From 593388e1807b81456cfdba3a1bb197038bd2182f Mon Sep 17 00:00:00 2001 From: thsrite Date: Wed, 15 May 2024 09:37:37 +0800 Subject: [PATCH] fix --- plugins/pluginautoupdate/__init__.py | 7 ++++++- plugins/pluginreinstall/__init__.py | 8 ++++++-- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/plugins/pluginautoupdate/__init__.py b/plugins/pluginautoupdate/__init__.py index 533ff95..793466f 100644 --- a/plugins/pluginautoupdate/__init__.py +++ b/plugins/pluginautoupdate/__init__.py @@ -267,7 +267,12 @@ class PluginAutoUpdate(_PluginBase): """ 注册插件API（先删除后新增） """ - for api in PluginManager().get_plugin_apis(plugin_id): + apis: List[Dict[str, Any]] = [] + for api in PluginManager().get_plugin_apis(): + if plugin_id in api.get("path"): + apis.append(api) + + for api in apis: for r in router.routes: if r.path == api.get("path"): router.routes.remove(r) diff --git a/plugins/pluginreinstall/__init__.py b/plugins/pluginreinstall/__init__.py index b011d92..d342ad7 100644 --- a/plugins/pluginreinstall/__init__.py +++ b/plugins/pluginreinstall/__init__.py @@ -225,13 +225,17 @@ class PluginReInstall(_PluginBase): """ 注册插件API（先删除后新增） """ - for api in PluginManager().get_plugin_apis(plugin_id): + apis: List[Dict[str, Any]] = [] + for api in PluginManager().get_plugin_apis(): + if plugin_id in api.get("path"): + apis.append(api) + + for api in apis: for r in router.routes: if r.path == api.get("path"): router.routes.remove(r) break router.add_api_route(**api) - def get_state(self) -> bool: return False