liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

模块管理优化

Browse Source
This commit is contained in:
liuwei
2025-03-20 10:47:54 +08:00
parent 05a6ee672c
commit d27d8c1a39
2 changed files with 257 additions and 367 deletions
Download Patch File Download Diff File Expand all files Collapse all files

173
plugin_common/plugin_manager.py
View File

@@ -3,7 +3,7 @@ import inspect
import logging
import os
import sys
from typing import Dict, List, Any, Optional, Type
from typing import Dict, List, Any, Optional, Type, Tuple
from plugin_common.plugin_interface import PluginInterface, PluginStatus
from plugin_common.message_plugin_interface import MessagePluginInterface
@@ -25,6 +25,7 @@ class PluginManager:
self.plugin_dir = plugin_dir
self.plugins: Dict[str, PluginInterface] = {} # 插件实例字典
self.plugin_modules = {} # 插件模块字典
self.module_to_plugin = {} # 模块名到插件名的映射
self.system_context = {} # 系统上下文
self.LOG = logging.getLogger(__name__)
@@ -94,12 +95,19 @@ class PluginManager:
加载插件
Args:
plugin_name: 插件名称
plugin_name: 插件名称（模块名）
Returns:
插件实例，加载失败返回None
"""
try:
# 检查是否已有同名模块的插件加载
for name, plugin in self.plugins.items():
module_name = plugin.__class__.__module__.split('.')[-2]
if module_name == plugin_name:
self.LOG.info(f"PluginManager：插件模块 {plugin_name} 已加载为 {name}")
return plugin
# 如果插件已加载，直接返回
if plugin_name in self.plugins:
return self.plugins[plugin_name]
@@ -193,9 +201,17 @@ class PluginManager:
# 注册插件
PluginRegistry().register(plugin)
# 在 load_plugin 方法中，修改存储插件实例的部分
# 存储插件实例
self.plugins[plugin.name] = plugin
# 添加模块名到插件名的映射
try:
module_name = plugin.__class__.__module__.split('.')[-2]
self.module_to_plugin[module_name] = plugin.name
except (IndexError, AttributeError):
self.LOG.warning(f"无法为插件 {plugin.name} 获取有效的模块名")
# 发布插件加载事件
EventSystem().publish(EventType.PLUGIN_LOADED, {"plugin": plugin})
@@ -210,40 +226,50 @@ class PluginManager:
卸载插件
Args:
plugin_name: 插件名称
plugin_name: 插件名称（可以是模块名或显示名称）
Returns:
卸载是否成功
"""
if plugin_name not in self.plugins:
# 查找插件
display_name, plugin = self.find_plugin_by_name(plugin_name)
if not plugin:
self.LOG.info(f"PluginManager：插件 {plugin_name} 未加载")
return False
plugin = self.plugins[plugin_name]
# 停止插件
if plugin.status == PluginStatus.RUNNING:
if not plugin.stop():
self.LOG.info(f"PluginManager：停止插件 {plugin_name} 失败")
self.LOG.info(f"PluginManager：停止插件 {display_name} 失败")
return False
plugin.status = PluginStatus.STOPPED # 确保状态更新
# 清理插件资源
if not plugin.cleanup():
self.LOG.info(f"PluginManager：清理插件 {plugin_name} 资源失败")
self.LOG.info(f"PluginManager：清理插件 {display_name} 资源失败")
return False
# 设置状态为未加载
plugin.status = PluginStatus.UNLOADED
# 注销插件
PluginRegistry().unregister(plugin_name)
PluginRegistry().unregister(display_name)
# 获取模块名，用于清理映射
try:
module_name = plugin.__class__.__module__.split('.')[-2]
# 清理模块名到插件名的映射
if module_name in self.module_to_plugin:
del self.module_to_plugin[module_name]
except (IndexError, AttributeError):
pass
# 移除插件实例
del self.plugins[plugin_name]
del self.plugins[display_name]
# 发布插件卸载事件
EventSystem().publish(EventType.PLUGIN_UNLOADED, {"plugin_name": plugin_name})
EventSystem().publish(EventType.PLUGIN_UNLOADED, {"plugin_name": display_name})
return True
@@ -252,31 +278,41 @@ class PluginManager:
重新加载插件
Args:
plugin_name: 插件名称
plugin_name: 插件名称（可以是模块名或显示名称）
Returns:
插件实例，重新加载失败返回None
"""
# 记录原插件状态
was_running = False
if plugin_name in self.plugins:
was_running = self.plugins[plugin_name].status == PluginStatus.RUNNING
# 查找插件
display_name, plugin = self.find_plugin_by_name(plugin_name)
if not plugin:
self.LOG.info(f"PluginManager：插件 {plugin_name} 未加载，无法重载")
return None
# 卸载插件
if not self.unload_plugin(plugin_name):
self.LOG.info(f"卸载插件 {plugin_name} 失败")
return None
# 记录原插件状态和模块名
was_running = plugin.status == PluginStatus.RUNNING
try:
module_name = plugin.__class__.__module__.split('.')[-2]
except (IndexError, AttributeError):
self.LOG.error(f"无法获取插件 {display_name} 的模块名，重载失败")
return None
# 卸载插件
if not self.unload_plugin(display_name):
self.LOG.info(f"卸载插件 {display_name} 失败，无法重载")
return None
# 重新导入模块
if plugin_name in self.plugin_modules:
if module_name in self.plugin_modules:
try:
importlib.reload(self.plugin_modules[plugin_name])
importlib.reload(self.plugin_modules[module_name])
except Exception as e:
self.LOG.info(f"重新导入插件模块 {plugin_name} 失败: {e}")
self.LOG.info(f"重新导入插件模块 {module_name} 失败: {e}")
return None
# 加载插件
plugin = self.load_plugin(plugin_name)
plugin = self.load_plugin(module_name)
# 如果原来是运行状态，则重新启动
if plugin and was_running:
@@ -289,28 +325,29 @@ class PluginManager:
启动插件
Args:
plugin_name: 插件名称
plugin_name: 插件名称（可以是模块名或显示名称）
Returns:
启动是否成功
"""
if plugin_name not in self.plugins:
# 查找插件
display_name, plugin = self.find_plugin_by_name(plugin_name)
if not plugin:
self.LOG.info(f"PluginManager：插件 {plugin_name} 未加载")
return False
plugin = self.plugins[plugin_name]
if plugin.status == PluginStatus.RUNNING:
self.LOG.info(f"PluginManager：插件 {plugin_name} 已经在运行")
self.LOG.info(f"PluginManager：插件 {display_name} 已经在运行")
return True
if plugin.start():
plugin.status = PluginStatus.RUNNING
self.LOG.info(f"PluginManager：插件 {plugin_name} 状态变更为在运行")
self.LOG.info(f"PluginManager：插件 {display_name} 状态变更为在运行")
return True
else:
plugin.status = PluginStatus.ERROR
self.LOG.info(f"PluginManager：插件 {plugin_name} 状态变更为异常")
self.LOG.info(f"PluginManager：插件 {display_name} 状态变更为异常")
return False
def stop_plugin(self, plugin_name: str) -> bool:
@@ -318,57 +355,31 @@ class PluginManager:
停止插件
Args:
plugin_name: 插件名称
plugin_name: 插件名称（可以是模块名或显示名称）
Returns:
停止是否成功
"""
if plugin_name not in self.plugins:
# 查找插件
display_name, plugin = self.find_plugin_by_name(plugin_name)
if not plugin:
self.LOG.info(f"插件 {plugin_name} 未加载")
return False
plugin = self.plugins[plugin_name]
if plugin.status != PluginStatus.RUNNING:
self.LOG.info(f"插件 {plugin_name} 未在运行")
self.LOG.info(f"插件 {display_name} 未在运行")
return True
if plugin.stop():
plugin.status = PluginStatus.STOPPED
self.LOG.info(f"插件 {plugin_name} 状态变更为已停止")
self.LOG.info(f"插件 {display_name} 状态变更为已停止")
return True
else:
plugin.status = PluginStatus.ERROR
self.LOG.info(f"插件 {plugin_name} 状态变更为异常")
self.LOG.info(f"插件 {display_name} 状态变更为异常")
return False
def reload_plugin(self, plugin_name: str) -> Optional[PluginInterface]:
"""
重新加载插件
Args:
plugin_name: 插件名称
Returns:
插件实例，重新加载失败返回None
"""
# 卸载插件
if plugin_name in self.plugins:
if not self.unload_plugin(plugin_name):
self.LOG.info(f"卸载插件 {plugin_name} 失败")
return None
# 重新导入模块
if plugin_name in self.plugin_modules:
try:
importlib.reload(self.plugin_modules[plugin_name])
except Exception as e:
self.LOG.info(f"重新导入插件模块 {plugin_name} 失败: {e}")
return None
# 加载插件
return self.load_plugin(plugin_name)
def shutdown_plugins(self) -> bool:
"""
卸载所有插件
@@ -394,3 +405,33 @@ class PluginManager:
success = False
return success
def find_plugin_by_name(self, plugin_name: str) -> Tuple[Optional[str], Optional[PluginInterface]]:
"""
根据插件名称或模块名查找插件
Args:
plugin_name: 插件名称或模块名
Returns:
(插件显示名称, 插件实例) 元组，未找到返回 (None, None)
"""
# 直接通过显示名称查找
if plugin_name in self.plugins:
return plugin_name, self.plugins[plugin_name]
# 通过模块名查找
if plugin_name in self.module_to_plugin:
display_name = self.module_to_plugin[plugin_name]
return display_name, self.plugins.get(display_name)
# 遍历所有插件查找匹配的模块名
for name, plugin in self.plugins.items():
try:
module_name = plugin.__class__.__module__.split('.')[-2]
if module_name == plugin_name:
return name, plugin
except (IndexError, AttributeError):
continue
return None, None