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

fix

Browse Source
This commit is contained in:
Kuroshini 2020-01-27 21:18:48 +03:00
parent 095e482358
commit 415c9b7851
1 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -81,21 +81,10 @@ class EmployeePositionsListView(generics.ListAPIView):
queryset = models.Establishment.objects.all()
serializer_class = serializers.EstablishmentPositionListSerializer
def list(self, request, *args, **kwargs):
def get_queryset(self):
pk = self.kwargs.get('pk')
employee = get_object_or_404(models.Employee, pk=pk)
queryset = employee.establishments.with_extended_related()
queryset = self.filter_queryset(queryset)
page = self.paginate_queryset(queryset)
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)
serializer = self.get_serializer(queryset, many=True)
return Response(serializer.data)
return employee.establishments.with_extended_related()
class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView):