add the status code in the Request exception

prune998 · prune998 · commit 27c281b12245 · 2014-05-23T11:34:10.000-04:00
Create a InfluxDBClientError class as Exception
This will give allow something similar to an HTTPError, including code and message instead of just message.
It will make it easier to parse the exception, for example when trying to create a database that already exists.
This may be a breaking change for some if they already parse the message.

FYI, messages and codes are defined in Influxdb as :

func errorToStatusCode(err error) int {
switch err.(type) {
case AuthenticationError:
return libhttp.StatusUnauthorized // HTTP 401
case AuthorizationError:
return libhttp.StatusForbidden // HTTP 403
case DatabaseExistsError:
return libhttp.StatusConflict // HTTP 409
default:
return libhttp.StatusBadRequest // HTTP 400
}
}
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -7,6 +7,9 @@
 import requests
 session = requests.Session()
 
+class InfluxDBClientError(Exception):
+    "Raised when an error occures in Influxdb Client"
+    pass
 
 class InfluxDBClient(object):
     """
@@ -95,9 +98,12 @@ 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()
+            error.message = "{0}: {1}".format(response.status_code, response.content)
+            error.code = response.status_code
+            raise error
 
     # Writing Data
     #
