Mail

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)