Mail

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

@@ -0,0 +1,37 @@
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class Scope(object):
+    EXTERNAL_ID = "https://auth.snapchat.com/oauth2/api/user.external_id"
+    DISPLAY_NAME = "https://auth.snapchat.com/oauth2/api/user.display_name"
+    BITMOJI = "https://auth.snapchat.com/oauth2/api/user.bitmoji.avatar"
+
+
+class SnapchatAccount(ProviderAccount):
+    def to_str(self):
+        dflt = super(SnapchatAccount, self).to_str()
+        return "%s (%s)" % (
+            self.account.extra_data.get("data").get("me").get("displayName", ""),
+            dflt,
+        )
+
+
+class SnapchatProvider(OAuth2Provider):
+    id = "snapchat"
+    name = "Snapchat"
+    account_class = SnapchatAccount
+
+    def get_default_scope(self):
+        scope = [Scope.EXTERNAL_ID, Scope.DISPLAY_NAME]
+        return scope
+
+    def extract_uid(self, data):
+        return str(data.get("data").get("me").get("externalId"))
+
+    def extract_common_fields(self, data):
+        user = data.get("data", {}).get("me")
+        return {"name": user.get("displayName")}
+
+
+provider_classes = [SnapchatProvider]

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

@@ -0,0 +1,26 @@
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import SnapchatProvider
+
+
+class SnapchatOAuth2Tests(OAuth2TestsMixin, TestCase):
+    provider_id = SnapchatProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """{
+                  "data":{
+                      "me":{
+                        "externalId":"CAESIPiRBp0e5gLDq7VVurQ3rVdmdbqxpOJWynjyBL/xlo0w",
+                        "displayName":"Karun Shrestha",
+                        "bitmoji":{
+                            "avatar":"https://sdk.bitmoji.com/render/panel/336d1e96-9055-4818-81aa-adde45ec030f-3aBXH5B0ZPCr~grPTZScjprXRT2RkU90oSd7X_PjDFFnBe3wuFkD1R-v1.png?transparent=1&palette=1",
+                            "id":"3aBXH5B0ZPCr~grPTZScjprXRT2RkU90oSd7X_PjDFFnBe3wuFkD1R"
+                        }
+                      }
+                  },
+                  "errors":[]
+            }""",
+        )  # noqa

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

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

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

@@ -0,0 +1,52 @@
+import requests
+
+from allauth.socialaccount import app_settings
+from allauth.socialaccount.providers.oauth2.client import OAuth2Error
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .provider import Scope, SnapchatProvider
+
+
+class SnapchatOAuth2Adapter(OAuth2Adapter):
+    provider_id = SnapchatProvider.id
+
+    access_token_url = "https://accounts.snapchat.com/accounts/oauth2/token"
+    authorize_url = "https://accounts.snapchat.com/accounts/oauth2/auth"
+    identity_url = "https://api.snapkit.com/v1/me"
+
+    def complete_login(self, request, app, token, **kwargs):
+        extra_data = self.get_data(token.token)
+        return self.get_provider().sociallogin_from_response(request, extra_data)
+
+    def get_data(self, token):
+        provider_id = SnapchatProvider.id
+        settings = app_settings.PROVIDERS.get(provider_id, {})
+        provider_scope = settings.get(
+            "SCOPE",
+            "['https://auth.snapchat.com/oauth2/api/user.external_id', 'https://auth.snapchat.com/oauth2/api/user.display_name']",
+        )
+
+        hed = {
+            "Authorization": "Bearer " + token,
+            "Content-Type": "application/json;charset=UTF-8",
+        }
+        if Scope.BITMOJI in provider_scope:
+            data = {"query": "{ me { externalId displayName bitmoji { avatar id } } }"}
+        else:
+            data = {"query": "{ me { externalId displayName } }"}
+        resp = requests.post(self.identity_url, headers=hed, json=data)
+        resp.raise_for_status()
+        resp = resp.json()
+
+        if not resp.get("data"):
+            raise OAuth2Error()
+
+        return resp
+
+
+oauth2_login = OAuth2LoginView.adapter_view(SnapchatOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(SnapchatOAuth2Adapter)