create webhook subscription after connection is complete

apps/travelperk/connector.py

@@ -2,8 +2,8 @@
 from django.conf import settings
 
 
-from apps.travelperk.models import TravelperkCredential
-from connectors.travelperk.core.client import Travelperk
+from apps.travelperk.models import TravelperkCredential, TravelPerk
+from connectors.travelperk import Travelperk
 
 
 logger = logging.getLogger(__name__)
@@ -26,3 +26,23 @@ def __init__(self, credentials_object: TravelperkCredential, org_id: int):
 
         credentials_object.refresh_token = self.connection.refresh_token
         credentials_object.save()
+
+
+    def create_webhook(self, data: dict):
+        """
+        Create Webhook in Travelperk
+        :param data: Webhook Data
+        :return: Webhook Data
+        """
+
+        response = self.connection.webhooks.create(data)
+        if response:
+            TravelPerk.objects.update_or_create(
+                org_id=self.org_id,
+                defaults={
+                    'webhook_id': response['id'],
+                    'webhook_enabled': response['enabled']
+                }
+            )
+
+        return response

apps/travelperk/migrations/0006_auto_20231214_1927.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.1.14 on 2023-12-14 19:27
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('travelperk', '0005_invoice_invoicelineitem_travelperkcredential'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='travelperk',
+            name='webhook_enabled',
+            field=models.BooleanField(help_text='If Webhook Is Enabled', null=True),
+        ),
+        migrations.AddField(
+            model_name='travelperk',
+            name='webhook_subscription_id',
+            field=models.CharField(help_text='Webhook Subscription Id', max_length=255, null=True),
+        ),
+    ]

apps/travelperk/models.py

@@ -19,8 +19,6 @@ class Meta:
         db_table = 'travelperk_credentials'
 
 
-from django.db import models
-
 class InvoiceLineItem(models.Model):
     """
     Travelperk Invoice Line Item Model
@@ -103,6 +101,8 @@ class TravelPerk(models.Model):
     is_s3_connected = models.BooleanField(null=True, help_text='If S3 Is Connectoed')
     is_travelperk_connected = models.BooleanField(null=True, help_text='If Travelperk Is Connected')
     travelperk_connection_id = models.IntegerField(null=True, help_text='Travelperk Connection Id')
+    webhook_subscription_id = models.CharField(max_length=255, null=True, help_text='Webhook Subscription Id')
+    webhook_enabled = models.BooleanField(null=True, help_text='If Webhook Is Enabled')
     created_at = models.DateTimeField(auto_now_add=True, help_text='Created at datetime')
     updated_at =  models.DateTimeField(auto_now=True, help_text='Updated at datetime')
 

apps/travelperk/views.py

@@ -12,8 +12,9 @@
 from apps.orgs.models import Org
 from apps.orgs.actions import create_connection_in_workato, upload_properties, post_folder, post_package
 from apps.travelperk.serializers import TravelperkSerializer, TravelPerkConfigurationSerializer
-from apps.travelperk.models import TravelPerk, TravelPerkConfiguration
+from apps.travelperk.models import TravelPerk, TravelPerkConfiguration, TravelperkCredential
 from apps.travelperk.actions import connect_travelperk
+from apps.travelperk.connector import TravelperkConnector
 
 from .helpers import get_refresh_token_using_auth_code
 
@@ -219,53 +220,39 @@ class ConnectTravelperkView(generics.CreateAPIView):
     """
 
     def post(self, request, *args, **kwargs):
+
         try:
-            connector = Workato()
             org = Org.objects.get(id=kwargs['org_id'])
-            travelperk = TravelPerk.objects.get(org_id=org.id)
-
             refresh_token = get_refresh_token_using_auth_code(request.data.get('code'), kwargs['org_id'])
-
-            properties_payload = {
-                'properties': {
-                    'TRAVELPERK_REFRESH_TOKEN': refresh_token
-                }
-            }
-
-            data = {
-                "input": {
-                    "key": "***"
+
+            if refresh_token:
+                travelperk_credential = TravelperkCredential.objects.get(org=org)
+                travelperk_connection = TravelperkConnector(travelperk_credential, kwargs['org_id'])
+
+                travelperk_webhook_data = {
+                    'name': 'travelperk webhook invoice',
+                    'url': 'https://webhook.site/e4ac2898-209f-4dd2-88cf-738dea95ecdc',
+                    'secret': 'some secret',
+                    'events': [
+                        'invoice.issued'
+                    ]
                 }
-            }
 
-            upload_properties(org.managed_user_id, properties_payload)
-
-            travelperk_connection = create_connection_in_workato(org.id, TRAVELPERK['connection'], org.managed_user_id, data)
-            if 'authorization_status' in travelperk_connection and travelperk_connection['authorization_status'] == 'success':
-                travelperk.is_connected = True
-                travelperk.save()
-
-                recipes = connector.recipes.get(org.managed_user_id)['result']
-                travelperk_configuration, _ =  TravelPerkConfiguration.objects.update_or_create(
-                    org_id=org.id,
-                    recipe_id=recipes[0]['id'],
+                created_webhook = travelperk_connection.create_webhook(travelperk_webhook_data)
+                print('created_webhook', created_webhook)
+                TravelPerk.objects.update_or_create(
+                    org=org,
                     defaults={
-                        'recipe_data': recipes[0]['code'],
-                        'is_recipe_enabled': True
+                        'webhook_id': created_webhook['id'],
                     }
                 )
-                connector.recipes.post(org.managed_user_id, travelperk_configuration.recipe_id, None, 'start')
+
                 return Response(
-                    data=travelperk_connection,
+                    data=created_webhook,
                     status=status.HTTP_200_OK
                 )
 
-            return Response(
-                data='Something went wrong while connecting to travelperk',
-                status=status.HTTP_400_BAD_REQUEST
-            )
-
-        except Exception as exception:
+        except Exception:
             error = traceback.format_exc()
             logger.error(error)
 

connectors/travelperk/apis/api_base.py

@@ -135,8 +135,7 @@ def _delete_request(self, api_url: str) -> Dict:
         )
 
         if response.status_code == 200:
-            result = json.loads(response.text)
-            return result
+            return response.text
 
         elif response.status_code == 400:
             raise BadRequestError('Something wrong with the request body', response.text)

tests/test_travelperk/fixtures.py

@@ -9,6 +9,8 @@
        "is_s3_connected":"dtmmy",
        "created_at":"2022-11-29T15:39:49.221955Z",
        "updated_at":"2022-11-29T15:41:59.535831Z",
+       "webhook_subscription_id": "123",
+       "webhook_enabled": True,
        "org":2
     },
     "configurations": {