Formatting

Author: jolpica

jolpica/formula_one/importer/importer.py

@@ -59,17 +59,14 @@ def get_model_import_priority(cls, import_key: ModelImport) -> int:
     def save_deserialisation_result_to_db(cls, result: DeserialisationResult) -> dict:
         prioritised_items = sorted(result.instances.items(), key=lambda x: cls.get_model_import_priority(x[0]))
 
-        import_stats = {"updated_count": 0, "created_count": 0, "models": {}}
+        import_stats: dict = {
+            "updated_count": 0,
+            "created_count": 0,
+            "models": defaultdict(lambda: {"updated_count": 0, "created_count": 0, "updated": [], "created": []}),
+        }
 
         for model_import, instances in prioritised_items:
             model_name = model_import.model_class.__name__
-            if model_name not in import_stats["models"]:
-                import_stats["models"][model_name] = {
-                    "updated_count": 0,
-                    "created_count": 0,
-                    "updated": [],
-                    "created": [],
-                }
 
             for ins in instances:
                 try:

jolpica_api/data_import/views.py

@@ -41,7 +41,7 @@ def put(self, request: Request) -> Response:
         try:
             request_data = ImportDataRequestData.model_validate(request.data)
         except ValidationError as ex:
-            errors = ex.errors(include_url=False)
+            errors: list = ex.errors(include_url=False)
             dry_run = request.data.get("dry_run", True)
             log_data_import_result(
                 request.user,