Mail

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

@@ -0,0 +1,41 @@
+from allauth.account.models import EmailAddress
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class OktaAccount(ProviderAccount):
+    def to_str(self):
+        dflt = super(OktaAccount, self).to_str()
+        return self.account.extra_data.get("name", dflt)
+
+
+class OktaProvider(OAuth2Provider):
+    id = "okta"
+    name = "Okta"
+    account_class = OktaAccount
+
+    def get_default_scope(self):
+        return ["openid", "profile", "email", "offline_access"]
+
+    def extract_uid(self, data):
+        return str(data["preferred_username"])
+
+    def extract_extra_data(self, data):
+        return data
+
+    def extract_email_addresses(self, data):
+        return [
+            EmailAddress(
+                email=data["email"], verified=bool(data["email_verified"]), primary=True
+            )
+        ]
+
+    def extract_common_fields(self, data):
+        return dict(
+            email=data["email"],
+            last_name=data["family_name"],
+            first_name=data["given_name"],
+        )
+
+
+provider_classes = [OktaProvider]

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

@@ -0,0 +1,28 @@
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import OktaProvider
+
+
+class OktaTests(OAuth2TestsMixin, TestCase):
+    provider_id = OktaProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+            {
+                "sub": "00u33ow83pjQpCQJr1j8",
+                "name": "Jon Smith",
+                "locale": "AE",
+                "email": "jsmith@example.com",
+                "nickname": "Jon Smith",
+                "preferred_username": "jsmith@example.com",
+                "given_name": "Jon",
+                "family_name": "Smith",
+                "zoneinfo": "America/Los_Angeles",
+                "updated_at": 1601285210,
+                "email_verified": true
+            }
+        """,
+        )

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

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

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

@@ -0,0 +1,59 @@
+import requests
+
+from allauth.socialaccount import app_settings
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .provider import OktaProvider
+
+
+class OktaOAuth2Adapter(OAuth2Adapter):
+    provider_id = OktaProvider.id
+
+    settings = app_settings.PROVIDERS.get(provider_id, {})
+    okta_base_url = settings.get("OKTA_BASE_URL")
+
+    @property
+    def access_token_url(self):
+        return "https://{}/oauth2/v1/token".format(self.okta_base_url)
+
+    @property
+    def authorize_url(self):
+        return "https://{}/oauth2/v1/authorize".format(self.okta_base_url)
+
+    @property
+    def userinfo_url(self):
+        return "https://{}/oauth2/v1/userinfo".format(self.okta_base_url)
+
+    @property
+    def access_token_method(self):
+        return "POST"
+
+    def complete_login(self, request, app, token, **kwargs):
+        """
+        Get the user info from userinfo endpoint and return a
+        A populated instance of the `SocialLogin` model (unsaved)
+
+        :param request:
+        :param app:
+        :param token:
+        :param kwargs:
+        :return:
+        """
+
+        resp = requests.get(
+            self.userinfo_url,
+            headers={"Authorization": "Bearer {}".format(token.token)},
+        )
+
+        resp.raise_for_status()
+        extra_data = resp.json()
+        login = self.get_provider().sociallogin_from_response(request, extra_data)
+        return login
+
+
+oauth2_login = OAuth2LoginView.adapter_view(OktaOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(OktaOAuth2Adapter)