refactor: let __init__ import export only

schemas/config.json

@@ -39,11 +39,6 @@
     },
     "Telegram": {
       "properties": {
-        "enabled": {
-          "default": true,
-          "title": "Enabled",
-          "type": "boolean"
-        },
         "bot_token": {
           "title": "Bot Token",
           "type": "string"
@@ -62,6 +57,11 @@
     },
     "TelegramNotify": {
       "properties": {
+        "enabled": {
+          "default": true,
+          "title": "Enabled",
+          "type": "boolean"
+        },
         "telegram": {
           "$ref": "#/$defs/Telegram"
         }

src/check_phat_nguoi/models/config/__init__.py

@@ -1,18 +1,5 @@
-from pydantic import BaseModel, ConfigDict
-
+from check_phat_nguoi.models.config.config import Config
 from check_phat_nguoi.models.config.log_level import LogLevel
-from check_phat_nguoi.models.config.notify.telegram_notify import TelegramNotify
 from check_phat_nguoi.models.config.plate_info import PlateInfo
 
-
-class Config(BaseModel):
-    model_config = ConfigDict(use_enum_values=True, validate_default=True)
-
-    data: list[PlateInfo] = []
-    notify: list[TelegramNotify] = []
-    havent_paid_only: bool = True
-    detail_log: bool = False
-    log_level: LogLevel = LogLevel.warning
-
-
-__all__ = ["Config"]
+__all__ = ["PlateInfo", "Config", "LogLevel"]

src/check_phat_nguoi/models/config/config.py

@@ -0,0 +1,18 @@
+from pydantic import BaseModel, ConfigDict
+
+from check_phat_nguoi.models.config.log_level import LogLevel
+from check_phat_nguoi.models.config.notify.telegram_notify import TelegramNotify
+from check_phat_nguoi.models.config.plate_info import PlateInfo
+
+
+class Config(BaseModel):
+    model_config = ConfigDict(use_enum_values=True, validate_default=True)
+
+    data: list[PlateInfo] = []
+    notify: list[TelegramNotify] = []
+    havent_paid_only: bool = True
+    detail_log: bool = False
+    log_level: LogLevel = LogLevel.warning
+
+
+__all__ = ["Config"]

src/check_phat_nguoi/models/config/notify/__init__.py

@@ -0,0 +1,3 @@
+from check_phat_nguoi.models.config.notify.telegram_notify import TelegramNotify
+
+__all__ = ["TelegramNotify"]

src/check_phat_nguoi/models/config/notify/base_notify.py

@@ -0,0 +1,8 @@
+from pydantic import BaseModel
+
+
+class BaseNotify(BaseModel):
+    enabled: bool = True
+
+
+__all__ = ["BaseNotify"]

src/check_phat_nguoi/models/config/notify/telegram.py

@@ -4,7 +4,6 @@
 
 
 class Telegram(BaseModel):
-    enabled: bool = True
     bot_token: str
     chat_id: str
 

src/check_phat_nguoi/models/config/notify/telegram_notify.py

@@ -1,9 +1,8 @@
-from pydantic import BaseModel
-
+from check_phat_nguoi.models.config.notify.base_notify import BaseNotify
 from check_phat_nguoi.models.config.notify.telegram import Telegram
 
 
-class TelegramNotify(BaseModel):
+class TelegramNotify(BaseNotify):
     telegram: Telegram
 
 

src/check_phat_nguoi/models/context/__init__.py

@@ -1,10 +1,3 @@
-from pydantic import BaseModel
-
-from check_phat_nguoi.models.context.plate_info import PlateInfo
-
-
-class Context(BaseModel):
-    data: list[PlateInfo] = []
-
+from check_phat_nguoi.models.context.context import Context
 
 __all__ = ["Context"]

src/check_phat_nguoi/models/context/context.py

@@ -0,0 +1,10 @@
+from pydantic import BaseModel
+
+from check_phat_nguoi.models.context.plate_info import PlateInfo
+
+
+class Context(BaseModel):
+    data: list[PlateInfo] = []
+
+
+__all__ = ["Context"]