Mail

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

@@ -0,0 +1,52 @@
+import requests
+from collections import OrderedDict
+
+from django.utils.http import urlencode
+
+from allauth.socialaccount.providers.oauth2.client import (
+    OAuth2Client,
+    OAuth2Error,
+)
+
+
+class WeixinOAuth2Client(OAuth2Client):
+    def get_redirect_url(self, authorization_url, extra_params):
+        params = {
+            "appid": self.consumer_key,
+            "redirect_uri": self.callback_url,
+            "scope": self.scope,
+            "response_type": "code",
+        }
+        if self.state:
+            params["state"] = self.state
+        params.update(extra_params)
+        sorted_params = OrderedDict()
+        for param in sorted(params):
+            sorted_params[param] = params[param]
+        return "%s?%s" % (authorization_url, urlencode(sorted_params))
+
+    def get_access_token(self, code, pkce_code_verifier=None):
+        data = {
+            "appid": self.consumer_key,
+            "redirect_uri": self.callback_url,
+            "grant_type": "authorization_code",
+            "secret": self.consumer_secret,
+            "scope": self.scope,
+            "code": code,
+        }
+        params = None
+        self._strip_empty_keys(data)
+        url = self.access_token_url
+        if self.access_token_method == "GET":
+            params = data
+            data = None
+        if data and pkce_code_verifier:
+            data["code_verifier"] = pkce_code_verifier
+        # TODO: Proper exception handling
+        resp = requests.request(self.access_token_method, url, params=params, data=data)
+        access_token = None
+        if resp.status_code == 200:
+            access_token = resp.json()
+        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/weixin/provider.py

@@ -0,0 +1,30 @@
+from allauth.socialaccount.providers.base import ProviderAccount
+from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
+
+
+class WeixinAccount(ProviderAccount):
+    def get_avatar_url(self):
+        return self.account.extra_data.get("headimgurl")
+
+    def to_str(self):
+        return self.account.extra_data.get(
+            "nickname", super(WeixinAccount, self).to_str()
+        )
+
+
+class WeixinProvider(OAuth2Provider):
+    id = "weixin"
+    name = "Weixin"
+    account_class = WeixinAccount
+
+    def extract_uid(self, data):
+        return data["openid"]
+
+    def get_default_scope(self):
+        return ["snsapi_login"]
+
+    def extract_common_fields(self, data):
+        return dict(username=data.get("nickname"), name=data.get("nickname"))
+
+
+provider_classes = [WeixinProvider]

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

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from allauth.socialaccount.tests import OAuth2TestsMixin
+from allauth.tests import MockedResponse, TestCase
+
+from .provider import WeixinProvider
+
+
+class WeixinTests(OAuth2TestsMixin, TestCase):
+    provider_id = WeixinProvider.id
+
+    def get_mocked_response(self):
+        return MockedResponse(
+            200,
+            """
+{"access_token":
+ "OezXcEiiBSKSxW0eoylIeO5cPxb4Ks1RpbXGMv9uiV35032zNHGzXcld-EKsSScE3gRZMrUU78skCbp1ShtZnR0dQB8Wr_LUf7FA-H97Lnd2HgQah_GnkQex-vPFsGEwPPcNAV6q1Vz3uRNgL0MUFg",
+ "city": "Pudong New District",
+ "country": "CN",
+ "expires_in": 7200,
+ "headimgurl":
+ "http://wx.qlogo.cn/mmopen/VkvLVEpoJiaibYsVyW8GzxHibzlnqSM7iaX09r6TWUJXCNQHibHz37krvN65HR1ibEpgH5K5sukcIzA3r1C4KQ9qyyX9XIUdY9lNOk/0",
+ "language": "zh_CN",
+ "nickname": "某某某",
+ "openid": "ohS-VwAJ9GEXlplngwybJ3Z-ZHrI",
+ "privilege": [],
+ "province": "Shanghai",
+ "refresh_token":
+ "OezXcEiiBSKSxW0eoylIeO5cPxb4Ks1RpbXGMv9uiV35032zNHGzXcld-EKsSScEbMnnMqVExcSpj7KRAuBA8BU2j2e_FK5dgBe-ro32k7OuHtznwqqBn5QR7LZGo2-P8G7gG0eitjyZ751sFlnTAw",
+ "scope": "snsapi_login",
+ "sex": 1,
+ "unionid": "ohHrhwKnD9TOunEW0eKTS45vS5Qo"}""",
+        )  # noqa

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

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

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

@@ -0,0 +1,44 @@
+import requests
+
+from allauth.socialaccount.providers.oauth2.views import (
+    OAuth2Adapter,
+    OAuth2CallbackView,
+    OAuth2LoginView,
+)
+
+from .client import WeixinOAuth2Client
+from .provider import WeixinProvider
+
+
+class WeixinOAuth2Adapter(OAuth2Adapter):
+    provider_id = WeixinProvider.id
+    access_token_url = "https://api.weixin.qq.com/sns/oauth2/access_token"
+    profile_url = "https://api.weixin.qq.com/sns/userinfo"
+    client_class = WeixinOAuth2Client
+
+    @property
+    def authorize_url(self):
+        settings = self.get_provider().get_settings()
+        url = settings.get(
+            "AUTHORIZE_URL", "https://open.weixin.qq.com/connect/qrconnect"
+        )
+        return url
+
+    def complete_login(self, request, app, token, **kwargs):
+        openid = kwargs.get("response", {}).get("openid")
+        resp = requests.get(
+            self.profile_url,
+            params={"access_token": token.token, "openid": openid},
+        )
+        resp.raise_for_status()
+        extra_data = resp.json()
+        nickname = extra_data.get("nickname")
+        if nickname:
+            extra_data["nickname"] = nickname.encode("raw_unicode_escape").decode(
+                "utf-8"
+            )
+        return self.get_provider().sociallogin_from_response(request, extra_data)
+
+
+oauth2_login = OAuth2LoginView.adapter_view(WeixinOAuth2Adapter)
+oauth2_callback = OAuth2CallbackView.adapter_view(WeixinOAuth2Adapter)