From 36b47a6cba7bca4606ae8a1bf04f540d6ba8cea6 Mon Sep 17 00:00:00 2001
From: Semyon Yekhmenin <s.yekhmenin@id-east.ru>
Date: Fri, 15 Nov 2019 09:13:28 +0000
Subject: [PATCH] Added establishment subtype to product view

---
 apps/establishment/serializers/common.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/apps/establishment/serializers/common.py b/apps/establishment/serializers/common.py
index 0053a424..46a8619e 100644
--- a/apps/establishment/serializers/common.py
+++ b/apps/establishment/serializers/common.py
@@ -131,7 +131,8 @@ class EstablishmentSubTypeBaseSerializer(serializers.ModelSerializer):
             'id',
             'name',
             'name_translated',
-            'establishment_type'
+            'establishment_type',
+            'index_name',
         ]
         extra_kwargs = {
             'name': {'write_only': True},
@@ -173,6 +174,7 @@ class EstablishmentShortSerializer(serializers.ModelSerializer):
     """Short serializer for establishment."""
     city = CitySerializer(source='address.city', allow_null=True)
     establishment_type = EstablishmentTypeGeoSerializer()
+    establishment_subtypes = EstablishmentSubTypeBaseSerializer(many=True)
 
     class Meta:
         """Meta class."""
@@ -183,7 +185,8 @@ class EstablishmentShortSerializer(serializers.ModelSerializer):
             'index_name',
             'slug',
             'city',
-            'establishment_type'
+            'establishment_type',
+            'establishment_subtypes',
         ]