Exceptions

Author: ingmferrer

microsoftgraph/client.py

@@ -1,6 +1,7 @@
 import base64
 import mimetypes
 import requests
+from microsoftgraph import exceptions
 from urllib.parse import urlencode, urlparse
 
 
@@ -373,6 +374,56 @@ def _request(self, method, url, headers=None, **kwargs):
         return self._parse(requests.request(method, url, headers=_headers, **kwargs))
 
     def _parse(self, response):
+        status_code = response.status_code
         if 'application/json' in response.headers['Content-Type']:
-            return response.json()
-        return response.text
+            r = response.json()
+        else:
+            r = response.text
+        if status_code == 200 or status_code == 201:
+            return r
+        elif status_code == 204:
+            return None
+        elif status_code == 400:
+            raise exceptions.BadRequest(r)
+        elif status_code == 401:
+            raise exceptions.Unauthorized(r)
+        elif status_code == 403:
+            raise exceptions.Forbidden(r)
+        elif status_code == 404:
+            raise exceptions.NotFound(r)
+        elif status_code == 405:
+            raise exceptions.MethodNotAllowed(r)
+        elif status_code == 406:
+            raise exceptions.NotAcceptable(r)
+        elif status_code == 409:
+            raise exceptions.Conflict(r)
+        elif status_code == 410:
+            raise exceptions.Gone(r)
+        elif status_code == 411:
+            raise exceptions.LengthRequired(r)
+        elif status_code == 412:
+            raise exceptions.PreconditionFailed(r)
+        elif status_code == 413:
+            raise exceptions.RequestEntityTooLarge(r)
+        elif status_code == 415:
+            raise exceptions.UnsupportedMediaType(r)
+        elif status_code == 416:
+            raise exceptions.RequestedRangeNotSatisfiable(r)
+        elif status_code == 422:
+            raise exceptions.UnprocessableEntity(r)
+        elif status_code == 429:
+            raise exceptions.TooManyRequests(r)
+        elif status_code == 500:
+            raise exceptions.InternalServerError(r)
+        elif status_code == 501:
+            raise exceptions.NotImplemented(r)
+        elif status_code == 503:
+            raise exceptions.ServiceUnavailable(r)
+        elif status_code == 504:
+            raise exceptions.GatewayTimeout(r)
+        elif status_code == 507:
+            raise exceptions.InsufficientStorage(r)
+        elif status_code == 509:
+            raise exceptions.BandwidthLimitExceeded(r)
+        else:
+            raise exceptions.UnknownError(r)

microsoftgraph/exceptions.py

@@ -0,0 +1,90 @@
+class BaseError(Exception):
+    pass
+
+
+class UnknownError(BaseError):
+    pass
+
+
+class BadRequest(BaseError):
+    pass
+
+
+class Unauthorized(BaseError):
+    pass
+
+
+class Forbidden(BaseError):
+    pass
+
+
+class NotFound(BaseError):
+    pass
+
+
+class MethodNotAllowed(BaseError):
+    pass
+
+
+class NotAcceptable(BaseError):
+    pass
+
+
+class Conflict(BaseError):
+    pass
+
+
+class Gone(BaseError):
+    pass
+
+
+class LengthRequired(BaseError):
+    pass
+
+
+class PreconditionFailed(BaseError):
+    pass
+
+
+class RequestEntityTooLarge(BaseError):
+    pass
+
+
+class UnsupportedMediaType(BaseError):
+    pass
+
+
+class RequestedRangeNotSatisfiable(BaseError):
+    pass
+
+
+class UnprocessableEntity(BaseError):
+    pass
+
+
+class TooManyRequests(BaseError):
+    pass
+
+
+class InternalServerError(BaseError):
+    pass
+
+
+class NotImplemented(BaseError):
+    pass
+
+
+class ServiceUnavailable(BaseError):
+    pass
+
+
+class GatewayTimeout(BaseError):
+    pass
+
+
+class InsufficientStorage(BaseError):
+    pass
+
+
+class BandwidthLimitExceeded(BaseError):
+    pass