From c8d2884a48731802d9db48c7d776736f657c85fb Mon Sep 17 00:00:00 2001
From: Kuroshini <o.hayatov@id-east.ru>
Date: Thu, 6 Feb 2020 15:17:21 +0300
Subject: [PATCH] filter employees by country for BO

---
 apps/establishment/views/back.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/apps/establishment/views/back.py b/apps/establishment/views/back.py
index 921470ad..b81db02e 100644
--- a/apps/establishment/views/back.py
+++ b/apps/establishment/views/back.py
@@ -608,6 +608,12 @@ class EmployeeListCreateView(generics.ListCreateAPIView):
         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):
     """