Feat: Check entity slide issue

apps/sage_intacct/connector.py

@@ -1291,7 +1291,8 @@ def __get_entity_slide_preference(self) -> bool:
             entity_slide_disabled = entity_slide_info['prefvalue'] == 'true'
         except Exception as e:
             logger.error(e.__dict__)
-
+            # TODO: Check by fetching locations scoped to one of the entities
+            # check the function check_entity_slide_error
         return entity_slide_disabled
 
     def sync_location_entities(self) -> None:

apps/sage_intacct/helpers.py

@@ -6,6 +6,10 @@
 from django_q.models import Schedule
 from django.core.cache import cache
 
+from cryptography.fernet import Fernet
+from intacctsdk import IntacctRESTSDK
+from sageintacctsdk import SageIntacctSDK
+
 from apps.workspaces.enums import CacheKeyEnum
 from apps.fyle.models import DependentFieldSetting
 from apps.sage_intacct.utils import SageIntacctConnector
@@ -167,3 +171,42 @@ def validate_rest_api_connection(workspace_id: int) -> None:
         sync_dimensions(workspace_id=workspace_id)
     except Exception as e:
         logger.info('REST API is not working for workspace_id - %s, error - %s', workspace_id, e)
+
+
+def check_entity_slide_error(workspace_id: int, entity_id: str) -> None:
+    """
+    Check if a location entity causes an entity slide error by attempting
+    to fetch locations scoped to that entity. Raises an exception if the
+    Sage Intacct API rejects the request (e.g. due to entity slide restrictions).
+    :param workspace_id: Workspace ID
+    :param entity_id: Location entity destination ID to validate
+    """
+    migrated_to_rest_api = FeatureConfig.get_feature_config(workspace_id=workspace_id, key='migrated_to_rest_api')
+    entity_id_for_sdk = entity_id if entity_id != 'top_level' else None
+
+    if migrated_to_rest_api:
+        rest_connector = SageIntacctRestConnector(workspace_id=workspace_id)
+        connection = IntacctRESTSDK(
+            username=rest_connector.username,
+            access_token=rest_connector.connection.access_token,
+            entity_id=entity_id_for_sdk,
+            client_id=settings.INTACCT_CLIENT_ID,
+            client_secret=settings.INTACCT_CLIENT_SECRET
+        )
+    else:
+        sage_intacct_credentials = SageIntacctCredential.get_active_sage_intacct_credentials(workspace_id)
+        cipher_suite = Fernet(settings.ENCRYPTION_KEY)
+        decrypted_password = cipher_suite.decrypt(
+            sage_intacct_credentials.si_user_password.encode('utf-8')
+        ).decode('utf-8')
+
+        connection = SageIntacctSDK(
+            sender_id=settings.SI_SENDER_ID,
+            sender_password=settings.SI_SENDER_PASSWORD,
+            user_id=sage_intacct_credentials.si_user_id,
+            company_id=sage_intacct_credentials.si_company_id,
+            user_password=decrypted_password,
+            entity_id=entity_id_for_sdk
+        )
+
+    connection.locations.count()

apps/sage_intacct/urls.py

@@ -5,6 +5,7 @@
 from apps.sage_intacct.views import (
     AuthorizationCodeView,
     DestinationAttributesView,
+    EntitySlideCheckView,
     PaginatedDestinationAttributesView,
     RefreshSageIntacctDimensionView,
     SageIntacctAttributesCountView,
@@ -18,6 +19,7 @@
     path('paginated_destination_attributes/', PaginatedDestinationAttributesView.as_view()),
     path('credentials/authorization_code/', AuthorizationCodeView.as_view()),
     path('attributes_count/', SageIntacctAttributesCountView.as_view()),
+    path('check_entity_slide/', EntitySlideCheckView.as_view()),
 ]
 
 sage_intacct_dimension_paths = [

apps/sage_intacct/views.py

@@ -17,7 +17,7 @@
 from sageintacctsdk.exceptions import InvalidTokenError
 from intacctsdk.exceptions import InvalidTokenError as IntacctRESTInvalidTokenError
 
-from apps.sage_intacct.helpers import sync_dimensions
+from apps.sage_intacct.helpers import sync_dimensions, check_entity_slide_error
 from apps.sage_intacct.models import SageIntacctAttributesCount
 from apps.sage_intacct.serializers import SageIntacctAttributesCountSerializer, SageIntacctFieldSerializer
 from apps.workspaces.enums import CacheKeyEnum
@@ -343,3 +343,43 @@
     serializer_class = SageIntacctAttributesCountSerializer
     lookup_field = 'workspace_id'
     lookup_url_kwarg = 'workspace_id'
+
+
+class EntitySlideCheckView(generics.CreateAPIView):
+    """
+    Check if a location entity causes an entity slide error
+    """
+    def post(self, request: Request, *args, **kwargs) -> Response:
+        entity_id = request.data.get('entity_id')
+        workspace_id = kwargs['workspace_id']
+
+        assert_valid(entity_id is not None, 'Entity ID is required')
+
+        try:
+            check_entity_slide_error(workspace_id=workspace_id, entity_id=entity_id)
+            return Response(
+                data={'entity_slide_error': False},
+                status=status.HTTP_200_OK
+            )
+        except SageIntacctCredential.DoesNotExist:
+            logger.info('Sage Intacct credentials not found for workspace_id - %s', workspace_id)
+            return Response(
+                data={'message': 'Sage Intacct credentials not found'},
+                status=status.HTTP_400_BAD_REQUEST
+            )
+        except (InvalidTokenError, IntacctRESTInvalidTokenError):
+            invalidate_sage_intacct_credentials(workspace_id)
+            logger.info('Invalid Sage Intacct token for workspace_id - %s', workspace_id)
+            return Response(
+                data={'message': 'Invalid Sage Intacct token'},
+                status=status.HTTP_400_BAD_REQUEST
+            )
+        except Exception as e:
+            logger.info(
+                'Entity slide error for workspace_id - %s, entity_id - %s: %s',
+                workspace_id, entity_id, str(e)
+            )
+            return Response(
+                data={'entity_slide_error': True, 'message': str(e)},
+                status=status.HTTP_200_OK
+            )