v6ole
7455d7e426
130 issues auto-fixed (import ordering, UP045/UP006 type annotations), 33 issues manually fixed (E712/E501/E402/E722 + N818 rename + per-file wechat ignore for N8xx naming conventions). All 33 tests pass.
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
import time
|
|
|
|
import httpx
|
|
|
|
from app.config import settings
|
|
|
|
|
|
class WeChatClient:
|
|
def __init__(self):
|
|
self._access_token: str | None = None
|
|
self._token_expires_at: float = 0
|
|
|
|
async def _get_access_token(self) -> str | None:
|
|
now = time.time()
|
|
if self._access_token and now < self._token_expires_at:
|
|
return self._access_token
|
|
|
|
url = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
|
|
params = {
|
|
"corpid": settings.wechat_corp_id,
|
|
"corpsecret": settings.wechat_secret,
|
|
}
|
|
async with httpx.AsyncClient(timeout=30) as client:
|
|
response = await client.get(url, params=params)
|
|
data = response.json()
|
|
if data.get("errcode") == 0:
|
|
self._access_token = data["access_token"]
|
|
self._token_expires_at = now + data.get("expires_in", 7200) - 300
|
|
return self._access_token
|
|
return None
|
|
|
|
async def send_text(self, content: str, to_user: str = "@all") -> bool:
|
|
return await self._send_message("text", {"content": content}, to_user)
|
|
|
|
async def send_markdown(self, content: str, to_user: str = "@all") -> bool:
|
|
return await self._send_message("markdown", {"content": content}, to_user)
|
|
|
|
async def send_textcard(
|
|
self,
|
|
title: str,
|
|
description: str,
|
|
url: str,
|
|
to_user: str = "@all",
|
|
btn_txt: str = "查看详情",
|
|
) -> bool:
|
|
return await self._send_message(
|
|
"textcard",
|
|
{
|
|
"title": title,
|
|
"description": description,
|
|
"url": url,
|
|
"btntxt": btn_txt,
|
|
},
|
|
to_user,
|
|
)
|
|
|
|
async def _send_message(
|
|
self, msgtype: str, msg_data: dict, to_user: str = "@all"
|
|
) -> bool:
|
|
token = await self._get_access_token()
|
|
if not token:
|
|
return False
|
|
|
|
url = "https://qyapi.weixin.qq.com/cgi-bin/message/send"
|
|
params = {"access_token": token}
|
|
body = {
|
|
"touser": to_user,
|
|
"msgtype": msgtype,
|
|
"agentid": int(settings.wechat_agent_id),
|
|
msgtype: msg_data,
|
|
}
|
|
|
|
async with httpx.AsyncClient(timeout=30) as client:
|
|
response = await client.post(url, params=params, json=body)
|
|
data = response.json()
|
|
return data.get("errcode") == 0
|