K2cr2O1
5d07a84283
本次提交对核心模块进行了深度重构，引入 Pydantic 增强配置管理的类型安全性，并全面优化了插件管理系统。 主要变更详情： 1. 核心架构与配置 - 重构配置加载模块：引入 Pydantic 模型 (`core/config_models.py`)，提供严格的配置项类型检查、验证及默认值管理。 - 统一模块结构：规范化模块导入路径，移除冗余的 `__init__.py` 文件，提升项目结构的清晰度。 - 性能优化：集成 Redis 缓存支持 (`RedisManager`)，有效降低高频 API 调用开销，提升响应速度。 2. 插件系统升级 - 实现热重载机制：新增插件文件变更监听功能，支持开发过程中自动重载插件，提升开发效率。 - 优化生命周期管理：改进插件加载与卸载逻辑，支持精确卸载指定插件及其关联的命令、事件处理器和定时任务。 3. 功能特性增强 - 新增媒体 API：引入 `MediaAPI` 模块，封装图片、语音等富媒体资源的获取与处理接口。 - 完善权限体系：重构权限管理系统，实现管理员与操作员的分级控制，支持更细粒度的命令权限校验。 4. 代码质量与稳定性 - 全面类型修复：解决 `mypy` 静态类型检查发现的大量类型错误（包括 `CommandManager`、`EventFactory` 及 `Bot` API 签名不匹配问题）。 - 增强错误处理：优化消息处理管道的异常捕获机制，完善关键路径的日志记录，提升系统运行稳定性。
240 lines
7.6 KiB
Python
240 lines
7.6 KiB
Python
"""
|
|
事件处理器模块
|
|
|
|
该模块定义了用于处理不同类型事件的处理器类。
|
|
每个处理器都负责注册和分发特定类型的事件。
|
|
"""
|
|
import inspect
|
|
from abc import ABC, abstractmethod
|
|
from typing import Any, Callable, Dict, List, Optional, Tuple
|
|
|
|
from ..bot import Bot
|
|
from ..config_loader import global_config
|
|
from ..managers.permission_manager import Permission
|
|
from ..utils.executor import run_in_thread_pool
|
|
|
|
|
|
class BaseHandler(ABC):
|
|
"""
|
|
事件处理器抽象基类
|
|
"""
|
|
def __init__(self):
|
|
self.handlers: List[Dict[str, Any]] = []
|
|
|
|
@abstractmethod
|
|
async def handle(self, bot: Bot, event: Any):
|
|
"""
|
|
处理事件
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
async def _run_handler(
|
|
self,
|
|
func: Callable,
|
|
bot: Bot,
|
|
event: Any,
|
|
args: Optional[List[str]] = None,
|
|
permission_granted: Optional[bool] = None
|
|
):
|
|
"""
|
|
智能执行事件处理器，并注入所需参数
|
|
"""
|
|
sig = inspect.signature(func)
|
|
params = sig.parameters
|
|
kwargs: Dict[str, Any] = {}
|
|
|
|
if "bot" in params:
|
|
kwargs["bot"] = bot
|
|
if "event" in params:
|
|
kwargs["event"] = event
|
|
if "args" in params and args is not None:
|
|
kwargs["args"] = args
|
|
if "permission_granted" in params and permission_granted is not None:
|
|
kwargs["permission_granted"] = permission_granted
|
|
|
|
if inspect.iscoroutinefunction(func):
|
|
result = await func(**kwargs)
|
|
else:
|
|
# 如果是同步函数，则放入线程池执行
|
|
result = await run_in_thread_pool(func, **kwargs)
|
|
return result is True
|
|
|
|
|
|
class MessageHandler(BaseHandler):
|
|
"""
|
|
消息事件处理器
|
|
"""
|
|
def __init__(self, prefixes: Tuple[str, ...]):
|
|
super().__init__()
|
|
self.prefixes = prefixes
|
|
self.commands: Dict[str, Dict] = {}
|
|
self.message_handlers: List[Dict[str, Any]] = []
|
|
|
|
def clear(self):
|
|
"""
|
|
清空所有已注册的消息和命令处理器
|
|
"""
|
|
self.commands.clear()
|
|
self.message_handlers.clear()
|
|
|
|
def unregister_by_plugin_name(self, plugin_name: str):
|
|
"""
|
|
根据插件名卸载相关的消息和命令处理器
|
|
"""
|
|
# 卸载命令
|
|
commands_to_remove = [name for name, info in self.commands.items() if info["plugin_name"] == plugin_name]
|
|
for name in commands_to_remove:
|
|
del self.commands[name]
|
|
|
|
# 卸载通用消息处理器
|
|
self.message_handlers = [h for h in self.message_handlers if h["plugin_name"] != plugin_name]
|
|
|
|
def on_message(self) -> Callable:
|
|
"""
|
|
注册通用消息处理器
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
module = inspect.getmodule(func)
|
|
plugin_name = module.__name__ if module else "Unknown"
|
|
self.message_handlers.append({"func": func, "plugin_name": plugin_name})
|
|
return func
|
|
return decorator
|
|
|
|
def command(
|
|
self,
|
|
*names: str,
|
|
permission: Optional[Permission] = None,
|
|
override_permission_check: bool = False
|
|
) -> Callable:
|
|
"""
|
|
注册命令处理器
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
module = inspect.getmodule(func)
|
|
plugin_name = module.__name__ if module else "Unknown"
|
|
for name in names:
|
|
self.commands[name] = {
|
|
"func": func,
|
|
"permission": permission,
|
|
"override_permission_check": override_permission_check,
|
|
"plugin_name": plugin_name,
|
|
}
|
|
return func
|
|
return decorator
|
|
|
|
async def handle(self, bot: Bot, event: Any):
|
|
"""
|
|
处理消息事件，分发给命令处理器或通用消息处理器
|
|
"""
|
|
from ..managers import permission_manager
|
|
for handler_info in self.message_handlers:
|
|
consumed = await self._run_handler(handler_info["func"], bot, event)
|
|
if consumed:
|
|
return
|
|
|
|
if not event.raw_message:
|
|
return
|
|
|
|
raw_text = event.raw_message.strip()
|
|
prefix_found = next((p for p in self.prefixes if raw_text.startswith(p)), None)
|
|
|
|
if not prefix_found:
|
|
return
|
|
|
|
command_parts = raw_text[len(prefix_found):].split()
|
|
if not command_parts:
|
|
return
|
|
|
|
command_name = command_parts[0]
|
|
args = command_parts[1:]
|
|
|
|
if command_name in self.commands:
|
|
command_info = self.commands[command_name]
|
|
func = command_info["func"]
|
|
permission = command_info.get("permission")
|
|
override_check = command_info.get("override_permission_check", False)
|
|
|
|
permission_granted = True
|
|
if permission:
|
|
permission_granted = await permission_manager.check_permission(event.user_id, permission)
|
|
|
|
if not permission_granted and not override_check:
|
|
permission_name = permission.name if isinstance(permission, Permission) else permission
|
|
message_template = global_config.bot.permission_denied_message
|
|
await bot.send(event, message_template.format(permission_name=permission_name))
|
|
return
|
|
|
|
await self._run_handler(
|
|
func,
|
|
bot,
|
|
event,
|
|
args=args,
|
|
permission_granted=permission_granted
|
|
)
|
|
|
|
|
|
class NoticeHandler(BaseHandler):
|
|
"""
|
|
通知事件处理器
|
|
"""
|
|
def clear(self):
|
|
self.handlers.clear()
|
|
|
|
def unregister_by_plugin_name(self, plugin_name: str):
|
|
"""
|
|
根据插件名卸载相关的通知处理器
|
|
"""
|
|
self.handlers = [h for h in self.handlers if h["plugin_name"] != plugin_name]
|
|
|
|
def register(self, notice_type: Optional[str] = None) -> Callable:
|
|
"""
|
|
注册通知处理器
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
module = inspect.getmodule(func)
|
|
plugin_name = module.__name__ if module else "Unknown"
|
|
self.handlers.append({"type": notice_type, "func": func, "plugin_name": plugin_name})
|
|
return func
|
|
return decorator
|
|
|
|
async def handle(self, bot: Bot, event: Any):
|
|
"""
|
|
处理通知事件
|
|
"""
|
|
for handler in self.handlers:
|
|
if handler["type"] is None or handler["type"] == event.notice_type:
|
|
await self._run_handler(handler["func"], bot, event)
|
|
|
|
|
|
class RequestHandler(BaseHandler):
|
|
"""
|
|
请求事件处理器
|
|
"""
|
|
def clear(self):
|
|
self.handlers.clear()
|
|
|
|
def unregister_by_plugin_name(self, plugin_name: str):
|
|
"""
|
|
根据插件名卸载相关的请求处理器
|
|
"""
|
|
self.handlers = [h for h in self.handlers if h["plugin_name"] != plugin_name]
|
|
|
|
def register(self, request_type: Optional[str] = None) -> Callable:
|
|
"""
|
|
注册请求处理器
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
module = inspect.getmodule(func)
|
|
plugin_name = module.__name__ if module else "Unknown"
|
|
self.handlers.append({"type": request_type, "func": func, "plugin_name": plugin_name})
|
|
return func
|
|
return decorator
|
|
|
|
async def handle(self, bot: Bot, event: Any):
|
|
"""
|
|
处理请求事件
|
|
"""
|
|
for handler in self.handlers:
|
|
if handler["type"] is None or handler["type"] == event.request_type:
|
|
await self._run_handler(handler["func"], bot, event)
|