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

Fixed structure view

Browse Source
This commit is contained in:
dormantman 2020-01-21 19:28:29 +03:00
parent 88e488e481
commit 0688c622cc
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -41,14 +41,8 @@ class EmployeeEstablishmentsListView(generics.ListAPIView):
def get_queryset(self): def get_queryset(self):
pk = self.kwargs.get('pk') pk = self.kwargs.get('pk')
employee = get_object_or_404(models.Employee, pk=pk)
try: return employee.establishments.with_extended_related()
employee = models.Employee.objects.get(pk=pk)
except ObjectDoesNotExist:
raise Http404
return employee.establishments.all()
class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView): class EstablishmentRUDView(generics.RetrieveUpdateDestroyAPIView):