Update and Create Employee on Webhook trigger from bamboohr

apps/bamboohr/migrations/0007_bamboohr_webhook_id.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.1.14 on 2024-01-02 16:10
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('bamboohr', '0006_bamboohr_is_credentials_expired'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='bamboohr',
+            name='webhook_id',
+            field=models.IntegerField(help_text='ID of the webhook created by BambooHr', null=True),
+        ),
+    ]

apps/bamboohr/models.py

@@ -19,6 +19,7 @@ class BambooHr(models.Model):
     updated_at =  models.DateTimeField(auto_now=True, help_text='Updated at datetime')
     employee_exported_at = models.DateTimeField(auto_now_add=True, help_text='Employee exported to Fyle at datetime')
     is_credentials_expired = models.BooleanField(default=False, help_text='BambooHr Credential Status')
+    webhook_id = models.IntegerField(null=True, help_text='ID of the webhook created by BambooHr')
 
     class Meta:
         db_table = 'bamboohr'

apps/bamboohr/signals.py

@@ -1,13 +1,46 @@
 
-from django.db.models.signals import pre_save
+from django.db.models.signals import pre_save, post_save, pre_delete
 from django.dispatch import receiver
+from admin_settings.settings import API_URL
 
 from workato import Workato
-from apps.bamboohr.models import BambooHrConfiguration
+
+from bamboosdk.bamboohrsdk import BambooHrSDK
+from apps.bamboohr.models import BambooHrConfiguration, BambooHr
 from apps.orgs.models import Org
 
 @receiver(pre_save, sender=BambooHrConfiguration)
 def run_pre_save_configuration_triggers(sender, instance: BambooHrConfiguration, **kwargs):
     connector = Workato()
     org = Org.objects.get(id=instance.org_id)
     connector.recipes.post(org.managed_user_id, instance.recipe_id, None, 'stop')
+
+
+@receiver(post_save, sender=BambooHr)
+def run_post_save_bamboohr_triggers(sender, instance: BambooHr, **kwargs):
+
+    # Disconnect the signal to avoid triggering it again
+    post_save.disconnect(run_post_save_bamboohr_triggers, sender=BambooHr)
+
+    bamboohrsdk = BambooHrSDK(api_token=instance.api_token, sub_domain=instance.sub_domain)
+
+    webhook_payload = {
+        'postFields': {
+            'firstName': 'firstName',
+            'lastName': 'lastName',
+            'department': 'department',
+            'workEmail': 'workEmail',
+            'status': 'status'
+        },
+        'name': instance.org.name,
+        'monitorFields': ['firstName', 'lastName', 'department', 'workEmail', 'status'],
+        'url': API_URL + f'/orgs/{instance.org.id}/bamboohr/webhook_callback/',
+        'format': 'json'
+    }
+
+    response = bamboohrsdk.webhook.post(payload=webhook_payload)
+    instance.webhook_id = int(response['id'])
+    instance.save()
+
+    # Reconnect the signal after saving
+    post_save.connect(run_post_save_bamboohr_triggers, sender=BambooHr)

apps/bamboohr/tasks.py

@@ -10,3 +10,31 @@ def refresh_employees(org_id: int, user: User):
     bambooHrImporter = BambooHrEmployeeImport(org_id=org_id, user=user)
     bambooHrImporter.sync_employees()
 
+def update_employee(org_id: int, user: User, payload: dict):
+
+    """
+        Update employee in fyle when employee in Bamboohr is added or updated
+    """
+    bamboohr = BambooHr.objects.get(org_id__in=[org_id], is_credentials_expired=False)
+    bamboohr_importer = BambooHrEmployeeImport(org_id=org_id, user=user)
+
+    employee_payload = {'employees': []}
+    payload = payload['employees'][0]
+    employee = {}
+    employee['id'] = payload['id']
+    employee['firstName'] = payload['fields']['firstName']['value']
+    employee['lastName'] = payload['fields']['lastName']['value']
+    for field in payload['changedFields']:
+        employee[field] = payload['fields'][field]['value']
+
+    employee_payload['employees'].append(employee)
+
+    bamboohr_importer.upsert_employees(employees=employee_payload)
+
+    hrms_employees = DestinationAttribute.objects.filter(
+            attribute_type='EMPLOYEE',
+            org_id=org_id,
+            updated_at__gte=bamboohr.employee_exported_at,
+        ).order_by('value', 'id')
+    bamboohr_importer.import_departments(hrms_employees=hrms_employees)
+    bamboohr_importer.fyle_employee_import(hrms_employees=hrms_employees)

apps/bamboohr/urls.py

