fix: updated migration steps

ohcnetwork · Aug 9, 2023 · 4c85bbe · 4c85bbe
1 parent 4286de9
commit 4c85bbe
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/care/facility/migrations/0346_add_recommend_discharge_action.py b/care/facility/migrations/0346_add_recommend_discharge_action.py
@@ -3,10 +3,16 @@
 
 def update_recommend_discharge(apps, schema_editor):
     patient_model = apps.get_model("facility", "PatientRegistration")
+    patient_objs = []
     for patient in patient_model.objects.all():
-        if patient.last_consultation.last_daily_round.recommend_discharge:
+        if (
+            patient.last_consultation
+            and patient.last_consultation.last_daily_round
+            and patient.last_consultation.last_daily_round.recommend_discharge
+        ):
             patient.action = 90
-            patient.save()
+            patient_objs.append(patient)
+    patient_model.objects.bulk_update(patient_objs, ["action"])
 
 
 class Migration(migrations.Migration):