first init

core/wechat/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding = utf-8 -*-
+# @Time : 2025/2/17 下午10:52
+# @File : __init__.py.py
+# @Software : PyCharm
+# @Author : xingc
+# @Desc :

core/wechat/api.py

@@ -0,0 +1,68 @@
+# -*- coding = utf-8 -*-
+# @Time : 2025/2/17 下午4:49
+# @File : api.py
+# @Software : PyCharm
+# @Author : xingc
+# @Desc :
+import time
+
+import httpx
+import ujson
+
+from core.db.engine import get_cache
+
+
+class WechatApi:
+    def __init__(self, appid, secret):
+        self.appid = appid
+        self.secret = secret
+        self._access_token = None
+        self._access_token_expire_time = None
+        self.session = httpx.AsyncClient()
+
+    @property
+    async def access_token(self):
+        if any([
+            self._access_token is None,
+            self._access_token_expire_time < time.time(),
+        ]):
+            await self.get_access_token()
+        return self._access_token
+
+    async def get_access_token(self):
+        async with get_cache() as redis_conn:
+            result_str = await redis_conn.get('access_token')
+
+            if result_str:
+                access_token, expire_time = result_str.split('|')
+                self._access_token = access_token
+                self._access_token_expire_time = int(expire_time)
+            else:
+                url = 'https://api.weixin.qq.com/cgi-bin/token'
+                params = {
+                    'appid': self.appid,
+                    'secret': self.secret,
+                    'grant_type': 'client_credential'
+                }
+                r = await self.session.get(url, params=params)
+                result = r.json()
+                if 'errcode' not in result_str:
+                    expires_in = result['expires_in']
+                    self._access_token_expire_time = time.time() + expires_in
+                    self._access_token = result['access_token']
+                    await redis_conn.set(
+                        'access_token', f'{self._access_token}|{self._access_token_expire_time}', ex=expires_in
+                    )
+
+            return self._access_token
+
+    async def code_to_session(self, code: str):
+        url = 'https://api.weixin.qq.com/sns/jscode2session'
+        params = {
+            'appid': self.appid,
+            'secret': self.secret,
+            'js_code': code,
+            'grant_type': 'authorization_code'
+        }
+        r = await self.session.get(url, params=params)
+        return r.json()