Update error.py

snowflakedb · Aug 26, 2024 · 2bd24dd · 2bd24dd
1 parent cd6eb68
commit 2bd24dd
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/snowflake/ingest/error.py b/snowflake/ingest/error.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
+# Copyright (c) 2012-2024 Snowflake Computing Inc. All rights reserved.
 from requests import Response
 
 class IngestResponseError(Exception):
@@ -15,13 +15,17 @@ def __init__(self, response: Response):
                                                                 response.reason)
             return
 
-        self.code = json_body[u'code']
-        self.success = json_body[u'success']
-        self._raw_message = json_body[u'message']
-        self.data = json_body[u'data']
-        self.message = 'Http Error: {}, Vender Code: {}, Message: {}'\
-            .format(self.http_error_code, self.code, self._raw_message)
-
+        try:
+            self.code = json_body[u'code']
+            self.success = json_body[u'success']
+            self._raw_message = json_body[u'message']
+            self.data = json_body[u'data']
+            self.message = 'Http Error: {}, Vender Code: {}, Message: {}' \
+                .format(self.http_error_code, self.code, self._raw_message)
+        except KeyError:
+            self.message = 'Http Error: {}, Message: {}, Body: {}'.format(self.http_error_code,
+                                                                          response.reason,
+                                                                          json_body)
         return
 
     def __str__(self):