feat: add print console

config.sample.json

@@ -52,5 +52,6 @@
   "asynchronous": true,
   "detail": true,
   "detail_log": false,
+  "print_console": true,
   "log_level": "INFO"
 }

src/check_phat_nguoi/__main__.py

@@ -5,6 +5,7 @@
 from check_phat_nguoi.context import plates_context
 from check_phat_nguoi.get_data import GetData
 from check_phat_nguoi.notify import SendNotifications
+from check_phat_nguoi.print_console import PrintConsole
 from check_phat_nguoi.utils import setup_logger
 
 logger = getLogger(__name__)
@@ -15,6 +16,7 @@ async def async_main() -> None:
     logger.debug(f"Config read: {config}")
     await GetData().get_data()
     logger.debug(f"Data got: {plates_context.plates}")
+    PrintConsole().print_console()
     await SendNotifications().send()
 
 

src/check_phat_nguoi/config/models/config.py

@@ -34,6 +34,11 @@ class Config(BaseModel):
         default=(ApiEnum.checkphatnguoi_vn),
         min_length=1,
     )
+    print_console: bool = Field(
+        title="In thông tin ra console",
+        description="In thông tin của các biển ra console",
+        default=True,
+    )
     pending_fines_only: bool = Field(
         title="Lọc chưa nộp phạt",
         description="Chỉ lọc các thông tin vi phạm chưa nộp phạt",

src/check_phat_nguoi/context/plates/models/plate_detail.py

@@ -27,54 +27,49 @@ def __eq__(self, other: Any):
             return self.plate == other.plate
         return False
 
-    def __str__(self):
+    # TODO: Handle show details later when main updates that option
+    def __str__(self) -> str:
         def create_violation_str(violation: ViolationDetail, index: int) -> str:
+            violation_str: str = (
+                f"Lỗi vi phạm thứ {index}:"
+                + (f"\nMàu biển: {violation.color}" if violation.color else "")
+                + (f"\nThời điểm vi phạm: {violation.date}" if violation.date else "")
+                + (
+                    f"\nVị trí vi phạm: {violation.location}"
+                    if violation.location
+                    else ""
+                )
+                + (
+                    f"\nHành vi vi phạm: {violation.violation}"
+                    if violation.violation
+                    else ""
+                )
+                + (
+                    f"\nTrạng thái: {'Chưa xử phạt' if not violation.status else 'Đã xử phạt'}"
+                    if violation.status is not None
+                    else ""
+                )
+                + (
+                    f"\nĐơn vị phát hiện vi phạm: {violation.enforcement_unit}"
+                    if violation.enforcement_unit
+                    else ""
+                )
+            )
             resolution_offices: str | None = (
-                f"""
-                   Nơi giải quyết vụ việc:
-                   {
-                    "\n".join(
-                        resolution_office_detail
-                        for resolution_office_detail in violation.resolution_offices_details
-                    )
-                }
-                """
+                "\n"
+                + "Nơi giải quyết vụ việc:"
+                + "\n"
+                + "\n".join(
+                    resolution_office_detail.strip()
+                    for resolution_office_detail in violation.resolution_offices_details
+                )
                 if violation.resolution_offices_details
                 else None
             )
-            # TODO: Keep going on
-            violation_str: str = (
-                f"Lỗi vi phạm thứ {index}:" + f"\nMàu biển: {violation.color}"
-                if violation.color
-                else ""
-            )
-            # violation_str = "\n".join(
-            #     line
-            #     for line in f"""
-            # Lỗi vi phạm thứ {index}:
-            #     Màu biển: {violation.color if violation.color else " "}
-            #     Thời điểm vi phạm: {violation.date if violation.date else " "}
-            #     Vị trí vi phạm: {violation.location if violation.location else " "}
-            #     Hành vi vi phạm: {violation.violation if violation.violation else " "}
-            #                     Trạng thái: {"Đã xử phạt" if violation.status else ("Chưa xử phạt" if not violation.status else " ")}
-            #     Đơn vị phát hiện vi phạm: {violation.enforcement_unit if violation.enforcement_unit else " "}
-            # """.splitlines()
-            #     if line.strip()
-            # )
-            return (
-                "\n".join([violation_str, resolution_offices])
-                if resolution_offices
-                else violation_str
-            )
+            return violation_str + (resolution_offices if resolution_offices else "")
 
-        # TODO: Ye going on hehehe
-        plate_detail: str = "\n".join(
-            line
-            for line in f"""
-        Biển số: {self.plate}
-        Chủ sở hữu: {self.owner if self.owner else " "}
-        """.splitlines()
-            if line.strip()
+        plate_detail: str = f"Biển số: {self.plate}" + (
+            f"\nChủ sở hữu: {self.owner}" if self.owner else ""
         )
 
         if self.violations:

src/check_phat_nguoi/print_console/__init__.py

@@ -0,0 +1,3 @@
+from .print_console import PrintConsole
+
+__all__ = ["PrintConsole"]

src/check_phat_nguoi/print_console/print_console.py

@@ -0,0 +1,20 @@
+from logging import getLogger
+
+from check_phat_nguoi.config.config_reader import config
+from check_phat_nguoi.context import PlateDetail, plates_context
+
+logger = getLogger(__name__)
+
+
+class PrintConsole:
+    def __init__(self):
+        self.plate_details: tuple[PlateDetail, ...] = plates_context.plates
+
+    def print_console(self) -> None:
+        if not config.print_console:
+            return
+        if not self.plate_details:
+            logger.info("No plate details. Skip printing")
+            return
+        for plate_detail in self.plate_details:
+            print(plate_detail)