Handle refresh dimension safely for new customers (#548)

fylein · Dec 27, 2023 · f77a41a · f77a41a
1 parent ab01a95
commit f77a41a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apps/quickbooks_online/actions.py b/apps/quickbooks_online/actions.py
@@ -67,12 +67,12 @@ def refresh_quickbooks_dimensions(workspace_id: int):
 
     mapping_settings = MappingSetting.objects.filter(workspace_id=workspace_id, import_to_fyle=True)
     credentials = QBOCredential.objects.get(workspace_id=workspace_id)
-    workspace_general_settings = WorkspaceGeneralSettings.objects.get(workspace_id=workspace_id)
+    workspace_general_settings = WorkspaceGeneralSettings.objects.filter(workspace_id=workspace_id).first()
 
     chain = Chain()
 
     for mapping_setting in mapping_settings:
-        if mapping_setting.source_field in ['PROJECT', 'COST_CENTER'] or mapping_setting.is_custom:
+        if workspace_general_settings and mapping_setting.source_field in ['PROJECT', 'COST_CENTER'] or mapping_setting.is_custom:
             chain.append(
                 'fyle_integrations_imports.tasks.trigger_import_via_schedule',
                 workspace_id,