import platform import threading from io import BufferedReader from typing import TypedDict from requests import Response, Session # Unpack is new in 3.11 from typing_extensions import Unpack from mdrsclient.__version__ import __version__ from mdrsclient.cache import CacheFile from mdrsclient.exceptions import MissingConfigurationException from mdrsclient.models import Laboratories, Token, User class _KwArgsMDRSConnectionGet(TypedDict, total=False): params: dict[str, str | int] stream: bool class _KwArgsMDRSConnectionPost(TypedDict, total=False): params: dict[str, str | int] data: dict[str, str | int] files: dict[str, BufferedReader] class _KwArgsMDRSConnectionPut(TypedDict, total=False): params: dict[str, str | int] data: dict[str, str | int] files: dict[str, BufferedReader] class _KwArgsMDRSConnectionDelete(TypedDict, total=False): params: dict[str, str | int] class MDRSConnection: url: str session: Session lock: threading.Lock __cache: CacheFile def __init__(self, remote: str, url: str) -> None: super().__init__() self.url = url self.session = Session() self.lock = threading.Lock() self.__cache = CacheFile(remote) self.__prepare_headers() def get(self, url: str, **kwargs: Unpack[_KwArgsMDRSConnectionGet]) -> Response: return self.session.get(self.__build_url(url), **kwargs) def post(self, url: str, **kwargs: Unpack[_KwArgsMDRSConnectionPost]) -> Response: return self.session.post(self.__build_url(url), **kwargs) def put(self, url: str, **kwargs: Unpack[_KwArgsMDRSConnectionPut]) -> Response: return self.session.put(self.__build_url(url), **kwargs) def delete(self, url: str, **kwargs: Unpack[_KwArgsMDRSConnectionDelete]) -> Response: return self.session.delete(self.__build_url(url), **kwargs) def logout(self) -> None: del self.__cache.user del self.__cache.token self.session.headers.update({"Authorization": ""}) @property def user(self) -> User | None: return self.__cache.user @user.setter def user(self, user: User) -> None: self.__cache.user = user @property def token(self) -> Token | None: return self.__cache.token @token.setter def token(self, token: Token) -> None: self.__cache.token = token self.__prepare_headers() @property def laboratories(self) -> Laboratories: return self.__cache.laboratories @laboratories.setter def laboratories(self, laboratories: Laboratories) -> None: self.__cache.laboratories = laboratories def __build_url(self, path: str) -> str: if self.url == "": raise MissingConfigurationException("remote host is not configured") return f"{self.url}/{path}" def __prepare_headers(self) -> None: self.session.headers.update( { "User-Agent": f"MdrsClient/{__version__} (Python {platform.python_version()} - {platform.platform()})", "Accept": "application/json", } ) if self.token is not None: self.session.headers.update({"Authorization": f"Bearer {self.token.access}"})