247 lines
8.0 KiB
Python
247 lines
8.0 KiB
Python
import asyncio
|
||
import time
|
||
from aiohttp import web, ClientSession, FormData, ClientTimeout
|
||
from wechatpy.crypto import WeChatCrypto
|
||
from wechatpy.exceptions import InvalidSignatureException, WeChatClientException
|
||
from wechatpy.messages import (
|
||
Message, TextMessage, ImageMessage, VoiceMessage, VideoMessage,
|
||
LocationMessage, LinkMessage, EventMessage
|
||
)
|
||
from wechatpy.replies import (
|
||
TextReply, ImageReply, VoiceReply, VideoReply,
|
||
ArticlesReply, MusicReply
|
||
)
|
||
from ahserver.configuredServer import add_cleanupctx
|
||
from ahserver.serverenv import ServerEnv
|
||
from appPublic.log import debug, exception, error
|
||
|
||
class WOAHandler:
|
||
media_types = ['image', 'voice', 'video', 'thumb']
|
||
def __init__(self, msg_handler):
|
||
self.app_id = os.environ.get('WOA_APPID')
|
||
self.token = os.environ.get('WOA_TOKEN')
|
||
self.aes_key = os.environ.get('WOA_AESKEY')
|
||
self.secret = os.environ.get('WOA_SECRET')
|
||
# 初始化加解密组件 (如果未开启加密，wechatpy 也能处理，但传入 key 更规范)
|
||
self.crypto = WeChatCrypto(token, aes_key, app_id)
|
||
self.msg_handler = msg_handler
|
||
self.client = WeChatClient(APP_ID, APP_SECRET)
|
||
self.client_token = None
|
||
self.client_token_expire_in = 0
|
||
|
||
async def get_client_access_token(self):
|
||
tim = time.time()
|
||
if tim > self.client_token_expire_in - 20:
|
||
f = awaitify(self.client.fetch_access_token)
|
||
data = await f()
|
||
self.client_token = data['access_token']
|
||
self.client_token_expire_in = time.time() + data['expire_in']
|
||
return self.client_token
|
||
|
||
async def upload_media(self, media_type:str, file_path:str) -> str:
|
||
"""
|
||
upload media file to wechat and return a media id
|
||
"""
|
||
media_id = ''
|
||
access_token = await self.get_client_access_token()
|
||
url = f"https://api.weixin.qq.com/cgi-bin/media/upload?access_token={access_token}&type={media_type}"
|
||
|
||
form = FormData()
|
||
# 'media' 是微信接口规定的字段名
|
||
with open(file_path, 'rb') as f:
|
||
form.add_field('media', f, filename=os.path.basename(file_path))
|
||
|
||
timeout = ClientTimeout(total=60) # 上传大文件需要更长超时
|
||
|
||
async with ClientSession(timeout=timeout) as session:
|
||
async with session.post(url, data=form) as resp:
|
||
result = await resp.json()
|
||
|
||
if resp.status == 200 and 'media_id' in result:
|
||
debug(f"临时素材上传成功: {result['media_id']}")
|
||
return result['media_id']
|
||
else:
|
||
exception(f"{file_path}:上传失败: {result}")
|
||
raise Exception(f"WeChat API Error: {result}")
|
||
|
||
async def media_id2url(self, media_id:str) -> str:
|
||
if self.client_token is None:
|
||
f = awaitify(self.client.fetch_access_token)
|
||
self.client_token = await f()
|
||
url = f"https://api.weixin.qq.com/cgi-bin/media/get?access_token={self.client_token}&media_id={media_id}"
|
||
return url
|
||
|
||
async def messagehandler(self, request:web.Request, msg:Message):
|
||
"""
|
||
|
||
call self.msg_handler handle message
|
||
"""
|
||
dic = {}
|
||
|
||
if self.msg_handler:
|
||
return await self.msg_handler(reqest, msg)
|
||
return None
|
||
|
||
def _verify_signature(self, request):
|
||
"""验证签名并解密消息体（如果开启加密）"""
|
||
signature = request.query.get('signature', '')
|
||
timestamp = request.query.get('timestamp', '')
|
||
nonce = request.query.get('nonce', '')
|
||
echo_str = request.query.get('echostr', '')
|
||
|
||
# 1. 验证签名
|
||
try:
|
||
if echo_str:
|
||
# GET 请求：验证 URL
|
||
echo_str = self.crypto.check_signature(signature, timestamp, nonce, echo_str)
|
||
return True, echo_str
|
||
else:
|
||
# POST 请求：验证签名 (不返回 echo_str，只返回 True)
|
||
self.crypto.check_signature(signature, timestamp, nonce)
|
||
return True, None
|
||
except InvalidSignatureException:
|
||
logger.warning("签名验证失败")
|
||
return False, None
|
||
|
||
async def handle_get(self, request: web.Request) -> web.Response:
|
||
"""处理微信服务器 URL 验证"""
|
||
is_valid, echo_str = self._verify_signature(request)
|
||
if is_valid:
|
||
debug("URL 验证成功")
|
||
return web.Response(text=echo_str)
|
||
else:
|
||
return web.Response(text="failed", status=403)
|
||
|
||
async def handle_post(self, request: web.Request) -> web.Response:
|
||
"""处理用户消息和事件"""
|
||
# 1. 验证签名
|
||
is_valid, _ = self._verify_signature(request)
|
||
if not is_valid:
|
||
return web.Response(text="failed", status=403)
|
||
|
||
# 2. 读取并解密消息体
|
||
try:
|
||
body = await request.read()
|
||
# decrypt_message 会自动处理解密和 XML 解析，返回 wechatpy 的消息对象
|
||
# 如果未开启加密，它也会正常解析 XML
|
||
msg = self.crypto.decrypt_message(body)
|
||
except Exception as e:
|
||
logger.error(f"消息解密或解析失败: {e}")
|
||
return web.Response(text="success") # 防止微信重试
|
||
|
||
debug(f"收到消息类型: {type(msg).__name__}")
|
||
|
||
# 3. 业务逻辑路由
|
||
reply = None
|
||
|
||
if isinstance(msg, TextMessage):
|
||
content = msg.content.strip()
|
||
debug(f"用户文本: {content},{msg}")
|
||
|
||
# --- 模拟多媒体回复逻辑 ---
|
||
if content == "图片":
|
||
reply = self._get_image_reply(msg)
|
||
elif content == "语音":
|
||
reply = self._get_voice_reply(msg)
|
||
elif content == "视频":
|
||
reply = self._get_video_reply(msg)
|
||
elif content == "图文":
|
||
reply = self._get_articles_reply(msg)
|
||
else:
|
||
# 默认文本回复
|
||
reply = TextReply(message=msg, content=f"收到：{content}\n发送'图片'、'语音'、'视频'或'图文'测试多媒体回复。")
|
||
|
||
elif isinstance(msg, EventMessage):
|
||
if msg.event == 'subscribe':
|
||
reply = TextReply(message=msg, content="欢迎关注！发送指令测试功能。")
|
||
elif msg.event == 'unsubscribe':
|
||
debug(f"用户 {msg.source} 取消关注")
|
||
# 取消关注通常不需要回复，或者回复了用户也收不到
|
||
return web.Response(text="success")
|
||
|
||
# 4. 生成响应 XML
|
||
if reply:
|
||
# render() 方法将回复对象转换为 XML 字符串
|
||
response_xml = reply.render()
|
||
# 如果开启了加密模式，需要再次加密返回给微信
|
||
if self.aes_key:
|
||
response_xml = self.crypto.encrypt_message(response_xml, request.query.get('nonce'), request.query.get('timestamp'))
|
||
|
||
return web.Response(text=response_xml, content_type='application/xml')
|
||
|
||
return web.Response(text="success")
|
||
|
||
# --- 多媒体回复构造 helper ---
|
||
|
||
def _get_image_reply(self, msg):
|
||
"""
|
||
返回图片消息
|
||
注意：media_id 必须是通过微信素材接口上传后获得的永久或临时素材 ID
|
||
这里仅做演示，实际运行需替换为有效的 media_id
|
||
"""
|
||
# 在实际生产中，你需要先调用 client.material.add(...) 上传获取 media_id
|
||
demo_media_id = "DEMO_IMAGE_MEDIA_ID"
|
||
logger.warning(f"演示模式：图片 media_id 为 {demo_media_id}，请替换为真实 ID")
|
||
|
||
return ImageReply(
|
||
message=msg,
|
||
media_id=demo_media_id
|
||
)
|
||
|
||
def _get_voice_reply(self, msg):
|
||
"""返回语音消息"""
|
||
demo_media_id = "DEMO_VOICE_MEDIA_ID"
|
||
logger.warning(f"演示模式：语音 media_id 为 {demo_media_id}")
|
||
return VoiceReply(
|
||
message=msg,
|
||
media_id=demo_media_id
|
||
)
|
||
|
||
def _get_video_reply(self, msg):
|
||
"""返回视频消息"""
|
||
demo_media_id = "DEMO_VIDEO_MEDIA_ID"
|
||
logger.warning(f"演示模式：视频 media_id 为 {demo_media_id}")
|
||
return VideoReply(
|
||
message=msg,
|
||
media_id=demo_media_id,
|
||
title="演示视频",
|
||
description="这是一个通过 wechatpy 生成的视频回复"
|
||
)
|
||
|
||
def _get_articles_reply(self, msg):
|
||
"""返回图文消息 (多图文)"""
|
||
reply = ArticlesReply(message=msg)
|
||
|
||
# 添加第一篇文章
|
||
reply.add_article({
|
||
'title': '欢迎使用 WeChatPy',
|
||
'description': '这是由 aiohttp + wechatpy 驱动的自动回复示例。',
|
||
'image': 'https://img.alicdn.com/tfs/TB1.R77XVY7gK0jSZKpXXbT6VXa-1024-1024.png', # 封面图链接
|
||
'url': 'https://wechatpy.readthedocs.io/'
|
||
})
|
||
|
||
# 添加第二篇文章 (可选)
|
||
reply.add_article({
|
||
'title': 'GitHub 仓库',
|
||
'description': '点击访问源代码。',
|
||
'image': '', # 可以不填
|
||
'url': 'https://github.com/wechatpy/wechatpy'
|
||
})
|
||
|
||
return reply
|
||
|
||
async def woa_get(request):
|
||
pass
|
||
|
||
async def woa_post(request):
|
||
pass
|
||
async def setup_route(app):
|
||
app.router.add_get('/woa/wechat', woa_get)
|
||
app.router.add_post('/woa/wechat', woa_post)
|
||
yield
|
||
return
|
||
|
||
def load_woa():
|
||
add_cleanupctx(setup_route)
|
||
|