plotly request exception

chriddyp · chriddyp · commit f1e600b08714 · 2014-10-31T14:06:08.000+01:00
diff --git a/plotly/exceptions.py b/plotly/exceptions.py
@@ -15,7 +15,7 @@
 
 
 """
-
+import json
 
 ## Base Plotly Error ##
 
@@ -27,6 +27,30 @@ class InputError(Exception):
     pass
 
 
+class PlotlyRequestError(Exception):
+    def __init__(self, requests_exception):
+        self.status_code = requests_exception.response.status_code
+        self.HTTPError = requests_exception
+        content_type = requests_exception.response.headers['content-type']
+        if 'json' in content_type:
+            content = requests_exception.response.content
+            if content != '':
+                res_payload = json.loads(requests_exception.response.content)
+                if 'detail' in res_payload:
+                    self.message = res_payload['detail']
+                else:
+                    self.message = ''
+            else:
+                self.message = ''
+        elif content_type == 'text/plain':
+            self.message = requests_exception.response.content
+        else:
+            self.message = requests_exception.message
+
+    def __str__(self):
+        return self.message
+
+
 ## Grid Errors ##
 
 COLUMN_NOT_YET_UPLOADED_MESSAGE = (
