first commit

mdrsclient/api/base.py

@@ -0,0 +1,51 @@
+from abc import ABC
+
+import requests
+from pydantic import parse_obj_as
+from requests import Response
+
+from mdrsclient.exceptions import (
+    BadRequestException,
+    ForbiddenException,
+    UnauthorizedException,
+    UnexpectedException,
+)
+from mdrsclient.models import DRFStandardizedErrors
+from mdrsclient.session import MDRSSession
+
+
+class BaseApi(ABC):
+    def __init__(self, session: MDRSSession) -> None:
+        self.session = session
+
+    def _get(self, url, *args, **kwargs) -> Response:
+        return self.session.get(self.__build_url(url), *args, **kwargs)
+
+    def _post(self, url, *args, **kwargs) -> Response:
+        return self.session.post(self.__build_url(url), *args, **kwargs)
+
+    def _put(self, url, *args, **kwargs) -> Response:
+        return self.session.put(self.__build_url(url), *args, **kwargs)
+
+    def _delete(self, url, *args, **kwargs) -> Response:
+        return self.session.delete(self.__build_url(url), *args, **kwargs)
+
+    def _patch(self, url, *args, **kwargs) -> Response:
+        return self.session.patch(self.__build_url(url), *args, **kwargs)
+
+    def _raise_response_error(self, response: Response) -> None:
+        if response.status_code >= 300:
+            if response.status_code < 400 or response.status_code >= 500:
+                raise UnexpectedException(f"Unexpected status code returned: {response.status_code}.")
+            errors = parse_obj_as(DRFStandardizedErrors, response.json())
+            if response.status_code == requests.codes.bad_request:
+                raise BadRequestException(errors.errors[0].detail)
+            elif response.status_code == requests.codes.unauthorized:
+                raise UnauthorizedException("Login required.")
+            elif response.status_code == requests.codes.forbidden:
+                raise ForbiddenException("You do not have enough permissions. Access is denied.")
+            else:
+                raise UnexpectedException(errors.errors[0].detail)
+
+    def __build_url(self, *args: tuple) -> str:
+        return self.session.build_url(*args)