change to update or create from create

codecov · Jan 31, 2025 · 4d1167b · 4d1167b
1 parent 7e5a947
commit 4d1167b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/core/management/commands/insert_data_to_db_from_csv.py b/core/management/commands/insert_data_to_db_from_csv.py
@@ -51,8 +51,14 @@ def handle(self, *args, **kwargs):
                         )
                         continue
 
-                Model.objects.create(**model_data)
-                self.stdout.write(self.style.SUCCESS(f"Inserted row: {row}"))
+                # Use update_or_create
+                Model.objects.update_or_create(
+                    defaults=model_data,
+                    # Specify the fields to identify the record
+                    # For example, if 'id' is the unique field:
+                    id=row.get("id"),
+                )
+                self.stdout.write(self.style.SUCCESS(f"Processed row: {row}"))
 
         self.stdout.write(
             self.style.SUCCESS(