Mawoka
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from aiohttp import ClientSession
|
|
from io import BytesIO
|
|
|
|
|
|
class DetaStorage:
|
|
def __init__(self, deta_base_url: str, deta_id: str, deta_key: str):
|
|
self.deta_url = deta_base_url
|
|
self.deta_id = deta_id
|
|
self.deta_key = deta_key
|
|
self.headers = {
|
|
"X-Api-Key": self.deta_key,
|
|
}
|
|
|
|
async def download(self, file_name: str) -> BytesIO | None:
|
|
"""
|
|
|
|
:param file_name: The name of the file to be downloaded
|
|
:return: Either bytes f successfull download or None if failed
|
|
"""
|
|
async with ClientSession(headers=self.headers) as session:
|
|
async with session.get(f"{self.deta_url}/files/download?name={file_name}") as response:
|
|
if response.status == 200:
|
|
return BytesIO(await response.read())
|
|
elif response.status == 404:
|
|
return None
|
|
else:
|
|
raise Exception("Download failed")
|
|
|
|
async def upload(self, file: bytes, file_name: str) -> None:
|
|
"""
|
|
:param file: The file in bytes
|
|
:param file_name: The name of the file
|
|
:return:
|
|
"""
|
|
async with ClientSession(headers=self.headers) as session:
|
|
async with session.post(f"{self.deta_url}/files?name={file_name}", data=file) as response:
|
|
if response.status == 201:
|
|
return None
|
|
else:
|
|
raise Exception("Upload failed")
|
|
|
|
async def delete(self, file_names: [str]) -> None:
|
|
async with ClientSession(headers=self.headers) as session:
|
|
async with session.delete(f"{self.deta_url}/files", json={"names": file_names}) as response:
|
|
if response.status == 200:
|
|
return None
|
|
else:
|
|
raise Exception("Delete failed")
|