🐛 修复潜在的 NoneBot 插件列表读取错误

nonebot · Oct 25, 2024 · 454a8d2 · 454a8d2
1 parent 23bf0c9
commit 454a8d2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/nb_cli_plugin_webui/app/handlers/plugin.py b/nb_cli_plugin_webui/app/handlers/plugin.py
@@ -1,3 +1,4 @@
+import re
 import json
 from pathlib import Path
 from typing import List, Optional
@@ -21,7 +22,13 @@ async def get_nonebot_plugin_list(
     raw_content = await run_python_script(
         python_path, await t.render_async(toml_path=config_file), cwd
     )
-    return raw_content.split(",")
+
+    matches = re.findall(r"nonebot_plugins:\[(.*?)\]", raw_content)
+    result = list()
+    if matches:
+        result = matches[0].split(",")
+
+    return result
 
 
 async def get_nonebot_plugin_config_detail(

diff --git a/nb_cli_plugin_webui/app/template/scripts/plugin/get_loaded_plugins.py.jinja b/nb_cli_plugin_webui/app/template/scripts/plugin/get_loaded_plugins.py.jinja
@@ -13,4 +13,4 @@ def get_loaded_plugins(toml_path: str):
 
 plugins = get_loaded_plugins("{{ toml_path }}")
 if plugins:
-    print(",".join([plugin.module_name for plugin in plugins]))
+    print(f"nonebot_plugins:[{','.join([plugin.module_name for plugin in plugins])}]")