Update 0047_auto_20230517_0944.py

Testing in production for the L.
nottinghamtec · May 19, 2023 · 7d8dddb · 7d8dddb
1 parent 1104f10
commit 7d8dddb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/RIGS/migrations/0047_auto_20230517_0944.py b/RIGS/migrations/0047_auto_20230517_0944.py
@@ -12,10 +12,9 @@ def migrate_old_data(apps, schema_editor):
     for ec in EventChecklist.objects.all():
         for crew in ec.crew.all():
             try:
-                vehicle = ec.vehicles.get(driver=crew.crewmember)
+                EventCheckIn.objects.create(event=ec.event, person=crew.crewmember, role=crew.role, time=crew.start, end_time=crew.end, vehicle=ec.vehicles.get(driver=crew.crewmember).vehicle)
             except ObjectDoesNotExist:
-                vehicle = None
-            EventCheckIn.objects.create(event=ec.event, person=crew.crewmember, role=crew.role, time=crew.start, end_time=crew.end, vehicle=vehicle.vehicle)
+                EventCheckIn.objects.create(event=ec.event, person=crew.crewmember, role=crew.role, time=crew.start, end_time=crew.end)
 
 
 def revert(apps, schema_editor):