feat: get the list of PlateInfoContextModel from get_data function

check-phat-nguoi · Dec 25, 2024 · 35037e1 · 35037e1
1 parent ee2df77
commit 35037e1
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/check_phat_nguoi/modules/get_data/checkphatnguoi.py b/src/check_phat_nguoi/modules/get_data/checkphatnguoi.py
@@ -80,6 +80,13 @@ def _create_violation_model(data: Dict):
     @override
     def get_data(self) -> list[PlateInfoContextModel]:
         self._multi_thread_get_data()
-        plate_infos = [
-            GetDataCheckPhatNguoi.get_plate_violation(plate) for plate in self.data_dict
+        plate_infos: list[PlateInfoContextModel] = [
+            PlateInfoContextModel(
+                plate=plate,
+                violation=GetDataCheckPhatNguoi.get_plate_violation(
+                    plate_violation_dict=plate_violation_dict
+                ),
+            )
+            for plate, plate_violation_dict in self.data_dict.items()
         ]
+        return plate_infos