🐛 fix module name load

nonebot · Feb 28, 2024 · 1c9e9f9 · 1c9e9f9
1 parent 58b4ae7
commit 1c9e9f9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/nb_cli/config/model.py b/nb_cli/config/model.py
@@ -9,28 +9,25 @@ class SimpleInfo(BaseModel):
 
 
 class Adapter(SimpleInfo):
+    __module_name__ = "adapters"
+
     project_link: str
     desc: str
 
-    class Config:
-        module_name = "adapters"
-
 
 class Plugin(SimpleInfo):
+    __module_name__ = "plugins"
+
     project_link: str
     desc: str
 
-    class Config:
-        module_name = "plugins"
-
 
 class Driver(SimpleInfo):
+    __module_name__ = "drivers"
+
     project_link: str
     desc: str
 
-    class Config:
-        module_name = "drivers"
-
 
 class NoneBotConfig(BaseModel):
 

diff --git a/nb_cli/handlers/store.py b/nb_cli/handlers/store.py
@@ -44,7 +44,7 @@ async def load_module_data(
             raise ValueError(
                 _("Invalid module type: {module_type}").format(module_type=module_type)
             )
-        module_name: str = getattr(module_class.__config__, "module_name")
+        module_name: str = module_class.__module_name__
 
         exceptions: list[Exception] = []
         urls = [