Tune editor

folkugat_web/services/files.py

@@ -0,0 +1,58 @@
+import mimetypes
+import re
+import uuid
+from pathlib import Path
+
+import magic
+from fastapi import HTTPException, UploadFile
+from folkugat_web.config import db
+
+
+async def get_mimetype(upload_file: UploadFile) -> str:
+    info = magic.detect_from_content(await upload_file.read(2048))
+    await upload_file.seek(0)
+    return info.mime_type
+
+
+ACCEPTED_MIMETYPES = [
+    re.compile(r"image/.+"),
+    re.compile(r".+/pdf"),
+]
+
+
+def check_mimetype(mimetype: str) -> None:
+    if not any(regex.match(mimetype) for regex in ACCEPTED_MIMETYPES):
+        raise HTTPException(status_code=400, detail=f"Unsupported file type: {mimetype}")
+
+
+def get_db_file_path(filepath: Path) -> str:
+    return f"{db.DB_FILES_URL}/{filepath.relative_to(db.DB_FILES_DIR)}"
+
+
+async def store_file(tema_id: int, upload_file: UploadFile) -> str:
+    if not upload_file.size:
+        raise HTTPException(status_code=400, detail="Couldn't find out the size of the file")
+    if upload_file.size > db.FILE_MAX_SIZE:
+        raise HTTPException(
+            status_code=400,
+            detail=f"The uploaded file is too big (max size = {db.FILE_MAX_SIZE} bytes)",
+        )
+
+    mimetype = await get_mimetype(upload_file)
+    check_mimetype(mimetype)
+
+    extension = mimetypes.guess_extension(mimetype) or ""
+    filename = str(uuid.uuid4().hex) + extension
+    filedir = db.DB_FILES_DIR / str(tema_id)
+    filedir.mkdir(exist_ok=True)
+    filepath = filedir / filename
+
+    with open(filepath, "wb") as f:
+        f.write(await upload_file.read())
+
+    return get_db_file_path(filepath)
+
+
+def list_files(tema_id: str) -> list[str]:
+    filedir = db.DB_FILES_DIR / str(tema_id)
+    return [get_db_file_path(f) for f in filedir.iterdir()]