Update client.py

prune998 · prune998 · commit 77ab88bcf062 · 2014-05-23T12:29:51.000-04:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -8,9 +8,11 @@
 session = requests.Session()
 
 class InfluxDBClientError(Exception):
-    "Raised when an error occures in Influxdb Client"
-    pass
-
+    "Raised when an error occures in the Request"
+    def __init__(self, message, code):
+      self.message = message
+      self.code = code
+      
 class InfluxDBClient(object):
     """
     InfluxDB Client
@@ -100,9 +102,7 @@ def request(self, url, method='GET', params=None, data=None,
             return response
             
         else:
-            error = InfluxDBClientError()
-            error.message = "{0}: {1}".format(response.status_code, response.content)
-            error.code = response.status_code
+            error = InfluxDBClientError("{0}: {1}".format(response.status_code, response.content), response.status_code)
             raise error
 
     # Writing Data
