优化代码

2026-09-29 08:39:59 +08:00 · 2022-12-04 16:46:33 +08:00
parent 994484f3b0
commit 31e4cadecb
18 changed files with 483 additions and 325 deletions
@@ -0,0 +1,121 @@
+from types import ModuleType
+from typing import Any
+
+from services import logger
+from utils.manager import (
+    plugin_data_manager,
+    plugins2settings_manager,
+    plugins2cd_manager,
+    plugins2block_manager,
+    plugins2count_manager,
+)
+from configs.config import Config
+from utils.manager.models import (
+    PluginType,
+    PluginSetting,
+    PluginCd,
+    PluginData,
+    PluginBlock,
+    PluginCount,
+)
+from utils.utils import get_matchers
+
+
+def get_attr(module: ModuleType, name: str, default: Any = None) -> Any:
+    """
+    说明:
+        获取属性
+    参数:
+        :param module: module
+        :param name: name
+        :param default: default
+    """
+    if hasattr(module, name):
+        return getattr(module, name)
+    return default
+
+
+def init_plugin_info():
+
+    for matcher in [x for x in get_matchers(True) if x.plugin]:
+        try:
+            plugin = matcher.plugin
+            metadata = plugin.metadata
+            if hasattr(plugin, "module"):
+                module = plugin.module
+                plugin_model = matcher.plugin_name
+                plugin_name = (
+                    metadata.name
+                    if metadata and metadata.name
+                    else get_attr(module, "__zx_plugin_name__", matcher.plugin_name)
+                )
+                if "[Admin]" in plugin_name:
+                    plugin_type = PluginType.ADMIN
+                    plugin_name = plugin_name.replace("[Admin]", "")
+                elif "[Hidden]" in plugin_name:
+                    plugin_type = PluginType.HIDDEN
+                    plugin_name = plugin_name.replace("[Hidden]", "")
+                elif "[Superuser]" in plugin_name:
+                    plugin_type = PluginType.SUPERUSER
+                    plugin_name = plugin_name.replace("[Superuser]", "")
+                else:
+                    plugin_type = PluginType.NORMAL
+                plugin_usage = (
+                    metadata.usage
+                    if metadata and metadata.usage
+                    else get_attr(module, "__plugin_usage__")
+                )
+                plugin_des = (
+                    metadata.description
+                    if metadata and metadata.description
+                    else get_attr(module, "__plugin_des__")
+                )
+                menu_type = get_attr(module, "__plugin_type__") or ("normal",)
+                plugin_setting = get_attr(module, "__plugin_settings__")
+                if plugin_setting:
+                    plugin_setting = PluginSetting(**plugin_setting)
+                    plugin_setting.plugin_type = menu_type
+                plugin_task = get_attr(module, "__plugin_task__")
+                plugin_version = get_attr(module, "__plugin_version__")
+                plugin_author = get_attr(module, "__plugin_author__")
+                plugin_cd = get_attr(module, "__plugin_cd_limit__")
+                if plugin_cd:
+                    plugin_cd = PluginCd(**plugin_cd)
+                plugin_block = get_attr(module, "__plugin_block_limit__")
+                if plugin_block:
+                    plugin_block = PluginBlock(**plugin_block)
+                plugin_count = get_attr(module, "__plugin_count_limit__")
+                if plugin_count:
+                    plugin_count = PluginCount(**plugin_count)
+                plugin_resources = get_attr(module, "__plugin_resources__")
+                plugin_configs = get_attr(module, "__plugin_configs__")
+                if settings := plugins2settings_manager.get(plugin_model):
+                    plugin_setting = settings
+                if plugin_cd_limit := plugins2cd_manager.get(plugin_model):
+                    plugin_cd = plugin_cd_limit
+                if plugin_block_limit := plugins2block_manager.get(plugin_model):
+                    plugin_block = plugin_block_limit
+                if plugin_count_limit := plugins2count_manager.get(plugin_model):
+                    plugin_count = plugin_count_limit
+                if plugin_cfg := Config.get(plugin_model):
+                    plugin_configs = plugin_cfg
+                plugin_data = PluginData(
+                    model=plugin_model,
+                    name=plugin_name.strip(),
+                    plugin_type=plugin_type,
+                    usage=plugin_usage,
+                    des=plugin_des,
+                    task=plugin_task,
+                    menu_type=menu_type,
+                    version=plugin_version,
+                    author=plugin_author,
+                    plugin_setting=plugin_setting,
+                    plugin_cd=plugin_cd,
+                    plugin_block=plugin_block,
+                    plugin_count=plugin_count,
+                    plugin_resources=plugin_resources,
+                    plugin_configs=plugin_configs,
+                )
+                plugin_data_manager.add_plugin_info(plugin_data)
+        except Exception as e:
+            logger.error(f"构造插件数据失败 {matcher.plugin_name} - {type(e)}：{e}")