K2cr2O1
5d07a84283
本次提交对核心模块进行了深度重构，引入 Pydantic 增强配置管理的类型安全性，并全面优化了插件管理系统。 主要变更详情： 1. 核心架构与配置 - 重构配置加载模块：引入 Pydantic 模型 (`core/config_models.py`)，提供严格的配置项类型检查、验证及默认值管理。 - 统一模块结构：规范化模块导入路径，移除冗余的 `__init__.py` 文件，提升项目结构的清晰度。 - 性能优化：集成 Redis 缓存支持 (`RedisManager`)，有效降低高频 API 调用开销，提升响应速度。 2. 插件系统升级 - 实现热重载机制：新增插件文件变更监听功能，支持开发过程中自动重载插件，提升开发效率。 - 优化生命周期管理：改进插件加载与卸载逻辑，支持精确卸载指定插件及其关联的命令、事件处理器和定时任务。 3. 功能特性增强 - 新增媒体 API：引入 `MediaAPI` 模块，封装图片、语音等富媒体资源的获取与处理接口。 - 完善权限体系：重构权限管理系统，实现管理员与操作员的分级控制，支持更细粒度的命令权限校验。 4. 代码质量与稳定性 - 全面类型修复：解决 `mypy` 静态类型检查发现的大量类型错误（包括 `CommandManager`、`EventFactory` 及 `Bot` API 签名不匹配问题）。 - 增强错误处理：优化消息处理管道的异常捕获机制，完善关键路径的日志记录，提升系统运行稳定性。
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
from core.handlers.event_handler import MessageHandler
|
|
from core.managers import command_manager, permission_manager
|
|
from core.permission import Permission
|
|
from models.events.message import MessageEvent
|
|
|
|
# 更新插件元信息以包含OP管理
|
|
__plugin_meta__ = {
|
|
"name": "权限管理",
|
|
"description": "管理机器人的管理员和操作员",
|
|
"usage": (
|
|
"/admin list - 列出所有管理员和操作员\n"
|
|
"/admin add_admin <QQ号> - 添加管理员\n"
|
|
"/admin remove_admin <QQ号> - 移除管理员\n"
|
|
"/admin add_op <QQ号> - 添加操作员\n"
|
|
"/admin remove_op <QQ号> - 移除操作员"
|
|
),
|
|
}
|
|
|
|
|
|
@command_manager.command("admin", permission=Permission.ADMIN)
|
|
async def admin_management(event: MessageEvent, args: str):
|
|
"""
|
|
处理所有权限管理相关的命令。
|
|
"""
|
|
parts = args.split()
|
|
if not parts:
|
|
await event.reply(f"用法不正确。\n\n{__plugin_meta__['usage']}")
|
|
return
|
|
|
|
subcommand = parts[0].lower()
|
|
|
|
if subcommand == "list":
|
|
await list_permissions(event)
|
|
return
|
|
|
|
# 处理需要QQ号的命令
|
|
if len(parts) < 2 or not parts[1].isdigit():
|
|
await event.reply(f"请提供有效的用户QQ号。\n用法: /admin {subcommand} <QQ号>")
|
|
return
|
|
|
|
try:
|
|
target_user_id = int(parts[1])
|
|
except ValueError:
|
|
await event.reply("无效的QQ号。")
|
|
return
|
|
|
|
# 安全检查
|
|
if target_user_id == event.user_id:
|
|
await event.reply("你不能操作自己的权限。")
|
|
return
|
|
if target_user_id == event.self_id:
|
|
await event.reply("你不能操作机器人自身的权限。")
|
|
return
|
|
|
|
# 根据子命令分发
|
|
if subcommand == "add_admin":
|
|
permission_manager.set_user_permission(target_user_id, Permission.ADMIN)
|
|
await event.reply(f"已成功添加管理员：{target_user_id}")
|
|
elif subcommand == "remove_admin":
|
|
permission_manager.set_user_permission(target_user_id, Permission.USER)
|
|
await event.reply(f"已成功移除管理员：{target_user_id}")
|
|
elif subcommand == "add_op":
|
|
permission_manager.set_user_permission(target_user_id, Permission.OP)
|
|
await event.reply(f"已成功添加操作员：{target_user_id}")
|
|
elif subcommand == "remove_op":
|
|
permission_manager.set_user_permission(target_user_id, Permission.USER)
|
|
await event.reply(f"已成功移除操作员：{target_user_id}")
|
|
else:
|
|
await event.reply(f"未知的子命令 '{subcommand}'。\n\n{__plugin_meta__['usage']}")
|
|
|
|
|
|
async def list_permissions(event: MessageEvent):
|
|
"""
|
|
列出所有具有特殊权限（管理员和操作员）的用户。
|
|
"""
|
|
permissions = permission_manager.get_all_user_permissions()
|
|
if not permissions:
|
|
await event.reply("当前没有配置任何特殊权限的用户。")
|
|
return
|
|
|
|
admins = {uid for uid, p in permissions.items() if p == 'admin'}
|
|
ops = {uid for uid, p in permissions.items() if p == 'op'}
|
|
|
|
reply_msg = "当前权限列表：\n"
|
|
if admins:
|
|
reply_msg += "--- 管理员 ---\n"
|
|
for user_id in admins:
|
|
reply_msg += f"- {user_id}\n"
|
|
if ops:
|
|
reply_msg += "--- 操作员 ---\n"
|
|
for user_id in ops:
|
|
reply_msg += f"- {user_id}\n"
|
|
|
|
await event.reply(reply_msg.strip())
|