Mail

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

@@ -0,0 +1,39 @@
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class AgaveAccount(ProviderAccount):
+    def get_profile_url(self):
+        return self.account.extra_data.get("web_url", "dflt")
+
+    def get_avatar_url(self):
+        return self.account.extra_data.get("avatar_url", "dflt")
+
+    def to_str(self):
+        dflt = super(AgaveAccount, self).to_str()
+        return self.account.extra_data.get("name", dflt)
+
+
+class AgaveProvider(OAuth2Provider):
+    id = "agave"
+    name = "Agave"
+    account_class = AgaveAccount
+
+    def extract_uid(self, data):
+        return str(data.get("create_time"))
+
+    def extract_common_fields(self, data):
+        return dict(
+            email=data.get("email"),
+            username=data.get("username", ""),
+            name=(
+                (data.get("first_name", "") + " " + data.get("last_name", "")).strip()
+            ),
+        )
+
+    def get_default_scope(self):
+        scope = ["PRODUCTION"]
+        return scope
+
+
+provider_classes = [AgaveProvider]

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

@@ -0,0 +1,31 @@
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import AgaveProvider
+
+
+class AgaveTests(OAuth2TestsMixin, TestCase):
+    provider_id = AgaveProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+        {
+        "status": "success",
+        "message": "User details retrieved successfully.",
+        "version": "2.0.0-SNAPSHOT-rc3fad",
+        "result": {
+          "first_name": "John",
+          "last_name": "Doe",
+          "full_name": "John Doe",
+          "email": "jon@doe.edu",
+          "phone": "",
+          "mobile_phone": "",
+          "status": "Active",
+          "create_time": "20180322043812Z",
+          "username": "jdoe"
+          }
+        }
+        """,
+        )

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

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

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

@@ -0,0 +1,39 @@
+import requests
+
+from allauth.socialaccount import app_settings
+from allauth.socialaccount.providers.agave.provider import AgaveProvider
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+
+class AgaveAdapter(OAuth2Adapter):
+    provider_id = AgaveProvider.id
+
+    settings = app_settings.PROVIDERS.get(provider_id, {})
+    provider_base_url = settings.get("API_URL", "https://public.agaveapi.co")
+
+    access_token_url = "{0}/token".format(provider_base_url)
+    authorize_url = "{0}/authorize".format(provider_base_url)
+    profile_url = "{0}/profiles/v2/me".format(provider_base_url)
+
+    def complete_login(self, request, app, token, response):
+        extra_data = requests.get(
+            self.profile_url,
+            params={"access_token": token.token},
+            headers={
+                "Authorization": "Bearer " + token.token,
+            },
+        )
+
+        user_profile = (
+            extra_data.json()["result"] if "result" in extra_data.json() else {}
+        )
+
+        return self.get_provider().sociallogin_from_response(request, user_profile)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(AgaveAdapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(AgaveAdapter)