Mail

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

@@ -0,0 +1,23 @@
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class JupyterHubAccount(ProviderAccount):
+    def to_str(self):
+        dflt = super(JupyterHubAccount, self).to_str()
+        return self.account.extra_data.get("name", dflt)
+
+
+class JupyterHubProvider(OAuth2Provider):
+    id = "jupyterhub"
+    name = "JupyterHub"
+    account_class = JupyterHubAccount
+
+    def extract_uid(self, data):
+        return str(data.get("name"))
+
+    def extract_common_fields(self, data):
+        return dict(name=data.get("name", ""))
+
+
+provider_classes = [JupyterHubProvider]

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

@@ -0,0 +1,25 @@
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import JupyterHubProvider
+
+
+class JupyterHubTests(OAuth2TestsMixin, TestCase):
+    provider_id = JupyterHubProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+        {
+        "kind": "user",
+        "name": "abc",
+        "admin": false,
+        "groups": [],
+        "server": null,
+        "pending": null,
+        "created": "2016-12-06T18:30:50.297567Z",
+        "last_activity": "2017-02-07T17:29:36.470236Z",
+        "servers": null}
+        """,
+        )

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

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

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

@@ -0,0 +1,35 @@
+import requests
+
+from allauth.socialaccount import app_settings
+from allauth.socialaccount.providers.jupyterhub.provider import (
+    JupyterHubProvider,
+)
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+
+class JupyterHubAdapter(OAuth2Adapter):
+    provider_id = JupyterHubProvider.id
+
+    settings = app_settings.PROVIDERS.get(provider_id, {})
+    provider_base_url = settings.get("API_URL", "")
+
+    access_token_url = "{0}/hub/api/oauth2/token".format(provider_base_url)
+    authorize_url = "{0}/hub/api/oauth2/authorize".format(provider_base_url)
+    profile_url = "{0}/hub/api/user".format(provider_base_url)
+
+    def complete_login(self, request, app, access_token, **kwargs):
+        headers = {"Authorization": "Bearer {0}".format(access_token)}
+
+        extra_data = requests.get(self.profile_url, headers=headers)
+
+        user_profile = extra_data.json()
+
+        return self.get_provider().sociallogin_from_response(request, user_profile)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(JupyterHubAdapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(JupyterHubAdapter)