Mail

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

@@ -0,0 +1,51 @@
+from allauth.socialaccount import app_settings
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class VKAccount(ProviderAccount):
+    def get_profile_url(self):
+        return "https://vk.com/id%s" % self.account.extra_data.get("id")
+
+    def get_avatar_url(self):
+        ret = None
+        photo_big_url = self.account.extra_data.get("photo_big")
+        photo_medium_url = self.account.extra_data.get("photo_medium")
+        if photo_big_url:
+            return photo_big_url
+        elif photo_medium_url:
+            return photo_medium_url
+        else:
+            return ret
+
+    def to_str(self):
+        first_name = self.account.extra_data.get("first_name", "")
+        last_name = self.account.extra_data.get("last_name", "")
+        name = " ".join([first_name, last_name]).strip()
+        return name or super(VKAccount, self).to_str()
+
+
+class VKProvider(OAuth2Provider):
+    id = "vk"
+    name = "VK"
+    account_class = VKAccount
+
+    def get_default_scope(self):
+        scope = []
+        if app_settings.QUERY_EMAIL:
+            scope.append("email")
+        return scope
+
+    def extract_uid(self, data):
+        return str(data["id"])
+
+    def extract_common_fields(self, data):
+        return dict(
+            email=data.get("email"),
+            last_name=data.get("last_name"),
+            username=data.get("screen_name"),
+            first_name=data.get("first_name"),
+        )
+
+
+provider_classes = [VKProvider]

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

@@ -0,0 +1,31 @@
+from __future__ import absolute_import
+
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import VKProvider
+
+
+class VKTests(OAuth2TestsMixin, TestCase):
+    provider_id = VKProvider.id
+
+    def get_mocked_response(self, verified_email=True):
+        return MockedResponse(
+            200,
+            """
+{"response": [{"last_name": "Penners", "university_name": "",
+"photo": "http://vk.com/images/camera_c.gif", "sex": 2,
+"photo_medium": "http://vk.com/images/camera_b.gif", "relation": "0",
+"timezone": 1, "photo_big": "http://vk.com/images/camera_a.gif",
+"id": 219004864, "universities": [], "city": "1430", "first_name": "Raymond",
+"faculty_name": "", "online": 1, "counters": {"videos": 0, "online_friends": 0,
+"notes": 0, "audios": 0, "photos": 0, "followers": 0, "groups": 0,
+"user_videos": 0, "albums": 0, "friends": 0}, "home_phone": "", "faculty": 0,
+"nickname": "", "screen_name": "id219004864", "has_mobile": 1,
+"country": "139", "university": 0, "graduation": 0, "activity": "",
+"last_seen": {"time": 1377805189}}]}
+""",
+        )
+
+    def get_login_response_json(self, with_refresh_token=True):
+        return '{"user_id": 219004864, "access_token":"testac"}'

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

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

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

@@ -0,0 +1,63 @@
+import requests
+
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .provider import VKProvider
+
+
+USER_FIELDS = [
+    "first_name",
+    "last_name",
+    "nickname",
+    "screen_name",
+    "sex",
+    "bdate",
+    "city",
+    "country",
+    "timezone",
+    "photo",
+    "photo_medium",
+    "photo_big",
+    "photo_max_orig",
+    "has_mobile",
+    "contacts",
+    "education",
+    "online",
+    "counters",
+    "relation",
+    "last_seen",
+    "activity",
+    "universities",
+]
+
+
+class VKOAuth2Adapter(OAuth2Adapter):
+    provider_id = VKProvider.id
+    access_token_url = "https://oauth.vk.com/access_token"
+    authorize_url = "https://oauth.vk.com/authorize"
+    profile_url = "https://api.vk.com/method/users.get"
+
+    def complete_login(self, request, app, token, **kwargs):
+        uid = kwargs["response"].get("user_id")
+        params = {
+            "v": "5.95",
+            "access_token": token.token,
+            "fields": ",".join(USER_FIELDS),
+        }
+        if uid:
+            params["user_ids"] = uid
+        resp = requests.get(self.profile_url, params=params)
+        resp.raise_for_status()
+        extra_data = resp.json()["response"][0]
+        email = kwargs["response"].get("email")
+        if email:
+            extra_data["email"] = email
+        return self.get_provider().sociallogin_from_response(request, extra_data)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(VKOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(VKOAuth2Adapter)