@@ -1,11 +1,12 @@
 from django.urls import path
 
 from .views import PostFolder, PostPackage, BambooHrConnection, BambooHrView, BambooHrConfigurationView, \
-    DisconnectView, SyncEmployeesView, HealthCheck
+    DisconnectView, SyncEmployeesView, HealthCheck, WebhookAPIView
 
 app_name = 'bamboohr'
 
 urlpatterns = [
+    path('webhook_callback/', WebhookAPIView.as_view(), name='webhook-callback'),
     path('health_check/', HealthCheck.as_view(), name='health-check'),
     path('', BambooHrView.as_view(), name='bamboohr'),
     path('packages/', PostPackage.as_view(), name='package'),

apps/bamboohr/views.py

@@ -48,6 +48,24 @@ def get(self, request, *args, **kwargs):
                 status=status.HTTP_404_NOT_FOUND
             )
 
+
+class WebhookAPIView(generics.CreateAPIView):
+
+    def post(self, request, *args, **kwargs):
+
+        org_id = kwargs['org_id']
+        user = self.request.user
+        payload = request.data
+
+        async_task('apps.bamboohr.tasks.update_employee', org_id, user, payload)
+
+        return Response(
+            {
+                'status': 'success'
+            },
+            status=status.HTTP_201_CREATED
+        )
+
 class BambooHrView(generics.ListAPIView):
     serializer_class = BambooHrSerializer
 
@@ -195,17 +213,12 @@ class DisconnectView(generics.CreateAPIView):
 
     def post(self, request, *args, **kwargs):
         try:
-            configuration = BambooHrConfiguration.objects.get(org__id=kwargs['org_id'])
             bamboohr = BambooHr.objects.filter(org__id=kwargs['org_id']).first()
-
-            connection = disconnect_bamboohr(kwargs['org_id'], configuration, bamboohr)
-
-            # in case of an error response
-            if isinstance(connection, Response):
-                return connection
-
+            webhook_id = request.data['webhook_id']
+            bambamboohrsdk = BambooHrSDK(api_token=bamboohr.api_token, sub_domain=bamboohr.sub_domain)
+            bambamboohrsdk.webhook.delete(id=webhook_id)
             return Response(
-                data=connection,
+                data='Succesfully disconnected Bamboohr',
                 status=status.HTTP_200_OK
             )
         except BambooHr.DoesNotExist:
@@ -215,11 +228,6 @@ def post(self, request, *args, **kwargs):
                 },
                 status = status.HTTP_404_NOT_FOUND
             )
-        except BambooHrConfiguration.DoesNotExist:
-            return Response(
-                data={'message': 'BambooHr Configuration does not exist for this Workspace'},
-                status=status.HTTP_404_NOT_FOUND
-            )
 
 
 class SyncEmployeesView(generics.UpdateAPIView):

fyle_employee_imports/bamboo_hr.py

@@ -19,18 +19,23 @@ def sync_hrms_employees(self):
     def upsert_employees(self, employees: Dict):
         attributes = []
         for employee in employees['employees']:
-            supervisor = [employee['supervisorEmail']]
-            active_status = True if employee['status'] == 'Active' else False
+            supervisor = [employee.get('supervisorEmail', None)]
+            active_status = True if employee.get('status', None) == 'Active' else False
+
+            display_name = employee.get('displayName', None)
+            if not display_name:
+                display_name = employee['firstName'] + ' ' + employee['lastName']
+
             detail = {
-                'email': employee['workEmail'] if employee['workEmail'] else None,
-                'department_name': employee['department'] if employee['department'] else None,
-                'full_name': employee['displayName'] if employee['displayName'] else None,
+                'email': employee.get('workEmail', None),
+                'department_name': employee.get('department', None),
+                'full_name': display_name,
                 'approver_emails': supervisor,
             }
 
             attributes.append({
                 'attribute_type': 'EMPLOYEE',
-                'value': employee['displayName'],
+                'value': display_name,
                 'destination_id': employee['id'],
                 'detail': detail,
                 'active': active_status

fyle_employee_imports/base.py

@@ -115,7 +115,6 @@ def get_employee_and_approver_payload(self, hrms_employees):
 
     def fyle_employee_import(self, hrms_employees):
         fyle_employee_payload, employee_approver_payload = self.get_employee_and_approver_payload(hrms_employees)
-
         if fyle_employee_payload:
             self.platform_connection.bulk_post_employees(employees_payload=fyle_employee_payload)
 
@@ -141,6 +140,6 @@ def sync_employees(self):
             org_id=self.org_id,
             updated_at__gte=self.bamboohr.employee_exported_at,
         ).order_by('value', 'id')
-
+        
         self.import_departments(hrms_employees)
         self.fyle_employee_import(hrms_employees)