Geo subtypes

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',
         ]