Kris
2348a04163
本次提交包含多项核心更新： 1. 新增微信4.0浅色主题UI模板资源与说明文档，补充了联系人图标、新的朋友入口等四个UI元素的图像匹配模板 2. 重构UI驱动层，将原2375行的单文件拆分为5个职责清晰的子驱动类，提升代码可维护性 3. 新增批量群发消息与聊天记录导出的完整数据模型、路由与后台协程，支持幂等校验与风控配置 4. 为所有业务接口新增post_verify校验逻辑，补充了verified字段返回操作结果 5. 优化媒体文件解密逻辑，修复了导出功能中的数据库错误处理与分辨率硬编码问题 6. 更新版本配置，将媒体发送功能标记为已落地，调整了能力列表与配置项
243 lines
8.4 KiB
Python
243 lines
8.4 KiB
Python
"""联系人域模型：联系人 / 联系人列表 / 群聊列表 / 群成员。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import enum
|
||
from typing import TYPE_CHECKING, Optional
|
||
|
||
from pydantic import BaseModel, Field
|
||
|
||
if TYPE_CHECKING:
|
||
from woc_bridge.messaging.friend_watcher import FriendRequestInfo
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 联系人接口
|
||
# ---------------------------------------------------------------------------
|
||
class Contact(BaseModel):
|
||
"""联系人。"""
|
||
|
||
wxid: str
|
||
nickname: str = ""
|
||
remark: str = ""
|
||
avatar_url: str = ""
|
||
type: str = Field(
|
||
default="person",
|
||
description="person / group / official / system",
|
||
)
|
||
|
||
# 扩展字段：从 contact.db 的 contact 表读取
|
||
alias: Optional[str] = Field(default=None, description="微信号/别名")
|
||
encrypt_username: Optional[str] = Field(default=None, description="加密用户名")
|
||
quan_pin: Optional[str] = Field(default=None, description="全拼")
|
||
pin_yin_initial: Optional[str] = Field(default=None, description="拼音首字母")
|
||
big_head_url: Optional[str] = Field(default=None, description="高清头像 URL")
|
||
small_head_url: Optional[str] = Field(default=None, description="缩略头像 URL")
|
||
description: Optional[str] = Field(default=None, description="个性签名/描述")
|
||
local_type: Optional[int] = Field(default=None, description="联系人类型标记")
|
||
verify_flag: Optional[int] = Field(default=None, description="认证标记")
|
||
delete_flag: Optional[int] = Field(default=None, description="删除标记")
|
||
chat_room_type: Optional[int] = Field(default=None, description="群类型标记")
|
||
|
||
|
||
class ContactsResponse(BaseModel):
|
||
"""联系人列表响应。"""
|
||
|
||
contacts: list[Contact] = Field(default_factory=list)
|
||
total: int = 0
|
||
|
||
|
||
class GroupsResponse(BaseModel):
|
||
"""群聊列表响应。"""
|
||
|
||
groups: list[Contact] = Field(default_factory=list)
|
||
total: int = 0
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 群成员接口
|
||
# ---------------------------------------------------------------------------
|
||
class GroupMember(BaseModel):
|
||
"""群成员。"""
|
||
|
||
wxid: str
|
||
nickname: str = ""
|
||
display_name: str = ""
|
||
is_admin: bool = False
|
||
|
||
|
||
class GroupMembersResponse(BaseModel):
|
||
"""群成员列表响应。"""
|
||
|
||
group_wxid: str
|
||
members: list[GroupMember] = Field(default_factory=list)
|
||
total: int = 0
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 修改好友备注（experimental）
|
||
# ---------------------------------------------------------------------------
|
||
class SetRemarkRequest(BaseModel):
|
||
"""修改好友备注请求（experimental）。"""
|
||
|
||
remark: str = Field(description="新备注名")
|
||
display_name: Optional[str] = Field(
|
||
default=None,
|
||
description="会话显示名（备注/昵称）用于搜索框定位；"
|
||
"不传则从 contact.db 解析（修改前的 remark 优先，其次 nickname，回退 wxid）",
|
||
)
|
||
|
||
|
||
class SetRemarkResponse(BaseModel):
|
||
"""修改好友备注响应。"""
|
||
|
||
success: bool = Field(default=False)
|
||
error: Optional[str] = Field(default=None, description="失败时的错误描述")
|
||
verified: Optional[bool] = Field(
|
||
default=None,
|
||
description="post_verify 校验结果：True=DB 中 remark 已更新为目标值，"
|
||
"False=超时仍为旧值，None=未执行校验",
|
||
)
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 添加好友（experimental）
|
||
# ---------------------------------------------------------------------------
|
||
class AddFriendRequest(BaseModel):
|
||
"""添加好友请求（experimental）。"""
|
||
|
||
keyword: str = Field(description="搜索关键词（wxid/手机号/微信号）")
|
||
message: str = Field(default="", description="可选好友验证消息")
|
||
|
||
|
||
class AddFriendResponse(BaseModel):
|
||
"""添加好友响应。"""
|
||
|
||
success: bool = Field(default=False)
|
||
error: Optional[str] = Field(default=None, description="失败时的错误描述")
|
||
verified: Optional[bool] = Field(
|
||
default=None,
|
||
description="post_verify 校验结果：True=检测到好友列表变化或输入框清空，"
|
||
"False=未检测到变化，None=未执行校验",
|
||
)
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 好友申请自动通过
|
||
# ---------------------------------------------------------------------------
|
||
class FriendRequestItem(BaseModel):
|
||
"""好友申请条目（API 响应用）。"""
|
||
stranger_wxid: str
|
||
nickname: str = ""
|
||
verify_message: str = ""
|
||
scene: str = ""
|
||
create_time: int
|
||
|
||
|
||
class FriendRequestsResponse(BaseModel):
|
||
"""好友申请列表响应。"""
|
||
requests: list[FriendRequestItem] = Field(default_factory=list)
|
||
total: int = 0
|
||
|
||
|
||
class AcceptFriendRequest(BaseModel):
|
||
"""手动通过好友申请请求。"""
|
||
stranger_wxid: str = Field(description="申请人 wxid（含 @stranger 后缀）")
|
||
nickname: Optional[str] = Field(default=None, description="申请人昵称（UI 定位用）")
|
||
|
||
|
||
class AcceptFriendResponse(BaseModel):
|
||
"""通过好友申请响应。"""
|
||
success: bool = Field(default=False)
|
||
error: Optional[str] = Field(default=None)
|
||
verified: Optional[bool] = Field(
|
||
default=None,
|
||
description="post_verify 校验结果：True=DB 中 @stranger 后缀消失已变为好友，"
|
||
"False=超时仍未通过，None=未执行校验",
|
||
)
|
||
|
||
|
||
class AutoAcceptStatus(BaseModel):
|
||
"""自动通过运行状态。"""
|
||
running: bool
|
||
enabled: bool
|
||
processed_count: int = 0
|
||
accepted_count: int = 0
|
||
rejected_count: int = 0
|
||
last_processed_time: Optional[int] = None
|
||
|
||
|
||
class AcceptRuleConfig(BaseModel):
|
||
"""自动通过规则配置。"""
|
||
enabled: bool = Field(default=False, description="全局开关")
|
||
accept_all: bool = Field(default=False, description="通过所有申请（忽略以下规则）")
|
||
whitelist_wxids: list[str] = Field(default_factory=list, description="白名单 wxid 列表")
|
||
whitelist_nicknames: list[str] = Field(default_factory=list, description="白名单昵称列表（精确匹配）")
|
||
keywords: list[str] = Field(default_factory=list, description="验证消息关键词列表（子串匹配）")
|
||
blacklist_wxids: list[str] = Field(default_factory=list, description="黑名单 wxid 列表")
|
||
blacklist_nicknames: list[str] = Field(default_factory=list, description="黑名单昵称列表")
|
||
allow_scenes: list[str] = Field(default_factory=list, description="允许的场景（空=不限制）")
|
||
|
||
|
||
class AcceptDecision(enum.Enum):
|
||
"""规则匹配决策结果。"""
|
||
ACCEPT = "accept"
|
||
REJECT = "reject"
|
||
SKIP = "skip"
|
||
|
||
|
||
class AcceptRuleEngine:
|
||
"""好友申请自动通过规则引擎。
|
||
|
||
按"黑名单 → 场景 → 白名单 → 关键词 → SKIP"优先级决策，
|
||
配置热更新受 asyncio.Lock 保护。
|
||
"""
|
||
|
||
def __init__(self, config: AcceptRuleConfig) -> None:
|
||
self._config = config
|
||
self._lock = asyncio.Lock()
|
||
|
||
async def get_config(self) -> AcceptRuleConfig:
|
||
"""获取当前规则配置（供 API 查询）。"""
|
||
async with self._lock:
|
||
return self._config
|
||
|
||
async def evaluate(self, req: "FriendRequestInfo") -> AcceptDecision:
|
||
async with self._lock:
|
||
cfg = self._config
|
||
|
||
if not cfg.enabled:
|
||
return AcceptDecision.SKIP
|
||
|
||
if cfg.accept_all:
|
||
return AcceptDecision.ACCEPT
|
||
|
||
# 黑名单优先
|
||
if req.stranger_wxid in cfg.blacklist_wxids:
|
||
return AcceptDecision.REJECT
|
||
if req.nickname in cfg.blacklist_nicknames:
|
||
return AcceptDecision.REJECT
|
||
|
||
# 场景过滤
|
||
if cfg.allow_scenes and req.scene not in cfg.allow_scenes:
|
||
return AcceptDecision.SKIP
|
||
|
||
# 白名单
|
||
if req.stranger_wxid in cfg.whitelist_wxids:
|
||
return AcceptDecision.ACCEPT
|
||
if req.nickname in cfg.whitelist_nicknames:
|
||
return AcceptDecision.ACCEPT
|
||
|
||
# 关键词
|
||
if cfg.keywords:
|
||
for kw in cfg.keywords:
|
||
if kw in req.verify_message:
|
||
return AcceptDecision.ACCEPT
|
||
|
||
return AcceptDecision.SKIP
|
||
|
||
async def update_config(self, config: AcceptRuleConfig) -> None:
|
||
async with self._lock:
|
||
self._config = config
|