Dev (#29)
* 滚木 * feat: 重构核心架构,增强类型安全与插件管理 本次提交对核心模块进行了深度重构,引入 Pydantic 增强配置管理的类型安全性,并全面优化了插件管理系统。 主要变更详情: 1. 核心架构与配置 - 重构配置加载模块:引入 Pydantic 模型 (`core/config_models.py`),提供严格的配置项类型检查、验证及默认值管理。 - 统一模块结构:规范化模块导入路径,移除冗余的 `__init__.py` 文件,提升项目结构的清晰度。 - 性能优化:集成 Redis 缓存支持 (`RedisManager`),有效降低高频 API 调用开销,提升响应速度。 2. 插件系统升级 - 实现热重载机制:新增插件文件变更监听功能,支持开发过程中自动重载插件,提升开发效率。 - 优化生命周期管理:改进插件加载与卸载逻辑,支持精确卸载指定插件及其关联的命令、事件处理器和定时任务。 3. 功能特性增强 - 新增媒体 API:引入 `MediaAPI` 模块,封装图片、语音等富媒体资源的获取与处理接口。 - 完善权限体系:重构权限管理系统,实现管理员与操作员的分级控制,支持更细粒度的命令权限校验。 4. 代码质量与稳定性 - 全面类型修复:解决 `mypy` 静态类型检查发现的大量类型错误(包括 `CommandManager`、`EventFactory` 及 `Bot` API 签名不匹配问题)。 - 增强错误处理:优化消息处理管道的异常捕获机制,完善关键路径的日志记录,提升系统运行稳定性。 * feat: 添加测试用例并优化代码结构 refactor(permission_manager): 调整初始化顺序和逻辑 fix(admin_manager): 修复初始化逻辑和目录创建问题 feat(ws): 优化Bot实例初始化条件 feat(message): 增强MessageSegment功能并添加测试 feat(events): 支持字符串格式的消息解析 test: 添加核心功能测试用例 refactor(plugin_manager): 改进插件路径处理 style: 清理无用导入和代码 chore: 更新依赖项 * refactor(handler): 移除TYPE_CHECKING并直接导入Bot类 简化类型注解,直接导入Bot类而非使用TYPE_CHECKING条件导入,提高代码可读性和维护性 * fix(command_manager): 修复插件卸载时元信息移除不精确的问题 修复 CommandManager 中 unload_plugin 方法移除插件元信息时使用 startswith 导致可能误删其他插件的问题,改为精确匹配 同时调整相关测试用例验证精确匹配行为
This commit is contained in:
@@ -6,7 +6,7 @@
|
|||||||
"""
|
"""
|
||||||
from ..config_loader import global_config
|
from ..config_loader import global_config
|
||||||
from .admin_manager import AdminManager
|
from .admin_manager import AdminManager
|
||||||
from .command_manager import CommandManager
|
from .command_manager import matcher as command_manager
|
||||||
from .permission_manager import PermissionManager
|
from .permission_manager import PermissionManager
|
||||||
from .plugin_manager import PluginManager
|
from .plugin_manager import PluginManager
|
||||||
from .redis_manager import RedisManager
|
from .redis_manager import RedisManager
|
||||||
@@ -20,7 +20,6 @@ admin_manager = AdminManager()
|
|||||||
permission_manager = PermissionManager()
|
permission_manager = PermissionManager()
|
||||||
|
|
||||||
# 命令与事件管理器 (别名 matcher)
|
# 命令与事件管理器 (别名 matcher)
|
||||||
command_manager = CommandManager(prefixes=tuple(global_config.bot.command))
|
|
||||||
matcher = command_manager
|
matcher = command_manager
|
||||||
|
|
||||||
# 插件管理器
|
# 插件管理器
|
||||||
|
|||||||
@@ -93,7 +93,7 @@ class CommandManager:
|
|||||||
self.request_handler.unregister_by_plugin_name(plugin_name)
|
self.request_handler.unregister_by_plugin_name(plugin_name)
|
||||||
|
|
||||||
# 移除插件元信息
|
# 移除插件元信息
|
||||||
plugins_to_remove = [name for name in self.plugins if name.startswith(plugin_name)]
|
plugins_to_remove = [name for name in self.plugins if name == plugin_name]
|
||||||
for name in plugins_to_remove:
|
for name in plugins_to_remove:
|
||||||
del self.plugins[name]
|
del self.plugins[name]
|
||||||
|
|
||||||
|
|||||||
@@ -170,13 +170,20 @@ class PermissionManager(Singleton):
|
|||||||
user_permission = await self.get_user_permission(user_id)
|
user_permission = await self.get_user_permission(user_id)
|
||||||
return user_permission >= required_permission
|
return user_permission >= required_permission
|
||||||
|
|
||||||
def get_all_user_permissions(self) -> Dict[str, str]:
|
async def get_all_user_permissions(self) -> Dict[str, str]:
|
||||||
"""
|
"""
|
||||||
获取所有已配置的用户权限
|
获取所有已配置的用户权限(包括 AdminManager 中的管理员)
|
||||||
|
|
||||||
:return: 一个包含所有用户权限的字典
|
:return: 一个包含所有用户权限的字典
|
||||||
"""
|
"""
|
||||||
return self._data["users"].copy()
|
permissions = self._data["users"].copy()
|
||||||
|
|
||||||
|
# 合并 AdminManager 中的管理员
|
||||||
|
admins = await admin_manager.get_all_admins()
|
||||||
|
for admin_id in admins:
|
||||||
|
permissions[str(admin_id)] = Permission.ADMIN.value
|
||||||
|
|
||||||
|
return permissions
|
||||||
|
|
||||||
def get_all_users(self) -> Dict[str, str]:
|
def get_all_users(self) -> Dict[str, str]:
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -18,11 +18,11 @@ __plugin_meta__ = {
|
|||||||
|
|
||||||
|
|
||||||
@command_manager.command("admin", permission=Permission.ADMIN)
|
@command_manager.command("admin", permission=Permission.ADMIN)
|
||||||
async def admin_management(event: MessageEvent, args: str):
|
async def admin_management(event: MessageEvent, args: list[str]):
|
||||||
"""
|
"""
|
||||||
处理所有权限管理相关的命令。
|
处理所有权限管理相关的命令。
|
||||||
"""
|
"""
|
||||||
parts = args.split()
|
parts = args
|
||||||
if not parts:
|
if not parts:
|
||||||
await event.reply(f"用法不正确。\n\n{__plugin_meta__['usage']}")
|
await event.reply(f"用法不正确。\n\n{__plugin_meta__['usage']}")
|
||||||
return
|
return
|
||||||
@@ -73,7 +73,7 @@ async def list_permissions(event: MessageEvent):
|
|||||||
"""
|
"""
|
||||||
列出所有具有特殊权限(管理员和操作员)的用户。
|
列出所有具有特殊权限(管理员和操作员)的用户。
|
||||||
"""
|
"""
|
||||||
permissions = permission_manager.get_all_user_permissions()
|
permissions = await permission_manager.get_all_user_permissions()
|
||||||
if not permissions:
|
if not permissions:
|
||||||
await event.reply("当前没有配置任何特殊权限的用户。")
|
await event.reply("当前没有配置任何特殊权限的用户。")
|
||||||
return
|
return
|
||||||
|
|||||||
60
tests/test_plugin_reload_meta.py
Normal file
60
tests/test_plugin_reload_meta.py
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
|
||||||
|
import pytest
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
from core.managers.command_manager import CommandManager
|
||||||
|
|
||||||
|
class TestPluginReloadMeta:
|
||||||
|
def test_plugin_meta_persistence(self):
|
||||||
|
"""
|
||||||
|
测试插件加载、卸载和重载过程中元信息的持久性
|
||||||
|
"""
|
||||||
|
# 初始化 CommandManager
|
||||||
|
command_manager = CommandManager(prefixes=("/",))
|
||||||
|
|
||||||
|
# 模拟插件名称和元信息
|
||||||
|
plugin_name = "plugins.test_plugin"
|
||||||
|
plugin_meta = {
|
||||||
|
"name": "测试插件",
|
||||||
|
"description": "这是一个测试插件",
|
||||||
|
"usage": "/test"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 1. 模拟加载插件
|
||||||
|
command_manager.plugins[plugin_name] = plugin_meta
|
||||||
|
|
||||||
|
# 验证元信息已注册
|
||||||
|
assert plugin_name in command_manager.plugins
|
||||||
|
assert command_manager.plugins[plugin_name] == plugin_meta
|
||||||
|
|
||||||
|
# 2. 模拟卸载插件
|
||||||
|
command_manager.unload_plugin(plugin_name)
|
||||||
|
|
||||||
|
# 验证元信息已移除
|
||||||
|
assert plugin_name not in command_manager.plugins
|
||||||
|
|
||||||
|
# 3. 模拟重载插件(重新注册元信息)
|
||||||
|
# 在实际运行中,PluginManager 会在 reload 后重新赋值
|
||||||
|
command_manager.plugins[plugin_name] = plugin_meta
|
||||||
|
|
||||||
|
# 验证元信息已恢复
|
||||||
|
assert plugin_name in command_manager.plugins
|
||||||
|
assert command_manager.plugins[plugin_name] == plugin_meta
|
||||||
|
|
||||||
|
def test_unload_plugin_exact_match(self):
|
||||||
|
"""
|
||||||
|
测试 unload_plugin 是否只移除精确匹配的插件元信息
|
||||||
|
"""
|
||||||
|
command_manager = CommandManager(prefixes=("/",))
|
||||||
|
|
||||||
|
plugin1 = "plugins.test"
|
||||||
|
plugin2 = "plugins.test_extra"
|
||||||
|
|
||||||
|
command_manager.plugins[plugin1] = {"name": "Test 1"}
|
||||||
|
command_manager.plugins[plugin2] = {"name": "Test 2"}
|
||||||
|
|
||||||
|
# 卸载 plugin1
|
||||||
|
command_manager.unload_plugin(plugin1)
|
||||||
|
|
||||||
|
# 验证 plugin1 被移除,但 plugin2 仍然存在
|
||||||
|
assert plugin1 not in command_manager.plugins
|
||||||
|
assert plugin2 in command_manager.plugins
|
||||||
Reference in New Issue
Block a user