Merge branch 'MoFox-Studio:dev' into dev

This commit is contained in:
喵吃鸟
2025-11-27 22:48:31 +08:00
committed by GitHub
8 changed files with 625 additions and 77 deletions

View File

@@ -58,6 +58,7 @@ from sqlalchemy import (
Table,
inspect,
text,
types as sqltypes,
)
from sqlalchemy.engine import Engine, Connection
from sqlalchemy.exc import SQLAlchemyError
@@ -191,7 +192,7 @@ def get_database_config_from_toml(db_type: str) -> dict | None:
def create_sqlite_engine(sqlite_path: str) -> Engine:
"""创建 SQLite 引擎"""
"""<EFBFBD><EFBFBD><EFBFBD><EFBFBD> SQLite <EFBFBD><EFBFBD><EFBFBD><EFBFBD>"""
if not os.path.isabs(sqlite_path):
sqlite_path = os.path.join(PROJECT_ROOT, sqlite_path)
@@ -200,28 +201,18 @@ def create_sqlite_engine(sqlite_path: str) -> Engine:
url = f"sqlite:///{sqlite_path}"
logger.info("使用 SQLite 数据库: %s", sqlite_path)
return create_engine(url, future=True)
def create_mysql_engine(
host: str,
port: int,
database: str,
user: str,
password: str,
charset: str = "utf8mb4",
) -> Engine:
"""创建 MySQL 引擎"""
# 延迟导入 pymysql以便友好提示
try:
import pymysql # noqa: F401
except ImportError:
logger.error("需要安装 pymysql 才能连接 MySQL: pip install pymysql")
raise
url = f"mysql+pymysql://{user}:{password}@{host}:{port}/{database}?charset={charset}"
logger.info("使用 MySQL 数据库: %s@%s:%s/%s", user, host, port, database)
return create_engine(url, future=True)
engine = create_engine(
url,
future=True,
connect_args={
"timeout": 30, # wait a bit if the db is locked
"check_same_thread": False,
},
)
# Increase busy timeout to reduce "database is locked" errors on SQLite
with engine.connect() as conn:
conn.execute(text("PRAGMA busy_timeout=30000"))
return engine
def create_postgresql_engine(
@@ -324,22 +315,35 @@ def get_table_row_count(conn: Connection, table: Table) -> int:
def copy_table_structure(source_table: Table, target_metadata: MetaData, target_engine: Engine) -> Table:
"""在目标数据库中创建与源表结构相同的表
"""复制表结构到目标数据库,使其结构保持一致"""
target_is_sqlite = target_engine.dialect.name == "sqlite"
target_is_pg = target_engine.dialect.name == "postgresql"
Args:
source_table: 源表对象
target_metadata: 目标元数据对象
target_engine: 目标数据库引擎
columns = []
for c in source_table.columns:
new_col = c.copy()
Returns:
Table: 目标表对象
"""
# 复制表结构
# SQLite 不支持 nextval 等 server_default
if target_is_sqlite:
new_col.server_default = None
# PostgreSQL 需要将部分 SQLite 特有类型转换
if target_is_pg:
col_type = new_col.type
# SQLite DATETIME -> 通用 DateTime
if isinstance(col_type, sqltypes.DateTime) or col_type.__class__.__name__ in {"DATETIME", "DateTime"}:
new_col.type = sqltypes.DateTime()
# TEXT(50) 等长度受限的 TEXT 在 PG 无效,改用 String(length)
elif isinstance(col_type, sqltypes.Text) and getattr(col_type, "length", None):
new_col.type = sqltypes.String(length=col_type.length)
columns.append(new_col)
# 为避免迭代约束集合时出现 “Set changed size during iteration”这里不复制表级约束
target_table = Table(
source_table.name,
target_metadata,
*[c.copy() for c in source_table.columns],
*[c.copy() for c in source_table.constraints],
*columns,
)
target_metadata.create_all(target_engine, tables=[target_table])
return target_table
@@ -383,8 +387,6 @@ def migrate_table_data(
logger.error("查询表 %s 失败: %s", source_table.name, e)
return 0, 1
columns = source_table.columns.keys()
def insert_batch(rows: list[dict]):
nonlocal migrated_rows, error_count
if not rows:
@@ -399,7 +401,8 @@ def migrate_table_data(
batch: list[dict] = []
for row in result:
row_dict = {col: row[col] for col in columns}
# Use column objects to access row mapping to avoid quoted_name keys
row_dict = {col.key: row._mapping[col] for col in source_table.columns}
batch.append(row_dict)
if len(batch) >= batch_size:
insert_batch(batch)
@@ -535,6 +538,14 @@ class DatabaseMigrator:
# 目标数据库配置
target_config = self._load_target_config()
# 防止源/目标 SQLite 指向同一路径导致自我覆盖及锁
if (
self.source_type == "sqlite"
and self.target_type == "sqlite"
and os.path.abspath(source_config.get("path", "")) == os.path.abspath(target_config.get("path", ""))
):
raise ValueError("源数据库与目标数据库不能是同一个 SQLite 文件,请为目标指定不同的路径")
# 创建引擎
self.source_engine = create_engine_by_type(self.source_type, source_config)
self.target_engine = create_engine_by_type(self.target_type, target_config)
@@ -589,11 +600,16 @@ class DatabaseMigrator:
return sorted_tables
def _drop_target_tables(self, conn: Connection):
"""删除目标数据库中已经存在的表(谨慎操作
def _drop_target_tables(self):
"""删除目标数据库中已有的表(如果有
这里为了避免冲突,迁移前会询问用户是否删除目标库中已经存在的同名表。
使用 Engine.begin() 进行连接以支持 autobegin 和 begin 兼容 SQLAlchemy 2.0 的写法
"""
if self.target_engine is None:
logger.warning("目标数据库引擎尚未初始化,无法删除表")
return
with self.target_engine.begin() as conn:
inspector = inspect(conn)
existing_tables = inspector.get_table_names()
@@ -601,20 +617,19 @@ class DatabaseMigrator:
logger.info("目标数据库中没有已存在的表,无需删除")
return
logger.info("目标数据库中当前存在的表: %s", ", ".join(existing_tables))
if confirm_action("是否删除目标数据库中有的所有表?此操作不可恢复!", default=False):
with conn.begin():
logger.info("目标数据库中当前表: %s", ", ".join(existing_tables))
if confirm_action("是否删除目标数据库中有的表列表?此操作不可撤销", default=False):
for table_name in existing_tables:
try:
logger.info("删除目标数据库表: %s", table_name)
logger.info("删除目标数据库表: %s", table_name)
conn.execute(text(f"DROP TABLE IF EXISTS {table_name} CASCADE"))
except SQLAlchemyError as e:
logger.error("删除 %s 失败: %s", table_name, e)
logger.error("删除 %s 失败: %s", table_name, e)
self.stats["errors"].append(
f"删除 {table_name} 失败: {e}"
f"删除 {table_name} 失败: {e}"
)
else:
logger.info("用户选择保留目标数据库中已有的表,可能会与迁移数据发生冲突。")
logger.info("跳过删除目标数据库中的表,继续迁移过程")
def migrate(self):
"""执行迁移操作"""
@@ -630,8 +645,7 @@ class DatabaseMigrator:
logger.info("按依赖顺序迁移表: %s", ", ".join(t.name for t in tables))
# 删除目标库中已有表(可选)
with self.target_engine.connect() as target_conn:
self._drop_target_tables(target_conn)
self._drop_target_tables()
# 开始迁移
with self.source_engine.connect() as source_conn, self.target_engine.connect() as target_conn:
@@ -937,7 +951,7 @@ def interactive_setup() -> dict:
if target_type == "sqlite":
target_path = _ask_str(
"目标 SQLite 文件路径(若不存在会自动创建)",
default="data/MaiBot_target.db",
default="data/MaiBot.db",
)
target_config = {"path": target_path}
else:

View File

@@ -367,6 +367,7 @@ class BotInterestManager:
self.embedding_dimension,
current_dim,
)
return embedding
else:
raise RuntimeError(f"❌ 返回的embedding为空: {embedding}")

View File

@@ -1129,6 +1129,10 @@ class DefaultReplyer:
if reply_to:
# 兼容旧的reply_to
sender, target = self._parse_reply_target(reply_to)
# 回退逻辑:为 'reply_to' 路径提供 platform 和 user_id 的回退值,以修复 UnboundLocalError
# 这样就不再强制要求必须有 user_id解决了QQ空间插件等场景下的崩溃问题
platform = chat_stream.platform
user_id = ""
else:
# 对于 respond 动作reply_message 可能为 None统一回应未读消息
# 对于 reply 动作reply_message 必须存在(针对特定消息回复)

View File

@@ -64,7 +64,7 @@ TEMPLATE_DIR = os.path.join(PROJECT_ROOT, "template")
# 考虑到实际上配置文件中的mai_version是不会自动更新的,所以采用硬编码
# 对该字段的更新请严格参照语义化版本规范https://semver.org/lang/zh-CN/
MMC_VERSION = "0.13.0-alpha.3"
MMC_VERSION = "0.13.0-alpha.4"
# 全局配置变量
_CONFIG_INITIALIZED = False

View File

@@ -50,6 +50,8 @@ class NapcatAdapter(BaseAdapter):
host = config_api.get_plugin_config(plugin.config, "napcat_server.host", "localhost")
port = config_api.get_plugin_config(plugin.config, "napcat_server.port", 8095)
access_token = config_api.get_plugin_config(plugin.config, "napcat_server.access_token", "")
mode_str = config_api.get_plugin_config(plugin.config, "napcat_server.mode", "reverse")
ws_mode = "client" if mode_str == "direct" else "server"
ws_url = f"ws://{host}:{port}"
headers = {}
@@ -58,10 +60,11 @@ class NapcatAdapter(BaseAdapter):
else:
ws_url = "ws://127.0.0.1:8095"
headers = {}
ws_mode = "server"
# 配置 WebSocket 传输
transport = WebSocketAdapterOptions(
mode="server",
mode=ws_mode,
url=ws_url,
headers=headers if headers else None,
)

View File

@@ -0,0 +1,37 @@
"""Napcat 适配器事件类型定义"""
class NapcatEvent:
"""Napcat 适配器事件类型"""
class ON_RECEIVED:
"""接收事件"""
FRIEND_INPUT = "napcat.on_received.friend_input" # 好友正在输入
EMOJI_LIEK = "napcat.on_received.emoji_like" # 表情回复(注意:保持原来的拼写)
POKE = "napcat.on_received.poke" # 戳一戳
GROUP_UPLOAD = "napcat.on_received.group_upload" # 群文件上传
GROUP_BAN = "napcat.on_received.group_ban" # 群禁言
GROUP_LIFT_BAN = "napcat.on_received.group_lift_ban" # 群解禁
FRIEND_RECALL = "napcat.on_received.friend_recall" # 好友消息撤回
GROUP_RECALL = "napcat.on_received.group_recall" # 群消息撤回
class MESSAGE:
"""消息相关事件"""
GET_MSG = "napcat.message.get_msg" # 获取消息
class GROUP:
"""群组相关事件"""
SET_GROUP_BAN = "napcat.group.set_group_ban" # 设置群禁言
SET_GROUP_WHOLE_BAN = "napcat.group.set_group_whole_ban" # 设置全员禁言
SET_GROUP_KICK = "napcat.group.set_group_kick" # 踢出群聊
class FRIEND:
"""好友相关事件"""
SEND_LIKE = "napcat.friend.send_like" # 发送点赞
__all__ = ["NapcatEvent"]

View File

@@ -79,7 +79,7 @@ class MessageHandler:
# 获取群聊配置
group_list_type = features_config.get("group_list_type", "blacklist")
group_list = features_config.get("group_list", [])
group_list = [str(item) for item in features_config.get("group_list", [])]
if group_list_type == "blacklist":
# 黑名单模式:如果在黑名单中就过滤
@@ -96,7 +96,7 @@ class MessageHandler:
elif message_type == "private":
# 获取私聊配置
private_list_type = features_config.get("private_list_type", "blacklist")
private_list = features_config.get("private_list", [])
private_list = [str(item) for item in features_config.get("private_list", [])]
if private_list_type == "blacklist":
# 黑名单模式:如果在黑名单中就过滤

View File

@@ -2,40 +2,529 @@
from __future__ import annotations
from typing import TYPE_CHECKING, Any, Dict, Optional
import time
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple
from mofox_wire import MessageBuilder, SegPayload, UserInfoPayload
from src.common.logger import get_logger
from src.plugin_system.apis import config_api
from ...event_models import ACCEPT_FORMAT, NoticeType, QQ_FACE, PLUGIN_NAME
from ..utils import get_group_info, get_member_info, get_self_info, get_stranger_info, get_message_detail
if TYPE_CHECKING:
from ...plugin import NapcatAdapter
from ....plugin import NapcatAdapter
logger = get_logger("napcat_adapter")
class NoticeHandler:
"""处理 Napcat 通知事件(戳一戳、表情回复等)"""
"""处理 Napcat 通知事件(戳一戳、表情回复、禁言、文件上传等)"""
def __init__(self, adapter: "NapcatAdapter"):
self.adapter = adapter
self.plugin_config: Optional[Dict[str, Any]] = None
# 戳一戳防抖时间戳
self.last_poke_time: float = 0.0
def set_plugin_config(self, config: Dict[str, Any]) -> None:
"""设置插件配置"""
self.plugin_config = config
async def handle_notice(self, raw: Dict[str, Any]):
"""处理通知事件"""
# 简化版本:返回一个空的 MessageEnvelope
import time
import uuid
def _get_config(self, key: str, default: Any = None) -> Any:
"""获取插件配置的辅助方法"""
if not self.plugin_config:
return default
return config_api.get_plugin_config(self.plugin_config, key, default)
return {
"direction": "incoming",
"message_info": {
"platform": "qq",
"message_id": str(uuid.uuid4()),
"time": time.time(),
},
"message_segment": {"type": "text", "data": "[通知事件]"},
"timestamp_ms": int(time.time() * 1000),
async def handle_notice(self, raw: Dict[str, Any]):
"""
处理通知事件
Args:
raw: OneBot 原始通知数据
Returns:
MessageEnvelope (dict) or None
"""
notice_type = raw.get("notice_type")
message_time: float = time.time()
self_id = raw.get("self_id")
group_id = raw.get("group_id")
user_id = raw.get("user_id")
target_id = raw.get("target_id")
handled_segment: SegPayload | None = None
user_info: UserInfoPayload | None = None
system_notice: bool = False
notice_config: Dict[str, Any] = {
"is_notice": False,
"is_public_notice": False,
"target_id": target_id,
}
match notice_type:
case NoticeType.friend_recall:
logger.info("好友撤回一条消息")
logger.info(f"撤回消息ID{raw.get('message_id')}, 撤回时间:{raw.get('time')}")
logger.warning("暂时不支持撤回消息处理")
return None
case NoticeType.group_recall:
logger.info("群内用户撤回一条消息")
logger.info(f"撤回消息ID{raw.get('message_id')}, 撤回时间:{raw.get('time')}")
logger.warning("暂时不支持撤回消息处理")
return None
case NoticeType.notify:
sub_type = raw.get("sub_type")
match sub_type:
case NoticeType.Notify.poke:
if self._get_config("features.enable_poke", True):
logger.debug("处理戳一戳消息")
handled_segment, user_info = await self._handle_poke_notify(raw, group_id, user_id)
if handled_segment and user_info:
notice_config["notice_type"] = "poke"
notice_config["is_notice"] = True
else:
logger.warning("戳一戳消息被禁用,取消戳一戳处理")
return None
case NoticeType.Notify.input_status:
from src.plugin_system.core.event_manager import event_manager
from ...event_types import NapcatEvent
await event_manager.trigger_event(
NapcatEvent.ON_RECEIVED.FRIEND_INPUT,
permission_group=PLUGIN_NAME
)
return None
case _:
logger.warning(f"不支持的notify类型: {notice_type}.{sub_type}")
return None
case NoticeType.group_msg_emoji_like:
if self._get_config("features.enable_emoji_like", True):
logger.debug("处理群聊表情回复")
handled_segment, user_info = await self._handle_group_emoji_like_notify(
raw, group_id, user_id
)
if handled_segment and user_info:
notice_config["notice_type"] = "emoji_like"
notice_config["is_notice"] = True
else:
logger.warning("群聊表情回复被禁用,取消群聊表情回复处理")
return None
case NoticeType.group_ban:
sub_type = raw.get("sub_type")
match sub_type:
case NoticeType.GroupBan.ban:
logger.info("处理群禁言")
handled_segment, user_info = await self._handle_ban_notify(raw, group_id)
if handled_segment and user_info:
system_notice = True
user_id_in_ban = raw.get("user_id")
if user_id_in_ban == 0:
notice_config["notice_type"] = "group_whole_ban"
else:
notice_config["notice_type"] = "group_ban"
notice_config["is_notice"] = True
case NoticeType.GroupBan.lift_ban:
logger.info("处理解除群禁言")
handled_segment, user_info = await self._handle_lift_ban_notify(raw, group_id)
if handled_segment and user_info:
system_notice = True
user_id_in_ban = raw.get("user_id")
if user_id_in_ban == 0:
notice_config["notice_type"] = "group_whole_lift_ban"
else:
notice_config["notice_type"] = "group_lift_ban"
notice_config["is_notice"] = True
case _:
logger.warning(f"不支持的group_ban类型: {notice_type}.{sub_type}")
return None
case NoticeType.group_upload:
logger.info("群文件上传")
if user_id == self_id:
logger.info("检测到机器人自己上传文件,忽略此通知")
return None
handled_segment, user_info = await self._handle_group_upload_notify(
raw, group_id, user_id, self_id
)
if handled_segment and user_info:
notice_config["notice_type"] = "group_upload"
notice_config["is_notice"] = True
case _:
logger.warning(f"不支持的notice类型: {notice_type}")
return None
if not handled_segment or not user_info:
logger.warning("notice处理失败或不支持")
return None
# 使用 MessageBuilder 构建消息
msg_builder = MessageBuilder()
(
msg_builder.direction("incoming")
.message_id("notice")
.timestamp_ms(int(message_time * 1000))
.from_user(
user_id=str(user_info.get("user_id", "")),
platform="qq",
nickname=user_info.get("user_nickname", ""),
cardname=user_info.get("user_cardname", ""),
)
)
# 如果是群消息,添加群信息
if group_id:
fetched_group_info = await get_group_info(group_id)
group_name: str | None = None
if fetched_group_info:
group_name = fetched_group_info.get("group_name")
else:
logger.warning("无法获取notice消息所在群的名称")
msg_builder.from_group(
group_id=str(group_id),
platform="qq",
name=group_name or "",
)
# 设置格式信息
content_format = [handled_segment.get("type", "text")]
if "notify" not in content_format:
content_format.append("notify")
msg_builder.format_info(
content_format=content_format,
accept_format=ACCEPT_FORMAT,
)
# 设置消息段
msg_builder.seg_list([handled_segment])
# 设置 additional_config包含 notice 相关配置)
res = msg_builder.build()["message_info"]["additional_config"] = notice_config
return res
async def _handle_poke_notify(
self, raw: Dict[str, Any], group_id: Any, user_id: Any
) -> Tuple[SegPayload | None, UserInfoPayload | None]:
"""处理戳一戳通知"""
self_info: dict | None = await get_self_info()
if not self_info:
logger.error("自身信息获取失败")
return None, None
self_id = raw.get("self_id")
target_id = raw.get("target_id")
# 防抖检查:如果是针对机器人的戳一戳,检查防抖时间
if self_id == target_id:
current_time = time.time()
debounce_seconds = self._get_config("features.poke_debounce_seconds", 2.0)
if self.last_poke_time > 0:
time_diff = current_time - self.last_poke_time
if time_diff < debounce_seconds:
logger.debug(
f"戳一戳防抖:用户 {user_id} 的戳一戳被忽略(距离上次戳一戳 {time_diff:.2f} 秒)"
)
return None, None
self.last_poke_time = current_time
target_name: str | None = None
raw_info: list = raw.get("raw_info", [])
if group_id:
user_qq_info: dict | None = await get_member_info(group_id, user_id)
else:
user_qq_info: dict | None = await get_stranger_info(user_id)
if user_qq_info:
user_name = user_qq_info.get("nickname", "QQ用户")
user_cardname = user_qq_info.get("card", "")
else:
user_name = "QQ用户"
user_cardname = ""
logger.debug("无法获取戳一戳对方的用户昵称")
# 计算显示名称
display_name = ""
if self_id == target_id:
target_name = self_info.get("nickname", "")
elif self_id == user_id:
# 不发送机器人戳别人的消息
return None, None
else:
# 如果配置为忽略不是针对自己的戳一戳则直接返回None
if self._get_config("features.ignore_non_self_poke", False):
logger.debug("忽略不是针对自己的戳一戳消息")
return None, None
if group_id:
fetched_member_info: dict | None = await get_member_info(group_id, target_id)
if fetched_member_info:
target_name = fetched_member_info.get("nickname", "QQ用户")
else:
target_name = "QQ用户"
logger.debug("无法获取被戳一戳方的用户昵称")
display_name = user_name
else:
return None, None
# 解析戳一戳文本
first_txt: str = "戳了戳"
second_txt: str = ""
try:
if len(raw_info) > 2:
first_txt = raw_info[2].get("txt", "戳了戳")
if len(raw_info) > 4:
second_txt = raw_info[4].get("txt", "")
except Exception as e:
logger.warning(f"解析戳一戳消息失败: {str(e)},将使用默认文本")
user_info: UserInfoPayload = {
"platform": "qq",
"user_id": str(user_id),
"user_nickname": user_name,
"user_cardname": user_cardname,
}
seg_data: SegPayload = {
"type": "text",
"data": f"{display_name}{first_txt}{target_name}{second_txt}这是QQ的一个功能用于提及某人但没那么明显",
}
return seg_data, user_info
async def _handle_group_emoji_like_notify(
self, raw: Dict[str, Any], group_id: Any, user_id: Any
) -> Tuple[SegPayload | None, UserInfoPayload | None]:
"""处理群聊表情回复通知"""
if not group_id:
logger.error("群ID不能为空无法处理群聊表情回复通知")
return None, None
user_qq_info: dict | None = await get_member_info(group_id, user_id)
if user_qq_info:
user_name = user_qq_info.get("nickname", "QQ用户")
user_cardname = user_qq_info.get("card", "")
else:
user_name = "QQ用户"
user_cardname = ""
logger.debug("无法获取表情回复对方的用户昵称")
# 触发事件
from src.plugin_system.core.event_manager import event_manager
from ...event_types import NapcatEvent
target_message = await get_message_detail(raw.get("message_id", ""))
target_message_text = ""
if target_message:
target_message_text = target_message.get("raw_message", "")
else:
logger.error("未找到对应消息")
return None, None
if len(target_message_text) > 15:
target_message_text = target_message_text[:15] + "..."
user_info: UserInfoPayload = {
"platform": "qq",
"user_id": str(user_id),
"user_nickname": user_name,
"user_cardname": user_cardname,
}
likes_list = raw.get("likes", [])
like_emoji_id = ""
if likes_list and len(likes_list) > 0:
like_emoji_id = str(likes_list[0].get("emoji_id", ""))
# 触发表情回复事件
await event_manager.trigger_event(
NapcatEvent.ON_RECEIVED.EMOJI_LIEK,
permission_group=PLUGIN_NAME,
group_id=group_id,
user_id=user_id,
message_id=raw.get("message_id", ""),
emoji_id=like_emoji_id,
)
emoji_text = QQ_FACE.get(like_emoji_id, f"[表情{like_emoji_id}]")
seg_data: SegPayload = {
"type": "text",
"data": f"{user_name}使用Emoji表情{emoji_text}回应了消息[{target_message_text}]",
}
return seg_data, user_info
async def _handle_group_upload_notify(
self, raw: Dict[str, Any], group_id: Any, user_id: Any, self_id: Any
) -> Tuple[SegPayload | None, UserInfoPayload | None]:
"""处理群文件上传通知"""
if not group_id:
logger.error("群ID不能为空无法处理群文件上传通知")
return None, None
user_qq_info: dict | None = await get_member_info(group_id, user_id)
if user_qq_info:
user_name = user_qq_info.get("nickname", "QQ用户")
user_cardname = user_qq_info.get("card", "")
else:
user_name = "QQ用户"
user_cardname = ""
logger.debug("无法获取上传文件的用户昵称")
file_info = raw.get("file")
if not file_info:
logger.error("群文件上传通知中缺少文件信息")
return None, None
user_info: UserInfoPayload = {
"platform": "qq",
"user_id": str(user_id),
"user_nickname": user_name,
"user_cardname": user_cardname,
}
file_name = file_info.get("name", "未知文件")
file_size = file_info.get("size", 0)
seg_data: SegPayload = {
"type": "text",
"data": f"{user_name} 上传了文件: {file_name} (大小: {file_size} 字节)",
}
return seg_data, user_info
async def _handle_ban_notify(
self, raw: Dict[str, Any], group_id: Any
) -> Tuple[SegPayload | None, UserInfoPayload | None]:
"""处理群禁言通知"""
if not group_id:
logger.error("群ID不能为空无法处理禁言通知")
return None, None
# 获取操作者信息
operator_id = raw.get("operator_id")
operator_nickname: str = "QQ用户"
operator_cardname: str = ""
member_info: dict | None = await get_member_info(group_id, operator_id)
if member_info:
operator_nickname = member_info.get("nickname", "QQ用户")
operator_cardname = member_info.get("card", "")
else:
logger.warning("无法获取禁言执行者的昵称,消息可能会无效")
operator_info: UserInfoPayload = {
"platform": "qq",
"user_id": str(operator_id),
"user_nickname": operator_nickname,
"user_cardname": operator_cardname,
}
# 获取被禁言者信息
user_id = raw.get("user_id")
banned_user_info: Dict[str, Any] | None = None
user_nickname: str = "QQ用户"
user_cardname: str = ""
sub_type: str = ""
duration = raw.get("duration")
if duration is None:
logger.error("禁言时长不能为空,无法处理禁言通知")
return None, None
if user_id == 0: # 全体禁言
sub_type = "whole_ban"
else: # 单人禁言
sub_type = "ban"
fetched_member_info: dict | None = await get_member_info(group_id, user_id)
if fetched_member_info:
user_nickname = fetched_member_info.get("nickname", "QQ用户")
user_cardname = fetched_member_info.get("card", "")
banned_user_info = {
"platform": "qq",
"user_id": str(user_id),
"user_nickname": user_nickname,
"user_cardname": user_cardname,
}
seg_data: SegPayload = {
"type": "notify",
"data": {
"sub_type": sub_type,
"duration": duration,
"banned_user_info": banned_user_info,
},
}
return seg_data, operator_info
async def _handle_lift_ban_notify(
self, raw: Dict[str, Any], group_id: Any
) -> Tuple[SegPayload | None, UserInfoPayload | None]:
"""处理解除群禁言通知"""
if not group_id:
logger.error("群ID不能为空无法处理解除禁言通知")
return None, None
# 获取操作者信息
operator_id = raw.get("operator_id")
operator_nickname: str = "QQ用户"
operator_cardname: str = ""
member_info: dict | None = await get_member_info(group_id, operator_id)
if member_info:
operator_nickname = member_info.get("nickname", "QQ用户")
operator_cardname = member_info.get("card", "")
else:
logger.warning("无法获取解除禁言执行者的昵称,消息可能会无效")
operator_info: UserInfoPayload = {
"platform": "qq",
"user_id": str(operator_id),
"user_nickname": operator_nickname,
"user_cardname": operator_cardname,
}
# 获取被解除禁言者信息
sub_type: str = ""
user_nickname: str = "QQ用户"
user_cardname: str = ""
lifted_user_info: Dict[str, Any] | None = None
user_id = raw.get("user_id")
if user_id == 0: # 全体禁言解除
sub_type = "whole_lift_ban"
else: # 单人禁言解除
sub_type = "lift_ban"
fetched_member_info: dict | None = await get_member_info(group_id, user_id)
if fetched_member_info:
user_nickname = fetched_member_info.get("nickname", "QQ用户")
user_cardname = fetched_member_info.get("card", "")
else:
logger.warning("无法获取解除禁言消息发送者的昵称,消息可能会无效")
lifted_user_info = {
"platform": "qq",
"user_id": str(user_id),
"user_nickname": user_nickname,
"user_cardname": user_cardname,
}
seg_data: SegPayload = {
"type": "notify",
"data": {
"sub_type": sub_type,
"lifted_user_info": lifted_user_info,
}
}
return seg_data, operator_info