"""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