gault-millau/apps/establishment/views.py
2019-08-27 12:33:16 +03:00

35 lines
1.2 KiB
Python

"""Establishment app views."""
from rest_framework import generics, permissions
from establishment import models, serializers
from utils.views import JWTGenericViewMixin
class EstablishmentListView(JWTGenericViewMixin, generics.ListAPIView):
"""Resource for getting a list of establishments."""
pagination_class = None
permission_classes = (permissions.AllowAny,)
serializer_class = serializers.EstablishmentSerializer
def get_queryset(self):
return models.Establishment.objects.annotate_localized_fields(
locale=self.request.locale)
class EstablishmentTypeListView(generics.ListAPIView):
"""Resource for getting a list of establishment types."""
pagination_class = None
permission_classes = (permissions.AllowAny,)
queryset = models.EstablishmentType.objects.all()
serializer_class = serializers.EstablishmentTypeSerializer
class EstablishmentSubTypeListView(generics.ListAPIView):
"""Resource for getting a list of establishment subtypes."""
pagination_class = None
permission_classes = (permissions.AllowAny,)
queryset = models.EstablishmentType.objects.all()
serializer_class = serializers.EstablishmentSubTypeSerializer