Use dict for HTTP status code → exception mapping.

pillarsdk/api.py

@@ -143,35 +143,15 @@ class Api(object):
         """Check HTTP response codes
         """
         status = response.status_code
-        if status in (301, 302, 303, 307):
-            raise exceptions.Redirection(response, content)
-        elif 200 <= status <= 299:
+
+        if 200 <= status <= 299:
             return json.loads(content) if content else {}
-        elif status == 400:
-            raise exceptions.BadRequest(response, content)
-        elif status == 401:
-            raise exceptions.UnauthorizedAccess(response, content)
-        elif status == 403:
-            raise exceptions.ForbiddenAccess(response, content)
-        elif status == 404:
-            raise exceptions.ResourceNotFound(response, content)
-        elif status == 405:
-            raise exceptions.MethodNotAllowed(response, content)
-        elif status == 409:
-            raise exceptions.ResourceConflict(response, content)
-        elif status == 410:
-            raise exceptions.ResourceGone(response, content)
-        elif status == 412:
-            raise exceptions.PreconditionFailed(response, content)
-        elif status == 422:
-            raise exceptions.ResourceInvalid(response, content)
-        elif 401 <= status <= 499:
-            raise exceptions.ClientError(response, content)
-        elif 500 <= status <= 599:
-            raise exceptions.ServerError(response, content)
-        else:
-            raise exceptions.ConnectionError(response,
-                                             content, "Unknown response code: #{response.code}")
+
+        exception = exceptions.exception_for_status(status)
+        if exception:
+            raise exception(response, content)
+
+        raise exceptions.ConnectionError(response, content, "Unknown response code: %s" % status)
 
     def headers(self):
         """Default HTTP headers