Support for Rate Limits Exception (#99)

fylein · Jan 27, 2023 · c2291d3 · c2291d3
1 parent dbe4a6f
commit c2291d3
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 8 deletions.
diff --git a/netsuitesdk/__init__.py b/netsuitesdk/__init__.py
@@ -8,6 +8,7 @@
     'NetSuiteLoginError',
     'NetSuiteRequestError',
     'NetSuiteTypeError',
+    'NetSuiteRateLimitError'
 ]
 
 name = "netsuitesdk"
diff --git a/netsuitesdk/internal/client.py b/netsuitesdk/internal/client.py
@@ -335,13 +335,22 @@ def request(self, name, *args, **kwargs):
         :return: the request response object
         :rtype: the exact type depends on the request
         """
-        method = getattr(self._service_proxy, name)
-        # call the service:
-        include_search_preferences = (name == 'search')
-        response = method(*args,
-                          _soapheaders=self._build_soap_headers(include_search_preferences=include_search_preferences)
-                          , **kwargs)
-        return response
+        try:
+            method = getattr(self._service_proxy, name)
+            # call the service:
+            include_search_preferences = (name == 'search')
+            response = method(*args,
+                            _soapheaders=self._build_soap_headers(include_search_preferences=include_search_preferences)
+                            , **kwargs)
+            return response
+        except Fault as e:
+            if 'SuiteTalk concurrent request limit exceeded. Request blocked' in str(e):
+                raise NetSuiteRateLimitError(str(e))
+            else:
+                raise
+        except Exception as e:
+            raise
+
 
     def get(self, recordType, internalId=None, externalId=None):
         """

diff --git a/netsuitesdk/internal/exceptions.py b/netsuitesdk/internal/exceptions.py
@@ -3,6 +3,7 @@
     :class:`NetSuiteLoginError`
     :class:`NetSuiteRequestError`
     :class:`NetSuiteTypeError`
+    :class:`NetSuiteRateLimitError`
 """
 
 class NetSuiteError(Exception):
@@ -38,4 +39,9 @@ class NetSuiteRequestError(NetSuiteError):
 class NetSuiteTypeError(NetSuiteError):
     """Exception raised when requested an invalid netsuite type"""
 
-    pass
+    pass
+
+class NetSuiteRateLimitError(NetSuiteError):
+    """Exception raised for errors during rate limit requests like get, search, .."""
+
+    pass