114 lines
4.4 KiB
Python
114 lines
4.4 KiB
Python
from typing import Final
|
|
|
|
from pydantic import parse_obj_as
|
|
from pydantic.dataclasses import dataclass
|
|
|
|
from mdrsclient.api.base import BaseApi
|
|
from mdrsclient.api.utils import token_check
|
|
from mdrsclient.exceptions import UnauthorizedException
|
|
from mdrsclient.models import Folder, FolderSimple
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class FolderCreateResponse:
|
|
id: str
|
|
|
|
|
|
class FolderApi(BaseApi):
|
|
ENTRYPOINT: Final[str] = "v2/folder/"
|
|
|
|
def list(self, laboratory_id: int, path: str) -> list[FolderSimple]:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT
|
|
params = {"path": path, "laboratory_id": laboratory_id}
|
|
token_check(self.connection)
|
|
response = self.connection.get(url, params=params)
|
|
self._raise_response_error(response)
|
|
ret: list[FolderSimple] = []
|
|
for data in response.json():
|
|
ret.append(parse_obj_as(FolderSimple, data))
|
|
return ret
|
|
|
|
def retrieve(self, id: str) -> Folder:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + id + "/"
|
|
token_check(self.connection)
|
|
response = self.connection.get(url)
|
|
self._raise_response_error(response)
|
|
ret = parse_obj_as(Folder, response.json())
|
|
return ret
|
|
|
|
def create(self, name: str, parent_id: str) -> str:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT
|
|
data = {"name": name, "parent_id": parent_id, "description": "", "template_id": -1}
|
|
token_check(self.connection)
|
|
response = self.connection.post(url, data=data)
|
|
self._raise_response_error(response)
|
|
ret = parse_obj_as(FolderCreateResponse, response.json())
|
|
return ret.id
|
|
|
|
def update(self, folder: FolderSimple) -> bool:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + folder.id + "/"
|
|
data = {
|
|
"name": folder.name,
|
|
"description": folder.description,
|
|
}
|
|
token_check(self.connection)
|
|
response = self.connection.put(url, data=data)
|
|
self._raise_response_error(response)
|
|
return True
|
|
|
|
def destroy(self, id: str, recursive: bool) -> bool:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + id + "/"
|
|
params = {"recursive": recursive}
|
|
token_check(self.connection)
|
|
response = self.connection.delete(url, params=params)
|
|
self._raise_response_error(response)
|
|
return True
|
|
|
|
def auth(self, id: str, password: str) -> bool:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + id + "/auth/"
|
|
data = {"password": password}
|
|
token_check(self.connection)
|
|
response = self.connection.post(url, data=data)
|
|
# stupid! always returned status code 200
|
|
self._raise_response_error(response)
|
|
data = response.json()
|
|
if data.get("status") is not True:
|
|
raise UnauthorizedException("Password is incorrect.")
|
|
return True
|
|
|
|
def acl(self, id: str, access_level: int, recursive: bool, password: str | None) -> bool:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + id + "/acl/"
|
|
data: dict[str, int | str] = {"access_level": access_level}
|
|
if password is not None:
|
|
data.update({"password": password})
|
|
if recursive is True:
|
|
data.update({"lower": 1})
|
|
token_check(self.connection)
|
|
response = self.connection.post(url, data=data)
|
|
self._raise_response_error(response)
|
|
return True
|
|
|
|
def move(self, folder: FolderSimple, folder_id: str) -> bool:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + folder.id + "/move/"
|
|
data = {"parent": folder_id}
|
|
token_check(self.connection)
|
|
response = self.connection.post(url, data=data)
|
|
self._raise_response_error(response)
|
|
return True
|
|
|
|
def metadata(self, id: str) -> dict:
|
|
# print(self.__class__.__name__ + "::" + sys._getframe().f_code.co_name)
|
|
url = self.ENTRYPOINT + id + "/metadata/"
|
|
token_check(self.connection)
|
|
response = self.connection.get(url)
|
|
self._raise_response_error(response)
|
|
return response.json()
|