Fix establishment retrieve view

This commit is contained in:
evgeniy-st 2019-09-01 14:53:46 +03:00
parent abecc6f99a
commit ddabac60d1

View File

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