diff --git a/nb_cli/config/model.py b/nb_cli/config/model.py index cdd075f..4b31c41 100644 --- 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 index 98cb3a0..3223996 100644 --- 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 = [