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

filter employees by country for BO

Browse Source
This commit is contained in:
Kuroshini 2020-02-06 15:17:21 +03:00
parent 0104400dbd
commit c8d2884a48
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -608,6 +608,12 @@ class EmployeeListCreateView(generics.ListCreateAPIView):
IsEstablishmentAdministrator, IsEstablishmentAdministrator,
) )
def get_queryset(self):
qs = super().get_queryset()
if self.request.country_code:
qs = qs.filter(establishments__address__city__country__code=self.request.country_code)
return qs
class EmployeesListSearchViews(generics.ListAPIView): class EmployeesListSearchViews(generics.ListAPIView):
""" """