Kris
7215418610
新增企业微信、微博、WhatsApp、Workplace 四个渠道扩展。 企业微信渠道扩展主要模块：config, gateway, webhook, webhook_bot, outbound, streaming, pairing, security, crypto, dedupe, persistent_dedupe, card, directory, events, externalcontact, media, mentions, menu, message, oauth, status 微博渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, passive_reply, broadcast, message, menu, media, subscription, template, status WhatsApp 渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, monitor, status Workplace 渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, challenge, groups, media, mentions, menu, monitor, persona, quick_reply, signature, subscriptions, template, threading, users, status
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
import logging
|
|
|
|
import httpx
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
AUTHORIZE_URL = "https://open.weixin.qq.com/connect/oauth2/authorize"
|
|
GET_USER_INFO_URL = "https://qyapi.weixin.qq.com/cgi-bin/user/getuserinfo"
|
|
GET_USER_DETAIL_URL = "https://qyapi.weixin.qq.com/cgi-bin/user/get"
|
|
|
|
|
|
def build_authorize_url(
|
|
corp_id: str,
|
|
redirect_uri: str,
|
|
state: str = "",
|
|
scope: str = "snsapi_base",
|
|
agent_id: int = 0,
|
|
) -> str:
|
|
params = {
|
|
"appid": corp_id,
|
|
"redirect_uri": redirect_uri,
|
|
"response_type": "code",
|
|
"scope": scope,
|
|
"state": state,
|
|
}
|
|
if agent_id > 0:
|
|
params["agentid"] = str(agent_id)
|
|
|
|
query = "&".join(f"{k}={v}" for k, v in params.items())
|
|
return f"{AUTHORIZE_URL}?{query}#wechat_redirect"
|
|
|
|
|
|
async def get_user_info(access_token: str, code: str) -> dict | None:
|
|
url = f"{GET_USER_INFO_URL}?access_token={access_token}&code={code}"
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
try:
|
|
resp = await client.get(url)
|
|
data = resp.json()
|
|
if data.get("errcode") == 0:
|
|
return {
|
|
"user_id": data.get("UserId"),
|
|
"device_id": data.get("DeviceId"),
|
|
"open_id": data.get("OpenId"),
|
|
}
|
|
logger.warning("WeCom OAuth get_user_info failed: %s", data)
|
|
return None
|
|
except Exception:
|
|
logger.exception("WeCom OAuth get_user_info error")
|
|
return None
|
|
|
|
|
|
async def get_user_detail(access_token: str, user_id: str) -> dict | None:
|
|
url = f"{GET_USER_DETAIL_URL}?access_token={access_token}&userid={user_id}"
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
try:
|
|
resp = await client.get(url)
|
|
data = resp.json()
|
|
if data.get("errcode") == 0:
|
|
return data
|
|
logger.warning("WeCom OAuth get_user_detail failed: %s", data)
|
|
return None
|
|
except Exception:
|
|
logger.exception("WeCom OAuth get_user_detail error")
|
|
return None
|