Merge pull request #67 from EillesWan/main

__liteyuki_plugin_meta__支持
This commit is contained in:
Snowykami 2024-08-24 01:46:39 +08:00 committed by GitHub
commit 93c287bbd9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 37 additions and 10 deletions

View File

@ -21,9 +21,9 @@ from liteyuki.utils import path_to_module_name
_plugins: dict[str, Plugin] = {} _plugins: dict[str, Plugin] = {}
__all__ = [ __all__ = [
"load_plugin", "load_plugin",
"load_plugins", "load_plugins",
"_plugins", "_plugins",
] ]
@ -34,19 +34,43 @@ def load_plugin(module_path: str | Path) -> Optional[Plugin]:
module_path: 插件名称 `path.to.your.plugin` module_path: 插件名称 `path.to.your.plugin`
或插件路径 `pathlib.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 module_path = (
path_to_module_name(Path(module_path))
if isinstance(module_path, Path)
else module_path
)
try: try:
module = import_module(module_path) module = import_module(module_path)
_plugins[module.__name__] = Plugin( _plugins[module.__name__] = Plugin(
name=module.__name__, name=module.__name__,
module=module, module=module,
module_name=module_path, module_name=module_path,
metadata=module.__dict__.get("__plugin_metadata__", None)
) )
display_name = module.__name__.split(".")[-1] if module.__dict__.get("__plugin_metadata__", None):
if module.__dict__.get("__plugin_meta__"): metadata: "PluginMetadata" = module.__dict__["__plugin_metadata__"]
display_name = module.__name__.split(".")[-1]
elif module.__dict__.get("__liteyuki_plugin_meta__", None):
metadata: "PluginMetadata" = module.__dict__["__liteyuki_plugin_meta__"]
display_name = format_display_name(
f"{metadata.name}({module.__name__.split('.')[-1]})", metadata.type
)
elif module.__dict__.get("__plugin_meta__", None):
metadata: "PluginMetadata" = module.__dict__["__plugin_meta__"] metadata: "PluginMetadata" = module.__dict__["__plugin_meta__"]
display_name = format_display_name(f"{metadata.name}({module.__name__.split('.')[-1]})", metadata.type) display_name = format_display_name(
f"{metadata.name}({module.__name__.split('.')[-1]})", metadata.type
)
else:
logger.opt(colors=True).warning(
f'The metadata of Liteyuki plugin "{module.__name__}" is not specified, use empty.'
)
metadata = PluginMetadata(
name=module.__name__,
)
display_name = module.__name__.split(".")[-1]
_plugins[module.__name__].metadata = metadata
logger.opt(colors=True).success( logger.opt(colors=True).success(
f'Succeeded to load liteyuki plugin "{display_name}"' f'Succeeded to load liteyuki plugin "{display_name}"'
@ -90,10 +114,12 @@ def load_plugins(*plugin_dir: str, ignore_warning: bool = True) -> set[Plugin]:
path = Path(os.path.join(dir_path, f)) path = Path(os.path.join(dir_path, f))
module_name = None module_name = None
if os.path.isfile(path) and f.endswith('.py') and f != '__init__.py': 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]}" 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')): elif os.path.isdir(path) and os.path.exists(
os.path.join(path, "__init__.py")
):
module_name = path_to_module_name(path) module_name = path_to_module_name(path)
if module_name: if module_name:

View File

@ -9,6 +9,7 @@ nonebot-adapter-onebot~=2.4.3
nonebot-plugin-alconna~=0.46.3 nonebot-plugin-alconna~=0.46.3
nonebot_plugin_apscheduler~=0.4.0 nonebot_plugin_apscheduler~=0.4.0
nonebot-adapter-satori~=0.11.5 nonebot-adapter-satori~=0.11.5
liteyukibot-plugin-htmlrender~=0.1.0
pyppeteer~=2.0.0 pyppeteer~=2.0.0
markdown>=3.3.6 markdown>=3.3.6
zhDateTime>=1.0.3 zhDateTime>=1.0.3