Update hospital_doctor.py

ohcnetwork · Dec 7, 2024 · fa6f6bd · fa6f6bd
1 parent bfd4f75
commit fa6f6bd
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/care/facility/api/serializers/hospital_doctor.py b/care/facility/api/serializers/hospital_doctor.py
@@ -8,11 +8,22 @@
 class HospitalDoctorSerializer(serializers.ModelSerializer):
     area_text = ChoiceField(choices=DOCTOR_TYPES, read_only=True, source="area")
     id = serializers.UUIDField(source="external_id", read_only=True)
+    total_doctors = serializers.IntegerField(read_only=True)
 
     class Meta:
         model = HospitalDoctors
         read_only_fields = (
             "id",
             "area_text",
+            "total_doctors"
         )
         exclude = (*TIMESTAMP_FIELDS, "facility", "external_id")
+
+        def to_representation(self, instance):
+            representation = super().to_representation(instance)
+            representation["total_doctors"] = (
+                HospitalDoctors.objects.filter(facility=instance.facility)
+                .aggregate(total_doctors=Sum("count"))["total_doctors"]
+                or 0
+            )
+            return representation