diff --git a/apps/establishment/views.py b/apps/establishment/views.py index 525118e2..10f47b01 100644 --- a/apps/establishment/views.py +++ b/apps/establishment/views.py @@ -5,24 +5,25 @@ from utils.views import JWTGenericViewMixin from establishment import filters -class EstablishmentListView(JWTGenericViewMixin, generics.ListAPIView): - """Resource for getting a list of establishments.""" +class EstablishmentMixin: + """Establishment mixin.""" permission_classes = (permissions.AllowAny,) serializer_class = serializers.EstablishmentSerializer - filter_class = filters.EstablishmentFilter def get_queryset(self): """Overrided method 'get_queryset'.""" return models.Establishment.objects.all().prefetch_actual_employees() -class EstablishmentRetrieveView(JWTGenericViewMixin, generics.RetrieveAPIView): - """Resource for getting a establishment.""" +class EstablishmentListView(EstablishmentMixin, JWTGenericViewMixin, generics.ListAPIView): + """Resource for getting a list of establishments.""" - permission_classes = (permissions.AllowAny,) - serializer_class = serializers.EstablishmentSerializer - queryset = models.Establishment.objects.all() + filter_class = filters.EstablishmentFilter + + +class EstablishmentRetrieveView(EstablishmentMixin, JWTGenericViewMixin, generics.RetrieveAPIView): + """Resource for getting a establishment.""" class EstablishmentTypeListView(JWTGenericViewMixin, generics.ListAPIView):