update 0.0.6.0

utils/manager/admin_manager.py

@@ -1,8 +1,5 @@
-from typing import Optional, Dict
 from .data_class import StaticData
-from utils.utils import FreqLimiter
-from services.log import logger
-from pathlib import Path
+from typing import List, Optional
 
 
 class AdminManager(StaticData):
@@ -13,15 +10,27 @@ class AdminManager(StaticData):
     def __init__(self):
         super().__init__(None)
 
-    def add_admin_command(self, plugin: str, level: int):
+    def add_admin_plugin_settings(self, plugin: str, cmd: List[str], level: int):
         """
         添加一个管理员命令
+        :param plugin: 模块
+        :param cmd: 别名
+        :param level: 等级
+        """
+        self._data[plugin] = {
+            "level": level,
+            "cmd": cmd,
+        }
+
+    def set_admin_level(self, plugin: str, level: int):
+        """
+        设置管理员命令等级
         :param plugin: 模块名
         :param level: 权限等级
         """
-        self._data[plugin] = level
+        self._data[plugin]["level"] = level
 
-    def remove_admin_command(self, plugin: str):
+    def remove_admin_plugin_settings(self, plugin: str):
         """
         删除一个管理员命令
         :param plugin: 模块名
@@ -36,7 +45,7 @@ class AdminManager(StaticData):
         :param level: 权限等级
         """
         if plugin in self._data.keys():
-            return level >= self._data[plugin]
+            return level >= self._data[plugin]["level"]
         return True
 
     def get_plugin_level(self, plugin: str) -> int:
@@ -45,7 +54,15 @@ class AdminManager(StaticData):
         :param plugin: 模块名
         """
         if plugin in self._data.keys():
-            return self._data[plugin]
+            return self._data[plugin]["level"]
         return 0
 
-
+    def get_plugin_module(self, cmd: str) -> Optional[str]:
+        """
+        根据 cmd 获取功能 module
+        :param cmd: 命令
+        """
+        for key in self._data.keys():
+            if self._data[key].get("cmd") and cmd in self._data[key]["cmd"]:
+                return key
+        return None