85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
# =======================================
|
||
# =============== 加载插件 ===============
|
||
# =======================================
|
||
|
||
import os
|
||
import site
|
||
import importlib
|
||
|
||
from umi_log import logger
|
||
from ..ocr.api import initOcrPlugins
|
||
|
||
# 插件 总目录
|
||
PLUGINS_PATH = "plugins"
|
||
# 插件组 组名
|
||
PLUGINS_GROUPS = ["ocr"]
|
||
|
||
|
||
# 插件控制器
|
||
class _PluginsControllerClass:
|
||
def __init__(self):
|
||
self.pluginsDict = {}
|
||
self.optionsDict = {}
|
||
for group in PLUGINS_GROUPS:
|
||
self.pluginsDict[group] = {}
|
||
self.optionsDict[group] = {}
|
||
|
||
# 初始化并加载插件
|
||
def init(self):
|
||
# 动态插件
|
||
errors = {} # 保存失败信息
|
||
# 添加包搜索路径
|
||
if not os.path.exists(PLUGINS_PATH):
|
||
os.makedirs(PLUGINS_PATH)
|
||
logger.error(f"插件目录不存在: {PLUGINS_PATH} ")
|
||
return None
|
||
site.addsitedir(PLUGINS_PATH) # 添加插件搜索路径
|
||
# 加载所有插件。在插件目录中搜索合法的包
|
||
plugList = os.listdir(PLUGINS_PATH)
|
||
for name in plugList:
|
||
initPath = os.path.join(PLUGINS_PATH, name, "__init__.py")
|
||
if os.path.exists(initPath): # 若包路径下存在 __init__.py ，则尝试加载该包
|
||
flag, res = self._loadPlugin(name)
|
||
if not flag: # 失败
|
||
errors[name] = res
|
||
logger.error(f"加载插件 {name} 失败：{res}")
|
||
# TODO: 静态插件
|
||
# 动态插件导入API管理器
|
||
ocrErrs = initOcrPlugins(self.pluginsDict["ocr"])
|
||
if ocrErrs:
|
||
errors.update(ocrErrs)
|
||
return {"options": self.optionsDict, "errors": errors}
|
||
|
||
# 加载一个组件python包。成功返回True，失败返回 False, 错误信息
|
||
def _loadPlugin(self, name):
|
||
# 加载包
|
||
try:
|
||
module = importlib.import_module(name)
|
||
except Exception as e:
|
||
return False, f"动态导入包失败：{e}"
|
||
# 验证信息
|
||
if not hasattr(module, "PluginInfo"):
|
||
return False, f"__init__.py 中未定义 PluginInfo 。"
|
||
pluginInfo = module.PluginInfo
|
||
if "group" not in pluginInfo:
|
||
return False, f"__init__.py 中未定义 group 。"
|
||
if "api_class" not in pluginInfo:
|
||
return False, f"__init__.py 中未定义 api_class 。"
|
||
if "global_options" not in pluginInfo:
|
||
pluginInfo["global_options"] = None
|
||
if "local_options" not in pluginInfo:
|
||
pluginInfo["local_options"] = None
|
||
group = pluginInfo["group"]
|
||
if not group or group not in PLUGINS_GROUPS:
|
||
return False, f'__init__.py group "{group}" 不属于已定义的插件类型。'
|
||
# 加载成功、验证成功，则记录信息
|
||
self.pluginsDict[group][name] = pluginInfo
|
||
self.optionsDict[group][name] = {
|
||
"global_options": pluginInfo["global_options"],
|
||
"local_options": pluginInfo["local_options"],
|
||
}
|
||
return True, ""
|
||
|
||
|
||
PluginsController = _PluginsControllerClass()
|