From 1f22139a7a246ee0221f7b693a37dcd7a4c150d4 Mon Sep 17 00:00:00 2001
From: Hrishabh Tiwari <74908943+Hrishabh17@users.noreply.github.com>
Date: Mon, 15 Jul 2024 09:56:39 +0530
Subject: [PATCH] fix created_at, updated_at null constraint (#149)

* fix created_at, updated_at null constraint

* fix expense fetching

* fix null refresh token
---
 apps/fyle/models.py         | 6 +++---
 apps/fyle/views.py          | 8 ++++++--
 apps/workspaces/tasks.py    | 2 +-
 tests/test_fyle/fixtures.py | 4 ++--
 4 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/apps/fyle/models.py b/apps/fyle/models.py
index 56d1e49..5e362ee 100644
--- a/apps/fyle/models.py
+++ b/apps/fyle/models.py
@@ -116,7 +116,9 @@ def create_expense_objects(expenses: List[Dict], workspace_id: int, skip_update:
                     'claim_number': expense['claim_number'],
                     'report_title': expense['report_title'],
                     'approved_at': expense['approved_at'],
-                    'payment_number': expense['payment_number']
+                    'payment_number': expense['payment_number'],
+                    'expense_created_at': expense['expense_created_at'],
+                    'expense_updated_at': expense['expense_updated_at']
                 }
 
             defaults = {
@@ -142,8 +144,6 @@ def create_expense_objects(expenses: List[Dict], workspace_id: int, skip_update:
                 'report_id': expense['report_id'],
                 'spent_at': expense['spent_at'],
                 'posted_at': expense['posted_at'],
-                'expense_created_at': expense['expense_created_at'],
-                'expense_updated_at': expense['expense_updated_at'],
                 'fund_source': SOURCE_ACCOUNT_MAP[expense['source_account_type']],
                 'verified_at': expense['verified_at'],
                 'custom_properties': expense['custom_properties'],
diff --git a/apps/fyle/views.py b/apps/fyle/views.py
index 6f1302a..091b9de 100644
--- a/apps/fyle/views.py
+++ b/apps/fyle/views.py
@@ -19,6 +19,7 @@
     ImportFyleAttributesSerializer,
 )
 from apps.accounting_exports.helpers import ExpenseSearchFilter
+from apps.workspaces.models import ExportSetting
 
 from ms_business_central_api.utils import LookupFieldMixin
 
@@ -97,9 +98,12 @@ def post(self, request, *args, **kwargs):
         """
         Post expense groups creation
         """
+        export_settings = ExportSetting.objects.get(workspace_id=kwargs['workspace_id'])
 
-        queue_import_reimbursable_expenses(kwargs['workspace_id'], synchronous=True)
-        queue_import_credit_card_expenses(kwargs['workspace_id'], synchronous=True)
+        if export_settings.reimbursable_expenses_export_type:
+            queue_import_reimbursable_expenses(kwargs['workspace_id'], synchronous=True)
+        if export_settings.credit_card_expense_export_type:
+            queue_import_credit_card_expenses(kwargs['workspace_id'], synchronous=True)
 
         return Response(
             status=status.HTTP_200_OK
diff --git a/apps/workspaces/tasks.py b/apps/workspaces/tasks.py
index 4412d58..d34b943 100644
--- a/apps/workspaces/tasks.py
+++ b/apps/workspaces/tasks.py
@@ -19,7 +19,7 @@
 
 def async_update_fyle_credentials(fyle_org_id: str, refresh_token: str):
     fyle_credentials = FyleCredential.objects.filter(workspace__org_id=fyle_org_id).first()
-    if fyle_credentials:
+    if fyle_credentials and refresh_token:
         fyle_credentials.refresh_token = refresh_token
         fyle_credentials.save()
 
diff --git a/tests/test_fyle/fixtures.py b/tests/test_fyle/fixtures.py
index 80d7861..70c87ae 100644
--- a/tests/test_fyle/fixtures.py
+++ b/tests/test_fyle/fixtures.py
@@ -43,7 +43,7 @@
             'name': 'Administration'
         },
         'cost_center_id': 23166,
-        'created_at': '2024-05-10T07:52:10.551260+00:00',
+        'created_at': None,
         'creator_user_id': 'usVN2WTtPqE7',
         'currency': 'USD',
         'custom_fields': [
@@ -179,7 +179,7 @@
         'tax_group': None,
         'tax_group_id': None,
         'travel_classes': [],
-        'updated_at': '2024-06-10T11:41:40.779611+00:00',
+        'updated_at': None,
         'user': {
             'email': 'admin1@fyleforimporrttest.in',
             'full_name': 'Theresa Brown',