Merge pull request #95 from mark-oppenheim/mark-oppenheim-safer-error…

…-logging Handle unexpected json structure in error payload
snowflakedb · Aug 26, 2024 · daa25c3 · daa25c3
2 parents cd6eb68 + bb19393
commit daa25c3
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/snowflake/ingest/error.py b/snowflake/ingest/error.py
@@ -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):