Fix establishment retrieve view
This commit is contained in:
parent
abecc6f99a
commit
ddabac60d1
|
|
@ -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):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user