From 202b03708081b1d35df26e37c68f701d7775f92c Mon Sep 17 00:00:00 2001
From: k9845 <khatririshi2430@gmail.com>
Date: Fri, 26 Apr 2024 14:08:50 +0545
Subject: [PATCH] Update local units health options

---
 local_units/serializers.py | 16 ++++++++++++++++
 local_units/views.py       |  6 +++++-
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/local_units/serializers.py b/local_units/serializers.py
index 7e267208b6..5fcff4c941 100644
--- a/local_units/serializers.py
+++ b/local_units/serializers.py
@@ -15,11 +15,25 @@
     HospitalType,
     BloodService,
     ProfessionalTrainingFacility,
+    GeneralMedicalService,
+    SpecializedMedicalService,
 )
 from api.models import Country
 from main.writable_nested_serializers import NestedCreateMixin, NestedUpdateMixin
 
 
+class GeneralMedicalServiceSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = GeneralMedicalService
+        fields = "__all__"
+
+
+class SpecializedMedicalServiceSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = SpecializedMedicalService
+        fields = "__all__"
+
+
 class AffiliationSerializer(serializers.ModelSerializer):
     class Meta:
         model = Affiliation
@@ -216,6 +230,8 @@ class LocalUnitOptionsSerializer(serializers.Serializer):
     hospital_type = HospitalTypeSerializer(many=True)
     blood_services = BloodServiceSerializer(many=True)
     professional_training_facilities = ProfessionalTrainingFacilitySerializer(many=True)
+    general_medical_services = GeneralMedicalServiceSerializer(many=True)
+    specialized_medical_services = SpecializedMedicalServiceSerializer(many=True)
 
 
 class MiniDelegationOfficeSerializer(serializers.ModelSerializer):
diff --git a/local_units/views.py b/local_units/views.py
index 7d6abf663d..a6e099fcbb 100644
--- a/local_units/views.py
+++ b/local_units/views.py
@@ -25,7 +25,9 @@
     HospitalType,
     BloodService,
     ProfessionalTrainingFacility,
-    VisibilityChoices
+    VisibilityChoices,
+    GeneralMedicalService,
+    SpecializedMedicalService,
 )
 from local_units.serializers import (
     LocalUnitSerializer,
@@ -84,6 +86,8 @@ def get_options(self, request, pk=None):
                     hospital_type=HospitalType.objects.all(),
                     blood_services=BloodService.objects.all(),
                     professional_training_facilities=ProfessionalTrainingFacility.objects.all(),
+                    general_medical_services=GeneralMedicalService.objects.all(),
+                    specialized_medical_services=SpecializedMedicalService.objects.all(),
                 )
             ).data
         )