Handle IntegrityError in bulk_update fallback logic

Wraps the fleet.objects.bulk_update call in a try/except block to catch IntegrityError exceptions. If a bulk update fails due to integrity issues, the code now falls back to updating vehicles individually, skipping those that still raise errors, and logs the process for better traceability.

Author: Kai-codin

tracking/management/commands/simulate_positions.py

@@ -5,6 +5,7 @@
 from datetime import timedelta
 from tracking.models import Trip
 from fleet.models import fleet
+from django.db import IntegrityError
 from routes.models import routeStop
 import time
 
@@ -196,12 +197,25 @@ def handle(self, *args, **kwargs):
         # ---------------------------------------------------------
         if vehicles_to_update:
             t3 = time.time()
-            fleet.objects.bulk_update(
-                vehicles_to_update,
-                ["sim_lat", "sim_lon", "sim_heading", "current_trip", "updated_at"],
-                batch_size=500
-            )
-            self.stdout.write(f"Updated {len(vehicles_to_update)} vehicles in {time.time() - t3:.2f}s")
+            try:
+                fleet.objects.bulk_update(
+                    vehicles_to_update,
+                    ["sim_lat", "sim_lon", "sim_heading", "current_trip", "updated_at"],
+                    batch_size=500
+                )
+                self.stdout.write(f"Updated {len(vehicles_to_update)} vehicles in {time.time() - t3:.2f}s")
+            except IntegrityError as e:
+                # Bulk update failed due to FK integrity (race or missing trip). Fall back
+                # to per-vehicle save so we can skip problematic updates.
+                self.stderr.write(f"Bulk update IntegrityError: {e}. Falling back to per-vehicle updates.")
+                updated = 0
+                for v in vehicles_to_update:
+                    try:
+                        v.save(update_fields=["sim_lat", "sim_lon", "sim_heading", "current_trip", "updated_at"])
+                        updated += 1
+                    except IntegrityError as e2:
+                        self.stderr.write(f"Skipping vehicle {v.id} due to IntegrityError: {e2}")
+                self.stdout.write(f"Fallback updated {updated} vehicles in {time.time() - t3:.2f}s")
 
         self.stdout.write(f"Total time: {time.time() - t0:.2f}s")