Update hospital_doctor.py

ohcnetwork · Dec 7, 2024 · 80aefde · 80aefde
1 parent fa6f6bd
commit 80aefde
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/care/facility/api/serializers/hospital_doctor.py b/care/facility/api/serializers/hospital_doctor.py
@@ -1,4 +1,5 @@
 from rest_framework import serializers
+from django.db.models import Sum
 
 from care.facility.api.serializers import TIMESTAMP_FIELDS
 from care.facility.models import DOCTOR_TYPES, HospitalDoctors
@@ -19,11 +20,14 @@ class Meta:
         )
         exclude = (*TIMESTAMP_FIELDS, "facility", "external_id")
 
-        def to_representation(self, instance):
-            representation = super().to_representation(instance)
+    def to_representation(self, instance):
+        representation = super().to_representation(instance)
+        try:
             representation["total_doctors"] = (
                 HospitalDoctors.objects.filter(facility=instance.facility)
                 .aggregate(total_doctors=Sum("count"))["total_doctors"]
                 or 0
             )
-            return representation
+        except Exception as e:
+            representation["total_doctors"] = 0
+        return representation