⚡ 添加进程及生命周期管理器，添加轻雪框架支持

liteyuki/plugin/__init__.py

@@ -0,0 +1,17 @@
+from liteyuki.plugin.model import Plugin, PluginMetadata
+from liteyuki.plugin.load import load_plugin, _plugins
+
+__all__ = [
+        "PluginMetadata",
+        "Plugin",
+        "load_plugin",
+]
+
+
+def get_loaded_plugins() -> dict[str, Plugin]:
+    """
+    获取已加载的插件
+    Returns:
+        dict[str, Plugin]: 插件字典
+    """
+    return _plugins

liteyuki/plugin/load.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+"""
+Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved 
+
+@Time    : 2024/7/23 下午11:59
+@Author  : snowykami
+@Email   : snowykami@outlook.com
+@File    : load.py
+@Software: PyCharm
+"""
+import os
+import traceback
+from pathlib import Path
+from typing import Optional
+
+from nonebot import logger
+
+from liteyuki.plugin.model import Plugin, PluginMetadata
+from importlib import import_module
+
+from liteyuki.utils import path_to_module_name
+
+_plugins: dict[str, Plugin] = {}
+
+
+def load_plugin(module_path: str | Path) -> Optional[Plugin]:
+    """加载单个插件，可以是本地插件或是通过 `pip` 安装的插件。
+
+    参数:
+        module_path: 插件名称 `path.to.your.plugin`
+        或插件路径 `pathlib.Path(path/to/your/plugin)`
+    """
+    module_path = path_to_module_name(Path(module_path)) if isinstance(module_path, Path) else module_path
+    try:
+        module = import_module(module_path)
+        _plugins[module.__name__] = Plugin(
+            name=module.__name__,
+            module=module,
+            module_name=module_path,
+            metadata=module.__dict__.get("__plugin_metadata__", None)
+        )
+        logger.opt(colors=True).success(
+            f'Succeeded to load liteyuki plugin "<y>{module.__name__.split(".")[-1]}</y>"'
+        )
+        return _plugins[module.__name__]
+
+    except Exception as e:
+        logger.opt(colors=True).success(
+            f'Failed to load liteyuki plugin "<r>{module_path}</r>"'
+        )
+        traceback.print_exc()
+        return None
+
+
+def load_plugins(*plugin_dir: str) -> set[Plugin]:
+    """导入文件夹下多个插件
+
+    参数:
+        plugin_dir: 文件夹路径
+    """
+    plugins = set()
+    for dir_path in plugin_dir:
+        # 遍历每一个文件夹下的py文件和包含__init__.py的文件夹，不递归
+        for f in os.listdir(dir_path):
+            path = Path(os.path.join(dir_path, f))
+
+            module_name = None
+            if os.path.isfile(path) and f.endswith('.py') and f != '__init__.py':
+                module_name = f"{path_to_module_name(Path(dir_path))}.{f[:-3]}"
+
+            elif os.path.isdir(path) and os.path.exists(os.path.join(path, '__init__.py')):
+                module_name = path_to_module_name(path)
+
+            if module_name:
+                load_plugin(module_name)
+                if _plugins.get(module_name):
+                    plugins.add(_plugins[module_name])
+    return plugins

liteyuki/plugin/manager.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved 
+
+@Time    : 2024/7/23 下午11:59
+@Author  : snowykami
+@Email   : snowykami@outlook.com
+@File    : manager.py
+@Software: PyCharm
+"""

liteyuki/plugin/model.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+"""
+Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved 
+
+@Time    : 2024/7/24 上午12:02
+@Author  : snowykami
+@Email   : snowykami@outlook.com
+@File    : model.py
+@Software: PyCharm
+"""
+from types import ModuleType
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class PluginMetadata(BaseModel):
+    """
+    轻雪插件元数据，由插件编写者提供
+    """
+    name: str
+    description: str
+    usage: str = ""
+    type: str = ""
+    homepage: str = ""
+    running_in_main: bool = True    # 是否在主进程运行
+
+
+class Plugin(BaseModel):
+    """
+    存储插件信息
+    """
+    model_config = {
+            'arbitrary_types_allowed': True
+    }
+    name: str
+    """插件名称 例如plugin_loader"""
+    module: ModuleType
+    """插件模块对象"""
+    module_name: str
+    """点分割模块路径 例如a.b.c"""
+    metadata: Optional[PluginMetadata] = None
+
+    def __hash__(self):
+        return hash(self.module_name)