From a0aa503cfa59211a1286d745f2a2c5bf9e99c024 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] Add index_name type field to geo

---
 apps/establishment/serializers/common.py | 25 ++++++++++++++++++++++++
 apps/establishment/views/web.py          |  2 +-
 2 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/apps/establishment/serializers/common.py b/apps/establishment/serializers/common.py
index 14be142a..12e217a9 100644
--- a/apps/establishment/serializers/common.py
+++ b/apps/establishment/serializers/common.py
@@ -118,6 +118,18 @@ class EstablishmentTypeBaseSerializer(serializers.ModelSerializer):
             'use_subtypes': {'write_only': True},
         }
 
+class EstablishmentTypeGeoSerializer(EstablishmentTypeBaseSerializer):
+    """Serializer for EstablishmentType model w/ index_name."""
+
+    class Meta(EstablishmentTypeBaseSerializer.Meta):
+        fields = EstablishmentTypeBaseSerializer.Meta.fields + [
+            'index_name'
+        ]
+        extra_kwargs = {
+            **EstablishmentTypeBaseSerializer.Meta.extra_kwargs,
+            'index_name': {'read_only': True},
+        }
+
 
 class EstablishmentSubTypeBaseSerializer(serializers.ModelSerializer):
     """Serializer for EstablishmentSubType models."""
@@ -185,6 +197,19 @@ class EstablishmentBaseSerializer(ProjectModelSerializer):
         ]
 
 
+class EstablishmentGeoSerializer(EstablishmentBaseSerializer):
+    """Serializer for Geo view."""
+
+    type = EstablishmentTypeGeoSerializer(source='establishment_type', read_only=True)
+
+    class Meta(EstablishmentBaseSerializer.Meta):
+        """Meta class."""
+
+        fields = EstablishmentBaseSerializer.Meta.fields + [
+            'type'
+        ]
+
+
 class EstablishmentDetailSerializer(EstablishmentBaseSerializer):
     """Serializer for Establishment model."""
 
diff --git a/apps/establishment/views/web.py b/apps/establishment/views/web.py
index 0699d9d0..63033d4b 100644
--- a/apps/establishment/views/web.py
+++ b/apps/establishment/views/web.py
@@ -152,7 +152,7 @@ class EstablishmentFavoritesCreateDestroyView(generics.CreateAPIView, generics.D
 class EstablishmentNearestRetrieveView(EstablishmentListView, generics.ListAPIView):
     """Resource for getting list of nearest establishments."""
 
-    serializer_class = serializers.EstablishmentBaseSerializer
+    serializer_class = serializers.EstablishmentGeoSerializer
     filter_class = filters.EstablishmentFilter
 
     def get_queryset(self):