Added page types

This commit is contained in:
dormantman 2019-12-11 13:03:31 +03:00
parent 096d5dab18
commit 19c42ab1d7

View File

@ -1,8 +1,5 @@
"""Tag views.""" """Tag views."""
from django.conf import settings from django.conf import settings
from django_elasticsearch_dsl_drf import constants
from django_elasticsearch_dsl_drf.filter_backends import FilteringFilterBackend
from elasticsearch_dsl import TermsFacet
from rest_framework import generics from rest_framework import generics
from rest_framework import mixins from rest_framework import mixins
from rest_framework import permissions from rest_framework import permissions
@ -81,7 +78,9 @@ class FiltersTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
result_list = serializer.data result_list = serializer.data
query_params = request.query_params query_params = request.query_params
if 'toque_number__in' in query_params: params_type = query_params['type']
if params_type == 'restaurant' and 'toque_number__in' in query_params:
toques = { toques = {
"index_name": "toque_number", "index_name": "toque_number",
"label_translated": "Toques", "label_translated": "Toques",
@ -94,7 +93,7 @@ class FiltersTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
} }
result_list.append(toques) result_list.append(toques)
if 'wine_region_id__in' in query_params: if params_type == 'winery' and 'wine_region_id__in' in query_params:
try: try:
wine_region_id = int(query_params['wine_region_id__in']) wine_region_id = int(query_params['wine_region_id__in'])
@ -114,7 +113,7 @@ class FiltersTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
except ValueError: except ValueError:
pass pass
if 'works_noon__in' in query_params: if params_type == 'restaurant' and 'works_noon__in' in query_params:
week_days = ("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday") week_days = ("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday")
works_noon = { works_noon = {
"index_name": "works_noon", "index_name": "works_noon",
@ -128,11 +127,11 @@ class FiltersTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
} }
result_list.append(works_noon) result_list.append(works_noon)
if 'works_evening__in' in query_params: if params_type == 'restaurant' and 'works_evening__in' in query_params:
week_days = ("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday") week_days = ("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday")
works_evening = { works_evening = {
"index_name": "works_evening", "index_name": "works_evening",
"label_translated": "Open noon", "label_translated": "Open evening",
"param_name": "works_evening__in", "param_name": "works_evening__in",
"filters": [{ "filters": [{
"id": weekday, "id": weekday,
@ -142,7 +141,7 @@ class FiltersTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
} }
result_list.append(works_evening) result_list.append(works_evening)
if 'works_now' in query_params: if params_type in ('restaurant', 'artisan') and 'works_now' in query_params:
works_now = { works_now = {
"index_name": "open_now", "index_name": "open_now",
"label_translated": "Open now", "label_translated": "Open now",
@ -151,6 +150,11 @@ class FiltersTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
} }
result_list.append(works_now) result_list.append(works_now)
if 'tags_id__in' in query_params:
# filtering by params_type and tags id
# todo: result_list.append( filtering_data )
pass
return Response(result_list) return Response(result_list)
@ -244,4 +248,4 @@ class TagCategoryBackOfficeViewSet(mixins.CreateModelMixin,
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT_TYPE: if obj_type == self.bind_object_serializer_class.ESTABLISHMENT_TYPE:
tag_category.establishment_types.remove(related_object) tag_category.establishment_types.remove(related_object)
elif obj_type == self.bind_object_serializer_class.NEWS_TYPE: elif obj_type == self.bind_object_serializer_class.NEWS_TYPE:
tag_category.news_types.remove(related_object) tag_category.news_types.remove(related_object)