mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2025-08-01 07:09:50 +00:00
✨ 支持开发热重载插件,支持独立测试函数
This commit is contained in:
@ -27,7 +27,7 @@ class Caller:
|
||||
self.func: ASYNC_FUNCTION_CALL_FUNC | None = None
|
||||
"""函数对象"""
|
||||
self.module_name: str = ""
|
||||
"""模块名"""
|
||||
"""模块名,仅为父级模块名,不一定是插件顶级模块名"""
|
||||
self._parameters: dict[str, Any] = {}
|
||||
"""声明参数"""
|
||||
|
||||
@ -137,14 +137,14 @@ class Caller:
|
||||
self.func = async_wrap(func) # type: ignore
|
||||
|
||||
if module := inspect.getmodule(func):
|
||||
module_name = module.__name__.split(".")[-1] + "."
|
||||
module_name = module.__name__.split(".")[-1]
|
||||
else:
|
||||
module_name = ""
|
||||
|
||||
self.module_name = module_name
|
||||
_caller_data[self.full_name] = self
|
||||
logger.opt(colors=True).debug(
|
||||
f"<y>加载函数 {module_name}{func.__name__}: {self._description}</y>"
|
||||
f"<y>加载函数 {self.full_name}: {self._description}</y>"
|
||||
)
|
||||
|
||||
return func
|
||||
@ -238,7 +238,7 @@ class Caller:
|
||||
@property
|
||||
def full_name(self) -> str:
|
||||
"""完整名"""
|
||||
return self.module_name + self._name
|
||||
return self.module_name + "." + self._name
|
||||
|
||||
@property
|
||||
def short_info(self) -> str:
|
||||
|
@ -45,7 +45,9 @@ def get_plugins() -> dict[str, Plugin]:
|
||||
return _plugins
|
||||
|
||||
|
||||
def load_plugin(module_path: str | Path) -> Optional[Plugin]:
|
||||
def load_plugin(
|
||||
module_path: str | Path, allow_reload: bool = False
|
||||
) -> Optional[Plugin]:
|
||||
"""加载单个插件,可以是本地插件或是通过 `pip` 安装的插件。
|
||||
该函数产生的副作用在于将插件加载到 `_plugins` 中。
|
||||
|
||||
@ -63,12 +65,15 @@ def load_plugin(module_path: str | Path) -> Optional[Plugin]:
|
||||
try:
|
||||
module = import_module(module_path) # 导入模块对象
|
||||
plugin = Plugin(
|
||||
name=module.__name__,
|
||||
name=module.__name__.split(".")[-1],
|
||||
module=module,
|
||||
module_name=module_path,
|
||||
module_path=module.__file__,
|
||||
)
|
||||
_plugins[plugin.name] = plugin
|
||||
if plugin.name in _plugins and not allow_reload:
|
||||
raise ValueError(f"插件名称重复: {plugin.name}")
|
||||
else:
|
||||
_plugins[plugin.name] = plugin
|
||||
|
||||
plugin.metadata = getattr(module, "__marsho_meta__", None)
|
||||
|
||||
@ -118,3 +123,33 @@ def load_plugins(*plugin_dirs: str) -> set[Plugin]:
|
||||
if module_name and (plugin := load_plugin(module_name)):
|
||||
plugins.add(plugin)
|
||||
return plugins
|
||||
|
||||
|
||||
def reload_plugin(plugin: Plugin) -> Optional[Plugin]:
|
||||
"""开发模式下的重新加载插件
|
||||
该方法无法保证没有副作用,因为插件可能会有自己的初始化方法
|
||||
如果出现异常请重启即可
|
||||
Args:
|
||||
plugin: 插件对象
|
||||
Returns:
|
||||
Optional[Plugin]: 插件对象
|
||||
"""
|
||||
try:
|
||||
if plugin.module_path:
|
||||
if new_plugin := load_plugin(plugin.module_name, True):
|
||||
logger.opt(colors=True).debug(
|
||||
f'重新加载插件 "<y>{new_plugin.name}</y>" 成功, 若出现异常或副作用请重启'
|
||||
)
|
||||
return new_plugin
|
||||
else:
|
||||
logger.opt(colors=True).error(
|
||||
f'重新加载插件失败 "<r>{plugin.name}</r>"'
|
||||
)
|
||||
return None
|
||||
else:
|
||||
logger.opt(colors=True).error(f'插件不支持重载 "<r>{plugin.name}</r>"')
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.opt(colors=True).error(f'重新加载插件失败 "<r>{plugin.name}</r>"')
|
||||
traceback.print_exc()
|
||||
return None
|
||||
|
@ -57,7 +57,7 @@ class Plugin(BaseModel):
|
||||
module: ModuleType
|
||||
"""插件模块对象"""
|
||||
module_name: str
|
||||
"""点分割模块路径 例如a.b.c"""
|
||||
"""点分或/割模块路径 例如a.b.c"""
|
||||
module_path: str | None
|
||||
"""实际路径,单文件为.py的路径,包为__init__.py路径"""
|
||||
metadata: PluginMetadata | None = None
|
||||
|
Reference in New Issue
Block a user