Merge pull request influxdata#33 from prune998/patch-3

toddboom · toddboom · commit 5f635fa3fe96 · 2014-06-06T16:03:51.000-04:00
add the status code in the Request exception
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -7,7 +7,12 @@
 import requests
 session = requests.Session()
 
-
+class InfluxDBClientError(Exception):
+    "Raised when an error occures in the Request"
+    def __init__(self, message, code):
+      self.message = message
+      self.code = code
+      
 class InfluxDBClient(object):
     """
     InfluxDB Client
@@ -97,9 +102,10 @@ def request(self, url, method='GET', params=None, data=None,
 
         if response.status_code == status_code:
             return response
+            
         else:
-            raise Exception(
-                "{0}: {1}".format(response.status_code, response.content))
+            error = InfluxDBClientError("{0}: {1}".format(response.status_code, response.content), response.status_code)
+            raise error
 
     # Writing Data
     #
