check if event type exists before running migration

ohcnetwork · Aug 14, 2024 · 6bc7099 · 6bc7099
1 parent 9020dea
commit 6bc7099
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/care/facility/migrations/0448_rename_ventilator_fi02_dailyround_ventilator_fio2.py b/care/facility/migrations/0448_rename_ventilator_fi02_dailyround_ventilator_fio2.py
@@ -11,7 +11,11 @@ def forward_rename_dailyround_events(apps, schema_editor):
             "facility", "PatientConsultationEvent"
         )
 
-        event_type_id = EventType.objects.get(name="RESPIRATORY_SUPPORT").id
+        event_type = EventType.objects.filter(name="RESPIRATORY_SUPPORT")
+        if not event_type.exists():
+            return
+        event_type_id = event_type.first().id
+
         paginator = Paginator(
             PatientConsultationEvent.objects.filter(
                 object_model="DailyRound",
@@ -36,7 +40,11 @@ def reverse_rename_dailyround_events(apps, schema_editor):
             "facility", "PatientConsultationEvent"
         )
 
-        event_type_id = EventType.objects.get(name="RESPIRATORY_SUPPORT").id
+        event_type = EventType.objects.filter(name="RESPIRATORY_SUPPORT")
+        if not event_type.exists():
+            return
+        event_type_id = event_type.first().id
+
         paginator = Paginator(
             PatientConsultationEvent.objects.filter(
                 object_model="DailyRound",