fix: use apis list, not allow single api

check-phat-nguoi · Jan 16, 2025 · 0115611 · 0115611
1 parent 030a4df
commit 0115611
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 21 deletions.
diff --git a/config.sample.json b/config.sample.json
@@ -18,21 +18,21 @@
     {
       "plate": "60A64685",
       "owner": "dad",
-      "api": "checkphatnguoi.vn",
+      "apis": "checkphatnguoi.vn",
       "type": 1
     },
     {
       "plate": "98A56604",
       "type": "car",
-      "api": ["csgt.vn", "checkphatnguoi.vn"]
+      "apis": ["csgt.vn", "checkphatnguoi.vn"]
     },
     {
       "plate": "30F88251",
       "type": 1,
       "enable": false
     }
   ],
-  "api": ["phatnguoi.vn", "checkphatnguoi.vn", "phatnguoi.vn", "csgt.vn"],
+  "apis": ["phatnguoi.vn", "checkphatnguoi.vn", "phatnguoi.vn", "csgt.vn"],
   "notifications": [
     {
       "telegram": {

diff --git a/src/check_phat_nguoi/config/models/config.py b/src/check_phat_nguoi/config/models/config.py
@@ -31,7 +31,7 @@ class Config(BaseModel):
         description="Danh sách các thiết lập để thông báo",
         default_factory=tuple,
     )
-    api: tuple[ApiEnum, ...] | ApiEnum = Field(
+    apis: tuple[ApiEnum, ...] = Field(
         title="API",
         description="Sử dụng API từ trang web nào. Mặc định sẽ là list các API như trong schema hiển thị và dừng khi 1 API lấy dữ liệu thành công. Có thể điền giá trị trùng để retry. Hoặc chỉ dùng 1 API",
         default=(ApiEnum.phatnguoi_vn, ApiEnum.checkphatnguoi_vn),

diff --git a/src/check_phat_nguoi/config/models/plate_info.py b/src/check_phat_nguoi/config/models/plate_info.py
@@ -30,7 +30,7 @@ class PlateInfo(BaseModel):
         description="Kích hoạt",
         default=True,
     )
-    api: tuple[ApiEnum, ...] | ApiEnum | None = Field(
+    apis: tuple[ApiEnum, ...] | None = Field(
         description='Sử dụng API từ trang web nào. Config giống "api" ở ngoài .Để trống sẽ sử dụng API define ở scope ngoài.',
         title="API",
         default=None,
@@ -49,7 +49,7 @@ def __hash__(self) -> int:
             hash(self.plate)
             + hash(self.type)
             + hash(self.enabled)
-            + hash(self.api)
+            + hash(self.apis)
             + hash(self.owner)
         )
 
@@ -65,14 +65,16 @@ def __eq__(self, other: Any):
                     all(
                         x == y
                         for x, y in zip(
-                            (self.api,) if isinstance(self.api, ApiEnum) else self.api,
-                            (other.api,)
-                            if isinstance(other.api, ApiEnum)
-                            else other.api,
+                            (self.apis,)
+                            if isinstance(self.apis, ApiEnum)
+                            else self.apis,
+                            (other.apis,)
+                            if isinstance(other.apis, ApiEnum)
+                            else other.apis,
                         )
                     )
-                    if self.api and other.api
-                    else (not self.api and not other.api)
+                    if self.apis and other.apis
+                    else (not self.apis and not other.apis)
                 )
             )
         return False

diff --git a/src/check_phat_nguoi/get_data/get_data.py b/src/check_phat_nguoi/get_data/get_data.py
@@ -24,15 +24,7 @@ def __init__(self) -> None:
 
     async def _get_data_for_plate(self, plate_info: PlateInfo) -> None:
         # NOTE: The config has constraint that config.api will be at least 1 api in tuple
-        apis: tuple[ApiEnum, ...] = (
-            plate_info.api
-            if isinstance(plate_info.api, tuple)
-            else (plate_info.api,)
-            if plate_info.api
-            else config.api
-            if isinstance(config.api, tuple)
-            else (config.api,)
-        )
+        apis: tuple[ApiEnum, ...] = plate_info.apis if plate_info.apis else config.apis
         engine: BaseGetDataEngine
         for api in apis:
             match api: