临时调整权限模块,备份

This commit is contained in:
liuwei
2025-06-09 14:12:31 +08:00
parent cedab1cefd
commit 9d15bf965b
30 changed files with 882 additions and 138 deletions

View File

@@ -5,13 +5,20 @@ from typing import Dict, Any, List, Optional, Tuple
from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
from utils.robot_cmd.robot_command import Feature, PermissionStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.decorator.points_decorator import plugin_points_cost
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from wechat_ipad import WechatAPIClient
class SystemUpdaterPlugin(MessagePluginInterface):
"""系统更新插件"""
# 功能权限常量
FEATURE_KEY = "SYSTEM_UPDATER"
FEATURE_DESCRIPTION = "🔄 系统更新功能 [更新系统, 系统更新]"
@property
def name(self) -> str:
return "系统更新"
@@ -36,11 +43,21 @@ class SystemUpdaterPlugin(MessagePluginInterface):
def commands(self) -> List[str]:
return self._commands
@property
def feature_key(self) -> Optional[str]:
return self.FEATURE_KEY
@property
def feature_description(self) -> Optional[str]:
return self.FEATURE_DESCRIPTION
def __init__(self):
super().__init__()
self.admin_wxids = []
self.wait_time = 5 # 默认等待15秒
self.bot: WechatAPIClient = None
# 注册功能权限
self.feature = self.register_feature()
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
@@ -87,47 +104,31 @@ class SystemUpdaterPlugin(MessagePluginInterface):
return False
@plugin_stats_decorator(plugin_name="系统更新")
@plugin_points_cost(1, "系统更新消耗积分", FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行: {self.name}{content}")
sender = message.get("sender")
roomid = message.get("roomid", "")
gbm = message.get("gbm", None)
gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot")
self.bot: WechatAPIClient = message.get("bot")
# 检查权限
if self.admin_wxids and sender not in self.admin_wxids:
await self.bot.send_text_message((roomid if roomid else sender), "⚠️ 您没有执行此操作的权限",
sender)
return True, "无权限"
if roomid and gbm.get_group_permission(roomid, self.feature) == PermissionStatus.DISABLED:
return False, "没有权限"
# 如果是群消息,检查群权限
if roomid and gbm and hasattr(gbm, 'get_group_permission'):
if gbm.get_group_permission(roomid, Feature.ROBOT) == PermissionStatus.DISABLED:
return False, "机器人功能已禁用"
# 检查是否是命令
if content in self._commands:
return True, "命令已接收"
# 提取等待时间参数
wait_time = self.wait_time
command = content.split(" ")[0]
# 检查是否是带参数的更新命令
for cmd in self._commands:
if content.startswith(f"{cmd} "):
return True, "带参数的命令已接收"
if len(content.split(" ")) > 1:
try:
param = content.split(" ")[1].strip()
if param.isdigit():
wait_time = int(param)
self.LOG.info(f"使用自定义等待时间: {wait_time}")
except:
pass
# 发送更新通知
await self.bot.send_text_message((roomid if roomid else sender),
f"🔄 系统即将更新并重启,等待时间设置为{wait_time}秒...",
sender)
# 执行系统更新脚本
self._execute_system_update()
return True, "系统更新中"
return False, "无法处理的消息"
def _execute_system_update(self):
"""执行系统更新操作"""