Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
def build_wecom_oauth_url(corp_id: str, redirect_uri: str, state: str = "") -> str:
|
|
params = {
|
|
"appid": corp_id,
|
|
"redirect_uri": redirect_uri,
|
|
"response_type": "code",
|
|
"scope": "snsapi_base",
|
|
}
|
|
query = "&".join(f"{k}={v}" for k, v in params.items())
|
|
url = f"https://open.weixin.qq.com/connect/oauth2/authorize?{query}"
|
|
if state:
|
|
url += f"&state={state}"
|
|
url += "#wechat_redirect"
|
|
return url
|
|
|
|
|
|
def build_mp_oauth_url(app_id: str, redirect_uri: str, state: str = "", scope: str = "snsapi_userinfo") -> str:
|
|
params = {
|
|
"appid": app_id,
|
|
"redirect_uri": redirect_uri,
|
|
"response_type": "code",
|
|
"scope": scope,
|
|
}
|
|
query = "&".join(f"{k}={v}" for k, v in params.items())
|
|
url = f"https://open.weixin.qq.com/connect/oauth2/authorize?{query}"
|
|
if state:
|
|
url += f"&state={state}"
|
|
url += "#wechat_redirect"
|
|
return url
|
|
|
|
|
|
async def handle_wecom_oauth_callback(code: str, corp_id: str, corp_secret: str) -> dict[str, Any]:
|
|
import httpx
|
|
|
|
url = "https://qyapi.weixin.qq.com/cgi-bin/auth/getuserinfo"
|
|
async with httpx.AsyncClient() as client:
|
|
resp = await client.get(url, params={"access_token": "", "code": code})
|
|
return resp.json()
|
|
|
|
|
|
async def handle_mp_oauth_callback(code: str, app_id: str, app_secret: str) -> dict[str, Any]:
|
|
import httpx
|
|
|
|
url = "https://api.weixin.qq.com/sns/oauth2/access_token"
|
|
async with httpx.AsyncClient() as client:
|
|
resp = await client.get(
|
|
url,
|
|
params={
|
|
"appid": app_id,
|
|
"secret": app_secret,
|
|
"code": code,
|
|
"grant_type": "authorization_code",
|
|
},
|
|
)
|
|
return resp.json()
|