ES. Establishment. filter by collection_type

apps/search_indexes/documents/establishment.py

@@ -55,6 +55,10 @@ class EstablishmentDocument(Document):
             ),
         }
     )
+    collections = fields.ObjectField(properties={
+        'id': fields.IntegerField(attr='collection.id'),
+        'collection_type': fields.IntegerField(attr='collection.collection_type'),
+    })
 
     class Django:
 

apps/search_indexes/serializers.py

@@ -60,6 +60,7 @@ class EstablishmentDocumentSerializer(DocumentSerializer):
             'description_translated',
             'tags',
             'address',
+            'collections',
         )
 
     @staticmethod

apps/search_indexes/views.py

@@ -91,5 +91,8 @@ class EstablishmentDocumentViewSet(BaseDocumentViewSet):
         },
         'tags_category_id': {
             'field': 'tags.category.id',
-        }
+        },
+        'collection_type': {
+            'field': 'collections.collection_type'
+        },
     }