fix modules

app/core/module.py

@@ -0,0 +1,83 @@
+from types import FunctionType
+from typing import Generator, Optional
+
+from app.core.config import settings
+from app.helper.module import ModuleHelper
+from app.log import logger
+from app.utils.singleton import Singleton
+
+
+class ModuleManager(metaclass=Singleton):
+    """
+    模块管理器
+    """
+
+    # 模块列表
+    _modules: dict = {}
+    # 运行态模块列表
+    _running_modules: dict = {}
+
+    def __init__(self):
+        self.load_modules()
+
+    def load_modules(self):
+        """
+        加载所有模块
+        """
+        # 扫描模块目录
+        modules = ModuleHelper.load(
+            "app.modules",
+            filter_func=lambda _, obj: hasattr(obj, 'init_module') and hasattr(obj, 'init_setting')
+        )
+        self._running_modules = {}
+        self._modules = {}
+        for module in modules:
+            module_id = module.__name__
+            self._modules[module_id] = module
+            # 生成实例
+            _module = module()
+            # 初始化模块
+            if self.check_setting(_module.init_setting()):
+                _module.init_module()
+                self._running_modules[module_id] = _module
+                logger.info(f"Moudle Loaded：{module_id}")
+
+    def stop(self):
+        """
+        停止所有模块
+        """
+        for _, module in self._running_modules.items():
+            if hasattr(module, "stop"):
+                module.stop()
+
+    @staticmethod
+    def check_setting(setting: Optional[tuple]) -> bool:
+        """
+        检查开关是否己打开
+        """
+        if not setting:
+            return True
+        switch, value = setting
+        if getattr(settings, switch) and value is True:
+            return True
+        if getattr(settings, switch) == value:
+            return True
+        return False
+
+    def get_modules(self, method: str) -> Generator:
+        """
+        获取模块列表
+        """
+
+        def check_method(func: FunctionType) -> bool:
+            """
+            检查函数是否已实现
+            """
+            return func.__code__.co_code != b'd\x01S\x00'
+
+        if not self._running_modules:
+            return []
+        for _, module in self._running_modules.items():
+            if hasattr(module, method) \
+                    and check_method(getattr(module, method)):
+                yield module