diff --git a/apps/establishment/models.py b/apps/establishment/models.py index ef3630c4..d258ba41 100644 --- a/apps/establishment/models.py +++ b/apps/establishment/models.py @@ -461,6 +461,11 @@ class Establishment(ProjectBaseMixin, URLImageMixin, TranslatedFieldsMixin): """ Used for indexing working by day """ return [ret.weekday for ret in self.schedule.all() if ret.works_at_noon] + @property + def works_at_weekday(self): + """ Used for indexing by working whole day criteria """ + return [ret.weekday for ret in self.schedule.all()] + @property def works_evening(self): """ Used for indexing working by day """ diff --git a/apps/search_indexes/documents/establishment.py b/apps/search_indexes/documents/establishment.py index 19d419aa..8db113c1 100644 --- a/apps/search_indexes/documents/establishment.py +++ b/apps/search_indexes/documents/establishment.py @@ -37,6 +37,9 @@ class EstablishmentDocument(Document): works_noon = fields.ListField(fields.IntegerField( attr='works_noon' )) + works_at_weekday = fields.ListField(fields.IntegerField( + attr='works_at_weekday' + )) works_now = fields.BooleanField(attr='works_now') tags = fields.ObjectField( properties={ diff --git a/apps/search_indexes/serializers.py b/apps/search_indexes/serializers.py index bbc61735..6aee06d0 100644 --- a/apps/search_indexes/serializers.py +++ b/apps/search_indexes/serializers.py @@ -108,6 +108,7 @@ class EstablishmentDocumentSerializer(DocumentSerializer): 'schedule', 'works_noon', 'works_evening', + 'works_at_weekday', # 'works_now', # 'collections', # 'establishment_type', diff --git a/apps/search_indexes/views.py b/apps/search_indexes/views.py index 87b0ba64..92d96a64 100644 --- a/apps/search_indexes/views.py +++ b/apps/search_indexes/views.py @@ -156,6 +156,12 @@ class EstablishmentDocumentViewSet(BaseDocumentViewSet): constants.LOOKUP_QUERY_IN, ], }, + 'works_at_weekday': { + 'field': 'works_at_weekday', + 'lookups': [ + constants.LOOKUP_QUERY_IN, + ], + }, 'works_evening': { 'field': 'works_evening', 'lookups': [