update serializer validation

ohcnetwork · Jun 6, 2024 · 4cbe387 · 4cbe387
1 parent 5841662
commit 4cbe387
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/care/facility/api/serializers/patient_consultation.py b/care/facility/api/serializers/patient_consultation.py
@@ -902,8 +902,10 @@ def validate(self, attrs):
                 "Only Home Facility Staff can create consent for a Consultation"
             )
 
-        if attrs.get("type") == ConsentType.PATIENT_CODE_STATUS and not attrs.get(
-            "patient_code_status"
+        if (
+            attrs.get("type", None)
+            and attrs.get("type") == ConsentType.PATIENT_CODE_STATUS
+            and not attrs.get("patient_code_status")
         ):
             raise ValidationError(
                 {
@@ -913,8 +915,10 @@ def validate(self, attrs):
                 }
             )
 
-        if attrs.get("type") != ConsentType.PATIENT_CODE_STATUS and attrs.get(
-            "patient_code_status"
+        if (
+            attrs.get("type", None)
+            and attrs["type"] != ConsentType.PATIENT_CODE_STATUS
+            and attrs.get("patient_code_status")
         ):
             raise ValidationError(
                 {