fix

Signed-off-by: Aakash Singh <[email protected]>
ohcnetwork · Aug 18, 2023 · a76402a · a76402a
1 parent 5383974
commit a76402a
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/care/facility/api/serializers/bed.py b/care/facility/api/serializers/bed.py
@@ -198,12 +198,18 @@ def validate(self, attrs):
                 raise ValidationError(
                     {"bed": "Bed is already in use by another patient"}
                 )
+            previous_consultation_bed = consultation.current_bed
 
-            if consultation.current_bed.bed == bed and set(
-                consultation.current_bed.assets.order_by("external_id").values_list(
-                    "external_id", flat=True
+            if (
+                previous_consultation_bed
+                and previous_consultation_bed.bed == bed
+                and set(
+                    previous_consultation_bed.assets.order_by(
+                        "external_id"
+                    ).values_list("external_id", flat=True)
                 )
-            ) == set(attrs.get("assets", [])):
+                == set(attrs.get("assets", []))
+            ):
                 raise ValidationError(
                     {"consultation": "These set of bed and assets are already assigned"}
                 )
@@ -257,8 +263,8 @@ def create(self, validated_data):
                     ]
                 )
 
-                consultation.current_bed = obj
-                consultation.save(update_fields=["current_bed"])
+            consultation.current_bed = obj
+            consultation.save(update_fields=["current_bed"])
             return obj
 
     def update(self, instance: ConsultationBed, validated_data) -> ConsultationBed: