54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
from abc import ABC
|
|
|
|
import requests
|
|
from pydantic import parse_obj_as
|
|
from requests import Response
|
|
|
|
from mdrsclient.connection import MDRSConnection
|
|
from mdrsclient.exceptions import (
|
|
BadRequestException,
|
|
ForbiddenException,
|
|
UnauthorizedException,
|
|
UnexpectedException,
|
|
)
|
|
from mdrsclient.models import DRFStandardizedErrors
|
|
|
|
|
|
class BaseApi(ABC):
|
|
connection: MDRSConnection
|
|
|
|
def __init__(self, connection: MDRSConnection) -> None:
|
|
self.connection = connection
|
|
|
|
def _get(self, url, *args, **kwargs) -> Response:
|
|
return self.connection.session.get(self.__build_url(url), *args, **kwargs)
|
|
|
|
def _post(self, url, *args, **kwargs) -> Response:
|
|
return self.connection.session.post(self.__build_url(url), *args, **kwargs)
|
|
|
|
def _put(self, url, *args, **kwargs) -> Response:
|
|
return self.connection.session.put(self.__build_url(url), *args, **kwargs)
|
|
|
|
def _delete(self, url, *args, **kwargs) -> Response:
|
|
return self.connection.session.delete(self.__build_url(url), *args, **kwargs)
|
|
|
|
def _patch(self, url, *args, **kwargs) -> Response:
|
|
return self.connection.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.connection.build_url(*args)
|