phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change api view method

Browse Source
This commit is contained in:
dormantman 2020-01-27 22:22:17 +03:00
parent d51d304077
commit bb38bb3393
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
apps/establishment/views/back.py
View File

@ -1,5 +1,4 @@
"""Establishment app views.""" """Establishment app views."""
from collections import defaultdict
from django.shortcuts import get_object_or_404 from django.shortcuts import get_object_or_404
from django_filters.rest_framework import DjangoFilterBackend from django_filters.rest_framework import DjangoFilterBackend
@ -33,21 +32,17 @@ class EstablishmentListCreateView(EstablishmentMixinViews, generics.ListCreateAP
serializer_class = serializers.EstablishmentListCreateSerializer serializer_class = serializers.EstablishmentListCreateSerializer
class EmployeeEstablishmentPositionsView(generics.GenericAPIView): class EmployeeEstablishmentPositionsView(generics.ListAPIView):
"""Establishment by employee view.""" """Establishment by employee view."""
permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager] permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
queryset = models.EstablishmentEmployee.objects.all() queryset = models.EstablishmentEmployee.objects.all()
serializer_class = serializers.EstablishmentEmployeePositionsSerializer serializer_class = serializers.EstablishmentEmployeePositionsSerializer
def get(self, request, *args, **kwargs): def get_queryset(self):
employee_pk = self.kwargs.get('pk') employee_pk = self.kwargs.get('pk')
return super().get_queryset().filter(employee__id=employee_pk).all().prefetch_related(
return Response(self.get_serializer( 'establishment').select_related('position')
self.get_queryset().filter(employee__id=employee_pk).all()
.prefetch_related('establishment').select_related('position'),
many=True
).data)
class EmployeeEstablishmentsListView(generics.ListAPIView): class EmployeeEstablishmentsListView(generics.ListAPIView):