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):