Kris
a1d9ba9683
本次提交包含多项优化与新增功能： 1. 清理多个文件中多余的空行与导入顺序 2. 修复voice.py中的多行字符串格式化问题 3. 新增微信公众号被动回复构建函数与配置项 4. 新增企业微信markdown消息发送支持 5. 新增消息去重TTL与最大条目配置 6. 新增markdown文本截断工具函数 7. 新增微信授权与OAuth相关工具方法 8. 重构消息去重逻辑，使用DedupPolicy替代本地字典实现 9. 新增子账号多租户支持功能 10. 新增消息动作处理适配器，支持send/reply等操作 11. 修复token持久化逻辑，新增状态存储支持
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
from yuxi.channels.models import MessageType
|
|
|
|
TRUNCATION_MARKER = "\n...(内容过长已截断)"
|
|
|
|
_MARKDOWN_BOUNDARY_PATTERN = re.compile(
|
|
r'(\n\n|\n(?=#{1,6}\s)|\n(?=>\s)|\n(?=-\s)|\n(?=\d+\.\s))'
|
|
)
|
|
|
|
|
|
def truncate_text(text: str, limit: int) -> str:
|
|
if len(text) <= limit:
|
|
return text
|
|
cut = max(0, limit - len(TRUNCATION_MARKER))
|
|
return text[:cut] + TRUNCATION_MARKER
|
|
|
|
|
|
def truncate_markdown(text: str, limit: int) -> str:
|
|
if len(text) <= limit:
|
|
return text
|
|
cut = max(0, limit - len(TRUNCATION_MARKER))
|
|
candidates = list(_MARKDOWN_BOUNDARY_PATTERN.finditer(text[: cut + 1]))
|
|
if candidates:
|
|
boundary = candidates[-1].start()
|
|
if boundary > cut * 0.7:
|
|
return text[:boundary] + TRUNCATION_MARKER
|
|
return text[:cut] + TRUNCATION_MARKER
|
|
|
|
|
|
def map_wecom_msg_type(msg_type: str) -> MessageType:
|
|
mapping = {
|
|
"text": MessageType.TEXT,
|
|
"image": MessageType.IMAGE,
|
|
"voice": MessageType.AUDIO,
|
|
"video": MessageType.VIDEO,
|
|
"file": MessageType.FILE,
|
|
"location": MessageType.LOCATION,
|
|
}
|
|
return mapping.get(msg_type, MessageType.TEXT)
|
|
|
|
|
|
def map_mp_msg_type(msg_type: str) -> MessageType:
|
|
mapping = {
|
|
"text": MessageType.TEXT,
|
|
"image": MessageType.IMAGE,
|
|
"voice": MessageType.AUDIO,
|
|
"video": MessageType.VIDEO,
|
|
"location": MessageType.LOCATION,
|
|
}
|
|
return mapping.get(msg_type, MessageType.TEXT)
|
|
|
|
|
|
def map_bridge_msg_type(msg_type: int) -> MessageType:
|
|
mapping = {
|
|
1: MessageType.TEXT,
|
|
3: MessageType.IMAGE,
|
|
34: MessageType.AUDIO,
|
|
43: MessageType.VIDEO,
|
|
48: MessageType.LOCATION,
|
|
49: MessageType.FILE,
|
|
}
|
|
return mapping.get(msg_type, MessageType.TEXT)
|