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

    permission_classes = (permissions.AllowAny,)
    serializer_class = serializers.EstablishmentSerializer
    queryset = models.Establishment.objects.all()


class EstablishmentTypeListView(JWTGenericViewMixin, generics.ListAPIView):
    """Resource for getting a list of establishment types."""

    permission_classes = (permissions.AllowAny,)
    serializer_class = serializers.EstablishmentTypeSerializer
    queryset = models.EstablishmentType.objects.all()