diff --git a/scripts/test_plugin_load.py b/scripts/test_plugin_load.py index a5bd7db..810aa8e 100644 --- a/scripts/test_plugin_load.py +++ b/scripts/test_plugin_load.py @@ -8,9 +8,12 @@ nonebot.load_plugins("konabot/plugins") plugins = nonebot.get_loaded_plugins() len_requires = len( - [f for f in ( - Path(__file__).parent.parent / "konabot" / "plugins" - ).iterdir() if f.is_dir() and (f / "__init__.py").exists()] + [ + f + for f in (Path(__file__).parent.parent / "konabot" / "plugins").iterdir() + if (f.is_dir() and (f / "__init__.py").exists()) + or ((not f.is_dir()) and f.suffix == ".py") + ] ) plugins = [p for p in plugins if p.module.__name__.startswith("konabot.plugins")]