275 lines
9.7 KiB
Python
275 lines
9.7 KiB
Python
"""Tag views."""
|
|
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 mixins
|
|
from rest_framework import permissions
|
|
from rest_framework import status
|
|
from rest_framework import viewsets
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from search_indexes import utils
|
|
from search_indexes.documents import EstablishmentDocument
|
|
from search_indexes.filters import CustomFacetedSearchFilterBackend
|
|
from search_indexes.filters import CustomSearchFilterBackend
|
|
from search_indexes.serializers import EstablishmentDocumentSerializer
|
|
from tag import filters
|
|
from tag import models
|
|
from tag import serializers
|
|
from utils.pagination import ESDocumentPagination
|
|
|
|
|
|
class ChosenTagsView(generics.ListAPIView, viewsets.GenericViewSet):
|
|
pagination_class = None
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.TagBaseSerializer
|
|
filterset_class = filters.TagsFilterSet
|
|
queryset = models.Tag.objects.all()
|
|
|
|
def get_queryset(self):
|
|
result_tags_ids = models.ChosenTagSettings.objects \
|
|
.by_country_code(self.request.country_code) \
|
|
.values('tag_id')
|
|
return models.Tag.objects \
|
|
.filter(id__in=result_tags_ids) \
|
|
.order_by_priority()
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
# TMP TODO remove it later
|
|
# Временный хардкод для демонстрации > 15 ноября, потом удалить!
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
|
|
page = self.paginate_queryset(queryset)
|
|
if page is not None:
|
|
serializer = self.get_serializer(page, many=True)
|
|
return self.get_paginated_response(serializer.data)
|
|
|
|
serializer = self.get_serializer(queryset, many=True)
|
|
result_list = serializer.data
|
|
if request.query_params.get('type') and (settings.ESTABLISHMENT_CHOSEN_TAGS or settings.NEWS_CHOSEN_TAGS):
|
|
ordered_list = settings.ESTABLISHMENT_CHOSEN_TAGS if request.query_params.get(
|
|
'type') == 'establishment' else settings.NEWS_CHOSEN_TAGS
|
|
result_list = sorted(result_list, key=lambda x: ordered_list.index(x['index_name']))
|
|
return Response(result_list)
|
|
|
|
|
|
# User`s views & viewsets
|
|
class TagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
|
|
"""ViewSet for TagCategory model."""
|
|
|
|
filterset_class = filters.TagCategoryFilterSet
|
|
pagination_class = None
|
|
permission_classes = (permissions.AllowAny,)
|
|
queryset = models.TagCategory.objects.with_tags().with_base_related(). \
|
|
distinct()
|
|
serializer_class = serializers.TagCategoryBaseSerializer
|
|
|
|
|
|
# User`s views & viewsets
|
|
class TestTagCategoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
|
|
"""ViewSet for TagCategory model."""
|
|
|
|
filterset_classes = [
|
|
filters.TagCategoryFilterSet,
|
|
FilteringFilterBackend,
|
|
CustomSearchFilterBackend,
|
|
CustomFacetedSearchFilterBackend,
|
|
]
|
|
|
|
document = EstablishmentDocument
|
|
pagination_class = ESDocumentPagination
|
|
permission_classes = (permissions.AllowAny,)
|
|
queryset = models.TagCategory.objects.with_tags().with_base_related(). \
|
|
distinct()
|
|
serializer_class = serializers.TestBaseSerializer
|
|
|
|
faceted_search_fields = {
|
|
'works_at_weekday': {
|
|
'field': 'works_at_weekday',
|
|
'facet': TermsFacet,
|
|
'enabled': True,
|
|
},
|
|
'toque_number': {
|
|
'field': 'toque_number',
|
|
'enabled': True,
|
|
'facet': TermsFacet,
|
|
},
|
|
'works_noon': {
|
|
'field': 'works_noon',
|
|
'facet': TermsFacet,
|
|
'enabled': True,
|
|
},
|
|
'works_evening': {
|
|
'field': 'works_evening',
|
|
'facet': TermsFacet,
|
|
'enabled': True,
|
|
},
|
|
'works_now': {
|
|
'field': 'works_now',
|
|
'facet': TermsFacet,
|
|
'enabled': True,
|
|
},
|
|
'tag': {
|
|
'field': 'tags.id',
|
|
'facet': TermsFacet,
|
|
'enabled': True,
|
|
'options': {
|
|
'size': utils.FACET_MAX_RESPONSE,
|
|
},
|
|
}
|
|
}
|
|
|
|
search_fields = {
|
|
'name': {
|
|
'fuzziness': 'auto:2,5',
|
|
'boost': 4
|
|
},
|
|
'transliterated_name': {
|
|
'fuzziness': 'auto:2,5',
|
|
'boost': 3
|
|
},
|
|
'description': {'fuzziness': 'auto:2,5'},
|
|
}
|
|
translated_search_fields = (
|
|
'description',
|
|
)
|
|
filter_fields = {
|
|
'slug': 'slug',
|
|
'tag': {
|
|
'field': 'tags.id',
|
|
'lookups': [constants.LOOKUP_QUERY_IN]
|
|
},
|
|
'toque_number': {
|
|
'field': 'toque_number',
|
|
'lookups': [
|
|
constants.LOOKUP_FILTER_RANGE,
|
|
constants.LOOKUP_QUERY_GT,
|
|
constants.LOOKUP_QUERY_GTE,
|
|
constants.LOOKUP_QUERY_LT,
|
|
constants.LOOKUP_QUERY_LTE,
|
|
constants.LOOKUP_QUERY_IN,
|
|
]
|
|
},
|
|
|
|
'works_noon': {
|
|
'field': 'works_noon',
|
|
'lookups': [
|
|
constants.LOOKUP_QUERY_IN,
|
|
],
|
|
},
|
|
'works_at_weekday': {
|
|
'field': 'works_at_weekday',
|
|
'lookups': [
|
|
constants.LOOKUP_QUERY_IN,
|
|
],
|
|
},
|
|
'works_evening': {
|
|
'field': 'works_evening',
|
|
'lookups': [
|
|
constants.LOOKUP_QUERY_IN,
|
|
],
|
|
},
|
|
'works_now': {
|
|
'field': 'works_now',
|
|
'lookups': [
|
|
constants.LOOKUP_FILTER_TERM,
|
|
]
|
|
},
|
|
}
|
|
|
|
|
|
# BackOffice user`s views & viewsets
|
|
class BindObjectMixin:
|
|
"""Bind object mixin."""
|
|
|
|
def get_serializer_class(self):
|
|
if self.action == 'bind_object':
|
|
return self.bind_object_serializer_class
|
|
return self.serializer_class
|
|
|
|
def perform_binding(self, serializer):
|
|
raise NotImplemented
|
|
|
|
def perform_unbinding(self, serializer):
|
|
raise NotImplemented
|
|
|
|
@action(methods=['post', 'delete'], detail=True, url_path='bind-object')
|
|
def bind_object(self, request, pk=None):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
if request.method == 'POST':
|
|
self.perform_binding(serializer)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
elif request.method == 'DELETE':
|
|
self.perform_unbinding(serializer)
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class TagBackOfficeViewSet(mixins.ListModelMixin, mixins.CreateModelMixin,
|
|
mixins.UpdateModelMixin, mixins.DestroyModelMixin,
|
|
BindObjectMixin, viewsets.GenericViewSet):
|
|
"""List/create tag view."""
|
|
|
|
pagination_class = None
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
queryset = models.Tag.objects.all()
|
|
serializer_class = serializers.TagBackOfficeSerializer
|
|
bind_object_serializer_class = serializers.TagBindObjectSerializer
|
|
|
|
def perform_binding(self, serializer):
|
|
data = serializer.validated_data
|
|
tag = data.pop('tag')
|
|
obj_type = data.get('type')
|
|
related_object = data.get('related_object')
|
|
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT:
|
|
tag.establishments.add(related_object)
|
|
elif obj_type == self.bind_object_serializer_class.NEWS:
|
|
tag.news.add(related_object)
|
|
|
|
def perform_unbinding(self, serializer):
|
|
data = serializer.validated_data
|
|
tag = data.pop('tag')
|
|
obj_type = data.get('type')
|
|
related_object = data.get('related_object')
|
|
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT:
|
|
tag.establishments.remove(related_object)
|
|
elif obj_type == self.bind_object_serializer_class.NEWS:
|
|
tag.news.remove(related_object)
|
|
|
|
|
|
class TagCategoryBackOfficeViewSet(mixins.CreateModelMixin,
|
|
mixins.UpdateModelMixin,
|
|
mixins.DestroyModelMixin,
|
|
mixins.RetrieveModelMixin,
|
|
BindObjectMixin,
|
|
TagCategoryViewSet):
|
|
"""ViewSet for TagCategory model for BackOffice users."""
|
|
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
queryset = TagCategoryViewSet.queryset.with_extended_related()
|
|
serializer_class = serializers.TagCategoryBackOfficeDetailSerializer
|
|
bind_object_serializer_class = serializers.TagCategoryBindObjectSerializer
|
|
|
|
def perform_binding(self, serializer):
|
|
data = serializer.validated_data
|
|
tag_category = data.pop('tag_category')
|
|
obj_type = data.get('type')
|
|
related_object = data.get('related_object')
|
|
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT_TYPE:
|
|
tag_category.establishment_types.add(related_object)
|
|
elif obj_type == self.bind_object_serializer_class.NEWS_TYPE:
|
|
tag_category.news_types.add(related_object)
|
|
|
|
def perform_unbinding(self, serializer):
|
|
data = serializer.validated_data
|
|
tag_category = data.pop('tag_category')
|
|
obj_type = data.get('type')
|
|
related_object = data.get('related_object')
|
|
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT_TYPE:
|
|
tag_category.establishment_types.remove(related_object)
|
|
elif obj_type == self.bind_object_serializer_class.NEWS_TYPE:
|
|
tag_category.news_types.remove(related_object)
|