catch each row that fails

Author: ajay-sentry

core/management/commands/insert_data_to_db_from_csv.py

@@ -51,13 +51,15 @@ def handle(self, *args, **kwargs):
                         )
                         continue
 
-                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"Inserted row: {row}"))
+                try:
+                    Model.objects.update_or_create(
+                        defaults=model_data,
+                        id=row.get("id"),
+                    )
+                    self.stdout.write(self.style.SUCCESS(f"Inserted row: {row}"))
+                except Exception as e:
+                    self.stdout.write(self.style.ERROR(f"Error inserting row: {e}"))
+                    continue
 
         self.stdout.write(
             self.style.SUCCESS(