From 0688c622cc795a088dc398ea26e2a86f2a8a7306 Mon Sep 17 00:00:00 2001
From: dormantman <dormantman@ya.ru>
Date: Tue, 21 Jan 2020 19:28:29 +0300
Subject: [PATCH] Fixed structure view

---
 apps/establishment/views/back.py | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/apps/establishment/views/back.py b/apps/establishment/views/back.py
index a19b2a95..4bbf9533 100644
--- 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):