Mail

venv/lib/python3.12/site-packages/allauth/socialaccount/providers/notion/client.py

@@ -0,0 +1,41 @@
+import requests
+from requests.auth import HTTPBasicAuth
+from urllib.parse import parse_qsl
+
+from django.utils.http import urlencode
+
+from allauth.socialaccount.providers.oauth2.client import (
+    OAuth2Client,
+    OAuth2Error,
+)
+
+
+class NotionOAuth2Client(OAuth2Client):
+    def get_redirect_url(self, authorization_url, extra_params):
+        params = {
+            "client_id": self.consumer_key,
+            "scope": self.scope,
+            "response_type": "code",
+            "owner": "user",
+        }
+        if self.state:
+            params["state"] = self.state
+        return "%s?%s" % (authorization_url, urlencode(params))
+
+    def get_access_token(self, code, pkce_code_verifier=None):
+        resp = requests.request(
+            self.access_token_method,
+            self.access_token_url,
+            auth=HTTPBasicAuth(self.consumer_key, self.consumer_secret),
+            json={"code": code, "grant_type": "authorization_code"},
+            headers=self.headers,
+        )
+        access_token = None
+        if resp.status_code in [200, 201]:
+            try:
+                access_token = resp.json()
+            except ValueError:
+                access_token = dict(parse_qsl(resp.text))
+        if not access_token or "access_token" not in access_token:
+            raise OAuth2Error("Error retrieving access token: %s" % resp.content)
+        return access_token

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

@@ -0,0 +1,53 @@
+from allauth.account.models import EmailAddress
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class NotionAccount(ProviderAccount):
+    def get_user(self):
+        return self.account.extra_data["owner"]["user"]
+
+    def get_name(self):
+        return self.get_user()["name"]
+
+    def get_avatar_url(self):
+        return self.get_user()["avatar_url"]
+
+    def get_workspace_name(self):
+        return self.account.extra_data["workspace_name"]
+
+    def get_workspace_icon(self):
+        return self.account.extra_data["workspace_icon"]
+
+    def to_str(self):
+        name = self.get_name()
+        workspace = self.get_workspace_name()
+        return f"{name} ({workspace})"
+
+
+class NotionProvider(OAuth2Provider):
+    id = "notion"
+    name = "Notion"
+    account_class = NotionAccount
+
+    def extract_uid(self, data):
+        """
+        The unique identifier for Notion is a combination of the User ID
+        and the Workspace ID they have authorized the application with.
+        """
+        user_id = data["owner"]["user"]["id"]
+        workspace_id = data["workspace_id"]
+        return "user-%s_workspace-%s" % (user_id, workspace_id)
+
+    def extract_common_fields(self, data):
+        user = data["owner"]["user"]
+        user["email"] = user["person"]["email"]
+        return user
+
+    def extract_email_addresses(self, data):
+        user = data["owner"]["user"]
+        email = user["person"]["email"]
+        return [EmailAddress(email=email, verified=True, primary=True)]
+
+
+provider_classes = [NotionProvider]

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

@@ -0,0 +1,89 @@
+from urllib.parse import parse_qs, urlparse
+
+from django.urls import reverse
+from django.utils.http import urlencode
+
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase, mocked_response
+
+from .provider import NotionProvider
+
+
+class NotionTests(OAuth2TestsMixin, TestCase):
+    provider_id = NotionProvider.id
+    pkce_enabled_default = False  # Notion does not support PKCE.
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+            {
+                "workspace_id": "workspace-abc",
+                "owner": {
+                    "user": {
+                        "id": "test123",
+                        "name": "John Doe",
+                        "avatar_url": "",
+                        "person": {"email": "john@example.com"},
+                    }
+                },
+            }
+            """,
+        )  # noqa
+
+    def get_login_response_json(self, with_refresh_token=False):
+        """
+        Docs here:
+        https://developers.notion.com/docs/authorization#step-4-notion-responds-with-an-access_token-and-additional-information
+        """
+        return """
+        {
+            "access_token": "test123",
+            "bot_id": "bot-abc",
+            "duplicated_template_id": "template-abc",
+            "owner": {
+            "workspace_id": "workspace-abc",
+                "user": {
+                    "id": "test123",
+                    "name": "John Doe",
+                    "avatar_url": "",
+                    "person": {
+                        "email": "john@example.com"
+                    }
+                }
+            },
+            "workspace_icon": "https://example.com/icon.png",
+            "workspace_id": "workspace-abc",
+            "workspace_name": "My Workspace"
+        }
+        """
+
+    def login(self, resp_mock=None, process="login", with_refresh_token=True):
+        resp = self.client.post(
+            reverse(self.provider.id + "_login")
+            + "?"
+            + urlencode(dict(process=process))
+        )
+
+        p = urlparse(resp["location"])
+        q = parse_qs(p.query)
+
+        complete_url = reverse(self.provider.id + "_callback")
+        response_json = self.get_login_response_json(
+            with_refresh_token=with_refresh_token
+        )
+
+        if isinstance(resp_mock, list):
+            resp_mocks = resp_mock
+        elif resp_mock is None:
+            resp_mocks = []
+        else:
+            resp_mocks = [resp_mock]
+
+        with mocked_response(
+            MockedResponse(200, response_json, {"content-type": "application/json"}),
+            *resp_mocks,
+        ):
+            resp = self.client.get(complete_url, self.get_complete_parameters(q))
+
+        return resp

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

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

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

@@ -0,0 +1,26 @@
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .client import NotionOAuth2Client
+from .provider import NotionProvider
+
+
+class NotionOAuth2Adapter(OAuth2Adapter):
+    provider_id = NotionProvider.id
+    basic_auth = True
+    client_class = NotionOAuth2Client
+
+    authorize_url = "https://api.notion.com/v1/oauth/authorize"
+    access_token_url = "https://api.notion.com/v1/oauth/token"
+
+    def complete_login(self, request, app, token, **kwargs):
+        return self.get_provider().sociallogin_from_response(
+            request, kwargs["response"]
+        )
+
+
+oauth2_login = OAuth2LoginView.adapter_view(NotionOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(NotionOAuth2Adapter)