使用FastAPI接收微信机器人webhook中的图片消息

在开发微信机器人webhook服务时，处理不同类型的消息是一个常见需求。本文将详细介绍如何使用FastAPI框架来接收并处理微信机器人发送的图片文件消息。

消息接收基础

微信机器人webhook通过HTTP POST请求发送消息到我们配置的接收端点。这些消息以表单形式(form-data)发送，包含以下几个关键字段：

• type: 消息类型，如图片消息为"file"
• content: 消息内容，对于图片消息这里是文件数据
• source: 消息来源信息，包含发送者等元数据

FastAPI实现方案

在FastAPI中，我们需要设计一个能够同时处理文本和文件消息的端点。以下是实现的核心要点：

1. 定义数据模型

首先创建一个Pydantic模型来描述接收的消息结构：

from fastapi import FastAPI, Form, File, UploadFile
from pydantic import BaseModel
from typing import Union, Optional

class WebhookPayload(BaseModel):
    type: str
    content: Union[str, UploadFile] = None
    source: Optional[str] = None
    isMentioned: Optional[str] = None
    isMsgFromSelf: Optional[str] = None

这个模型使用了Union类型，因为content字段可能是字符串(文本消息)或文件(图片等文件消息)。

2. 创建接收端点

app = FastAPI()

@app.post("/receive_msg")
async def receive_message(payload: WebhookPayload = Form()):
    if payload.type == 'file':
        # 处理文件消息
        filename = payload.content.filename
        sender_info = json.loads(payload.source)
        sender_name = sender_info['from']['payload']['name']
        
        # 保存文件
        file_content = await payload.content.read()
        with open(filename, 'wb') as f:
            f.write(file_content)
            
        return {"status": "file received"}
    else:
        # 处理文本消息
        return {"status": "text message received"}

3. 文件处理细节

当收到文件类型消息时，我们需要：

1. 从content字段获取文件对象(UploadFile类型)
2. 从source字段解析出发送者信息
3. 读取文件内容并保存到本地

UploadFile对象提供了方便的接口来访问文件名(filename属性)和文件内容(read()方法)。

注意事项

1. 文件大小限制：FastAPI默认有文件大小限制，大文件需要配置
2. 安全性：应验证文件名，防止路径遍历攻击
3. 错误处理：添加适当的异常处理
4. 性能考虑：大文件处理可能需要异步存储方案

总结

通过上述方法，我们可以构建一个能够同时处理文本和图片消息的微信机器人webhook接收端点。FastAPI的表单处理和文件上传功能使得这一实现变得简单而高效。开发者可以根据实际需求扩展这一基础实现，添加更多消息类型的处理逻辑。