use concurrent for the multiple file uploads instead of multiprocess.
This commit is contained in:
60
mdrsclient/connection.py
Normal file
60
mdrsclient/connection.py
Normal file
@ -0,0 +1,60 @@
|
||||
import requests
|
||||
|
||||
from mdrsclient.cache import CacheFile
|
||||
from mdrsclient.exceptions import MissingConfigurationException
|
||||
from mdrsclient.models import Laboratories, Token, User
|
||||
|
||||
|
||||
class MDRSConnection:
|
||||
url: str
|
||||
session: requests.Session
|
||||
__cache: CacheFile
|
||||
|
||||
def __init__(self, remote: str, url: str) -> None:
|
||||
super().__init__()
|
||||
self.url = url
|
||||
self.session = requests.Session()
|
||||
self.__cache = CacheFile(remote)
|
||||
self.__prepare_headers()
|
||||
|
||||
def build_url(self, *args) -> str:
|
||||
if self.url == "":
|
||||
raise MissingConfigurationException("remote host is not configured")
|
||||
parts = [self.url]
|
||||
parts.extend(args)
|
||||
return "/".join(parts)
|
||||
|
||||
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 __prepare_headers(self) -> None:
|
||||
self.session.headers.update({"accept": "application/json"})
|
||||
if self.token is not None:
|
||||
self.session.headers.update({"Authorization": f"Bearer {self.token.access}"})
|
Reference in New Issue
Block a user