83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
from fastapi import APIRouter, Request, Response
|
|
from app.core.mpclient import mp_client
|
|
from app.core.response import success_response, error_response
|
|
from app.models.database import get_db
|
|
from app.models.user import UserDB
|
|
import xml.etree.ElementTree as ET
|
|
from datetime import datetime
|
|
import hashlib
|
|
import time
|
|
from app.core.config import settings
|
|
import logging
|
|
|
|
router = APIRouter()
|
|
|
|
def check_signature(signature: str, timestamp: str, nonce: str) -> bool:
|
|
"""验证微信服务器签名"""
|
|
token = settings.MP_TOKEN # 在配置文件中添加 MP_TOKEN
|
|
# 按字典序排序
|
|
temp_list = [token, timestamp, nonce]
|
|
temp_list.sort()
|
|
# 拼接字符串
|
|
temp_str = "".join(temp_list)
|
|
# SHA1加密
|
|
sign = hashlib.sha1(temp_str.encode('utf-8')).hexdigest()
|
|
# 与微信发送的签名对比
|
|
return sign == signature
|
|
|
|
@router.get("")
|
|
async def verify_server(
|
|
signature: str,
|
|
timestamp: str,
|
|
nonce: str,
|
|
echostr: str
|
|
):
|
|
"""验证服务器地址的有效性"""
|
|
if check_signature(signature, timestamp, nonce):
|
|
return Response(content=echostr, media_type="text/plain")
|
|
return Response(status_code=403)
|
|
|
|
@router.post("")
|
|
async def handle_server(request: Request):
|
|
"""处理微信服务器推送的消息和事件"""
|
|
try:
|
|
# 读取原始XML数据
|
|
body = await request.body()
|
|
root = ET.fromstring(body)
|
|
|
|
logging.info(f"微信公众号消息：{root}")
|
|
|
|
# 解析基本信息
|
|
msg_type = root.find('MsgType').text
|
|
from_user = root.find('FromUserName').text
|
|
to_user = root.find('ToUserName').text
|
|
create_time = int(root.find('CreateTime').text)
|
|
|
|
# 获取数据库会话
|
|
db = next(get_db())
|
|
|
|
# 处理不同类型的消息和事件
|
|
if msg_type == 'event':
|
|
event = root.find('Event').text.lower()
|
|
|
|
if event == 'subscribe': # 关注事件
|
|
# 获取用户信息
|
|
user_info = await mp_client.get_user_info(from_user)
|
|
if user_info:
|
|
# 查找或创建用户
|
|
user = db.query(UserDB).filter(
|
|
UserDB.unionid == user_info.get('unionid')
|
|
).first()
|
|
|
|
if user:
|
|
# 更新用户信息
|
|
user.mp_openid = from_user
|
|
db.commit()
|
|
|
|
|
|
return Response(content="", media_type="text/plain")
|
|
|
|
except Exception as e:
|
|
logging.exception("处理微信消息异常")
|
|
# 返回空字符串表示接收成功
|
|
return Response(content="", media_type="text/plain") |