Fixed structure view

2020-01-21 19:28:29 +03:00 · 2020-01-21 19:28:29 +03:00 · 0688c622cc
commit 0688c622cc
parent 88e488e481
1 changed files with 2 additions and 8 deletions
--- a/apps/establishment/views/back.py
+++ b/apps/establishment/views/back.py
@ -41,14 +41,8 @@ class EmployeeEstablishmentsListView(generics.ListAPIView):

    def get_queryset(self):
        pk = self.kwargs.get('pk')
-
-        try:
-            employee = models.Employee.objects.get(pk=pk)
-
-        except ObjectDoesNotExist:
-            raise Http404
-
-        return employee.establishments.all()
+        employee = get_object_or_404(models.Employee, pk=pk)
+        return employee.establishments.with_extended_related()


 class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView):