Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from yuxi.channels.models import Attachment, ChannelResponse
|
|
|
|
|
|
class WeChatAttachmentAdapter:
|
|
MAX_FILE_SIZE = 20 * 1024 * 1024
|
|
|
|
async def prepare(self, response: ChannelResponse) -> dict[str, Any]:
|
|
result: dict[str, Any] = {"ok": True}
|
|
for att in response.attachments:
|
|
if att.size and att.size > self.MAX_FILE_SIZE:
|
|
result["ok"] = False
|
|
result.setdefault("errors", []).append(
|
|
f"File too large: {att.filename} ({att.size} > {self.MAX_FILE_SIZE})"
|
|
)
|
|
return result
|
|
|
|
@staticmethod
|
|
def build_attachment_inbound(attachment_data: dict) -> Attachment:
|
|
return Attachment(
|
|
file_id=attachment_data.get("media_id", ""),
|
|
filename=attachment_data.get("filename", ""),
|
|
mime_type=attachment_data.get("mime_type", "application/octet-stream"),
|
|
size=attachment_data.get("size"),
|
|
url=attachment_data.get("url"),
|
|
)
|
|
|
|
@staticmethod
|
|
def map_inbound_type(wechat_msg_type: str, mode: str = "mp") -> str:
|
|
if wechat_msg_type == "image":
|
|
return "image/png"
|
|
elif wechat_msg_type == "voice":
|
|
return "audio/amr"
|
|
elif wechat_msg_type == "video":
|
|
return "video/mp4"
|
|
elif wechat_msg_type == "file":
|
|
return "application/octet-stream"
|
|
return "application/octet-stream"
|
|
|
|
@staticmethod
|
|
def build_attachbounds(message: dict) -> list[dict[str, Any]]:
|
|
return []
|
|
|
|
@staticmethod
|
|
def adapt_response(response: dict, incoming_messages: list[dict]) -> dict:
|
|
return response
|