Mail

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/robinhood/provider.py

@@ -0,0 +1,30 @@
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class RobinhoodAccount(ProviderAccount):
+    def get_avatar_url(self):
+        return None
+
+    def to_str(self):
+        return self.account.extra_data.get(
+            "username", super(RobinhoodAccount, self).to_str()
+        )
+
+
+class RobinhoodProvider(OAuth2Provider):
+    id = "robinhood"
+    name = "Robinhood"
+    account_class = RobinhoodAccount
+
+    def get_default_scope(self):
+        return ["read"]
+
+    def extract_uid(self, data):
+        return data["id"]
+
+    def extract_common_fields(self, data):
+        return dict(username=data.get("username"))
+
+
+provider_classes = [RobinhoodProvider]

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/robinhood/tests.py

@@ -0,0 +1,19 @@
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import RobinhoodProvider
+
+
+class RobinhoodTests(OAuth2TestsMixin, TestCase):
+    provider_id = RobinhoodProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+{
+  "username": "test_username",
+  "id": "1234-5678-910"
+}
+        """,
+        )

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/robinhood/urls.py

@@ -0,0 +1,6 @@
+from allauth.socialaccount.providers.oauth2.urls import default_urlpatterns
+
+from .provider import RobinhoodProvider
+
+
+urlpatterns = default_urlpatterns(RobinhoodProvider)

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/robinhood/views.py

@@ -0,0 +1,37 @@
+import requests
+
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .provider import RobinhoodProvider
+
+
+class RobinhoodOAuth2Adapter(OAuth2Adapter):
+    provider_id = RobinhoodProvider.id
+
+    @property
+    def authorize_url(self):
+        return "https://www.robinhood.com/oauth2/authorize/"
+
+    @property
+    def access_token_url(self):
+        return "https://api.robinhood.com/oauth2/token/"
+
+    @property
+    def profile_url(self):
+        return "https://api.robinhood.com/user/id/"
+
+    def complete_login(self, request, app, token, **kwargs):
+        response = requests.get(
+            self.profile_url,
+            headers={"Authorization": "Bearer %s" % token.token},
+        )
+        extra_data = response.json()
+        return self.get_provider().sociallogin_from_response(request, extra_data)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(RobinhoodOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(RobinhoodOAuth2Adapter)