Made the requested changes

ohcnetwork · Oct 11, 2022 · dbe5e4e · dbe5e4e
1 parent c4a376a
commit dbe5e4e
Showing 1 changed file with 16 additions and 7 deletions.
diff --git a/care/facility/migrations/0315_auto_20220905_1008.py b/care/facility/migrations/0315_auto_20220905_1008.py
@@ -23,12 +23,17 @@ def get_occupation(apps, schema_editor):
     patients = PatientRegistration.objects.all()
 
     for patient in patients:
-        occupation_list.append(
-            {
-                "patient": patient,
-                "occupation": OccupationChoices[patient.meta_info.occupation],
-            }
-        )
+        occupation = ""
+
+        if patient.meta_info and patient.meta_info.occupation:
+            occupation = OccupationChoices[patient.meta_info.occupation]
+
+            occupation_list.append(
+                {
+                    "patient": patient,
+                    "occupation": occupation,
+                }
+            )
 
 
 def populate_occupation(apps, schema_editor):
@@ -51,7 +56,11 @@ class Migration(migrations.Migration):
         migrations.AlterField(
             model_name="patientmetainfo",
             name="occupation",
-            field=models.CharField(blank=True, max_length=100, null=True),
+            field=models.CharField(
+                blank=True,
+                max_length=100,
+                null=True,
+            ),
         ),
         migrations.RunPython(populate_occupation, migrations.RunPython.noop),
     ]