From d4792cb342e4cf68c985021a104b89067b85379e Mon Sep 17 00:00:00 2001
From: Kuroshini <o.hayatov@id-east.ru>
Date: Wed, 16 Oct 2019 14:11:07 +0300
Subject: [PATCH] Geo subtypes

---
 apps/establishment/serializers/common.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/apps/establishment/serializers/common.py b/apps/establishment/serializers/common.py
index 3f42c15b..176cd2a7 100644
--- a/apps/establishment/serializers/common.py
+++ b/apps/establishment/serializers/common.py
@@ -149,6 +149,18 @@ class EstablishmentSubTypeBaseSerializer(serializers.ModelSerializer):
             'establishment_type': {'write_only': True}
         }
 
+class EstablishmentSubTypeGeoSerializer(EstablishmentSubTypeBaseSerializer):
+    """Serializer for EstablishmentSuType model w/ index_name."""
+
+    class Meta(EstablishmentSubTypeBaseSerializer.Meta):
+        fields = EstablishmentSubTypeBaseSerializer.Meta.fields + [
+            'index_name'
+        ]
+        extra_kwargs = {
+            **EstablishmentSubTypeBaseSerializer.Meta.extra_kwargs,
+            'index_name': {'read_only': True},
+        }
+
 
 class EstablishmentEmployeeSerializer(serializers.ModelSerializer):
     """Serializer for actual employees."""
@@ -200,12 +212,14 @@ class EstablishmentGeoSerializer(EstablishmentBaseSerializer):
     """Serializer for Geo view."""
 
     type = EstablishmentTypeGeoSerializer(source='establishment_type', read_only=True)
+    subtypes = EstablishmentSubTypeGeoSerializer(many=True, source='establishment_subtypes')
 
     class Meta(EstablishmentBaseSerializer.Meta):
         """Meta class."""
 
         fields = EstablishmentBaseSerializer.Meta.fields + [
-            'type'
+            'type',
+            'subtypes',
         ]