Added new case for 449

qubole · Jun 10, 2019 · e90918f · e90918f
1 parent cf452bd
commit e90918f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/qds_sdk/connection.py b/qds_sdk/connection.py
@@ -152,9 +152,14 @@ def _handle_error(response):
         elif code == 422:
             sys.stderr.write(response.text + "\n")
             raise ResourceInvalid(response)
-        elif code in (449, 502, 503, 504):
+        elif code in (502, 503, 504):
             sys.stderr.write(response.text + "\n")
             raise RetryWithDelay(response)
+        elif code == 449:
+            message = "Data requested is unavailable. Retrying ..."
+            sys.stderr.write(response.text + "\n")
+            sys.stderr.write(message + "\n")
+            raise RetryWithDelay(response, message)
         elif 401 <= code < 500:
             sys.stderr.write(response.text + "\n")
             raise ClientError(response)

diff --git a/qds_sdk/exception.py b/qds_sdk/exception.py
@@ -11,8 +11,11 @@ def __init__(self, message, usage):
 class Error(Exception):
     """A general error derived from Exception."""
 
-    def __init__(self, request):
-        Exception.__init__(self, request.text)
+    def __init__(self, request, message = ""):
+        response = request.text
+        if message:
+           response = message
+        Exception.__init__(self, response)
         self.request = request