Merge branch 'refresh_employees' into configuration

apps/bamboohr/tasks.py

@@ -3,8 +3,10 @@
 from fyle_employee_imports.bamboo_hr import BambooHrEmployeeImport
 from apps.users.models import User
 
-def refresh_employees(org_id, user):
-
+def refresh_employees(org_id: int, user: User):
+    """
+        Sync Employees from BambooHR to Fyle
+    """
     bambooHrImporter = BambooHrEmployeeImport(org_id=org_id, user=user)
     bambooHrImporter.sync_employees()
 

apps/bamboohr/views.py

@@ -248,7 +248,7 @@ class SyncEmployeesView(generics.UpdateAPIView):
 
     def post(self, request, *args, **kwargs):
 
-        async_task('apps.bamboohr.tasks.refresh_employees')
+        async_task('apps.bamboohr.tasks.refresh_employees', kwargs['org_id'], self.request.user)
 
         return Response(
             data = {'message': 'success'},

fyle_employee_imports/bamboo_hr.py

@@ -10,8 +10,7 @@ class BambooHrEmployeeImport(FyleEmployeeImport):
 
     def __init__(self, org_id: int, user: User):
         super().__init__(org_id, user)
-        bamboo_hr = BambooHr.objects.get(org_id__in=org_id)
-        self.bamboohr_sdk = BambooHrSDK(api_token=bamboo_hr.api_token, sub_domain=bamboo_hr.sub_domain)
+        self.bamboohr_sdk = BambooHrSDK(api_token=self.bamboohr.api_token, sub_domain=self.bamboohr.sub_domain)
 
     def sync_hrms_employees(self):
         employees = self.bamboohr_sdk.employees.get_all()

fyle_employee_imports/base.py

@@ -12,7 +12,7 @@ class FyleEmployeeImport():
     def __init__(self, org_id: int, user):
         self.org_id = org_id
         self.user = user
-        self.bamboohr = BambooHr.objects.get(org_id__in=self.org_id)
+        self.bamboohr = BambooHr.objects.get(org_id__in=[self.org_id])
         refresh_token = AuthToken.objects.get(user__user_id=self.user).refresh_token
         cluster_domain = Org.objects.get(user__user_id=self.user).cluster_domain
         self.platform_connection = PlatformConnector(refresh_token, cluster_domain)