UpcloudAPIError

Author: Elias Nygren

upcloud_api/__init__.py

@@ -13,6 +13,7 @@
 __license__ = 'MIT'
 __copyright__ = 'Copyright (c) 2015 Elias Nygren'
 
+from upcloud_api.errors import UpCloudAPIError
 from upcloud_api.storage import Storage
 from upcloud_api.ip_address import IP_address
 from upcloud_api.server import Server, login_user_block

upcloud_api/base.py

@@ -1,7 +1,10 @@
 from __future__ import unicode_literals
+
 import json
 import requests
 
+from upcloud_api import UpCloudAPIError
+
 
 class BaseAPI(object):
     """
@@ -65,6 +68,8 @@ def __error_middleware(self, res, res_json):
         Middleware that raises an exception when HTTP statuscode is an error code.
         """
         if(res.status_code in [400, 401, 402, 403, 404, 405, 406, 409]):
-            raise Exception(res_json)
+            err_dict = res_json.get('error', {})
+            raise UpCloudAPIError(error_code=err_dict.get('error_code'),
+                                  error_message=err_dict.get('error_message'))
 
         return res_json

upcloud_api/errors.py

@@ -0,0 +1,13 @@
+
+class UpCloudAPIError(Exception):
+    """Custom Error class for UpCloud API error responses.
+
+    Each API call returns an `error_code` and `error_message` that
+    are available as attributes via instances of this class.
+    """
+    def __init__(self, error_code, error_message):
+        self.error_code = error_code
+        self.error_message = error_message
+
+    def __str__(self):
+        return '{0} {1}'.format(self.error_code, self.error_message)