This commit is contained in:
jxxghp
2023-11-01 22:05:48 +08:00
parent fbe306ba90
commit cd4229a915
3 changed files with 25 additions and 40 deletions

View File

@ -15,25 +15,23 @@ router = APIRouter()
@router.get("/", summary="所有插件", response_model=List[schemas.Plugin])
def all_plugins(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
查询所有插件清单
查询所有插件清单,包括本地插件和在线插件
"""
# 查询本地插件
plugins = []
# 本地插件
local_plugins = PluginManager().get_local_plugins()
# 在线插件
online_plugins = PluginManager().get_online_plugins()
# 全并去重,在线插件有的以在线插件为准
plugins = []
if not local_plugins:
return online_plugins
for plugin in local_plugins:
for online_plugin in online_plugins:
if plugin["id"] == online_plugin["id"]:
plugins.append(online_plugin)
break
else:
plugins.append(plugin)
# 已安装插件IDS
installed_ids = SystemConfigOper().get(SystemConfigKey.UserInstalledPlugins) or []
# 已经安装的本地
plugins.extend([plugin for plugin in local_plugins if plugin["id"] in installed_ids])
# 未安装的线上插件或者有更新的插件
for plugin in online_plugins:
if plugin not in plugins:
if plugin["id"] not in installed_ids:
plugins.append(plugin)
elif plugin.get("has_update"):
plugin["installed"] = False
plugins.append(plugin)
return plugins
@ -56,27 +54,20 @@ def install_plugin(plugin_id: str,
"""
# 已安装插件
install_plugins = SystemConfigOper().get(SystemConfigKey.UserInstalledPlugins) or []
# 重载标志
reload_flag = False
# 如果是非本地括件,或者强制安装时,则需要下载安装
if repo_url and (force or plugin_id not in PluginManager().get_plugin_ids()):
# 下载安装
state, msg = PluginHelper().install(pid=plugin_id, repo_url=repo_url)
if state:
# 安装成功
reload_flag = True
else:
if not state:
# 安装失败
return schemas.Response(success=False, msg=msg)
# 安装插件
if plugin_id not in install_plugins:
reload_flag = True
install_plugins.append(plugin_id)
# 保存设置
SystemConfigOper().set(SystemConfigKey.UserInstalledPlugins, install_plugins)
# 重载插件管理器
if reload_flag:
PluginManager().init_config()
PluginManager().init_config()
return schemas.Response(success=True)