✨ Added basic upload endpoint

classquiz/routers/storage.py

@@ -1,14 +1,17 @@
 #  This Source Code Form is subject to the terms of the Mozilla Public
 #  License, v. 2.0. If a copy of the MPL was not distributed with this
 #  file, You can obtain one at https://mozilla.org/MPL/2.0/.
-
 import re
+from datetime import datetime
 
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, UploadFile, File, Depends
 from fastapi.responses import StreamingResponse, RedirectResponse
 
-from classquiz.config import settings, storage
+from classquiz.auth import get_current_user
+from classquiz.config import settings, storage, arq
+from classquiz.db.models import User, StorageItem
 from classquiz.storage.errors import DownloadingFailedError
+from uuid import uuid4
 
 settings = settings()
 
@@ -39,3 +42,32 @@ async def download_file(file_name: str):
         media_type="image/*",
         headers={"Cache-Control": "public, immutable, max-age=31536000"},
     )
+
+
+@router.post("/")
+async def upload_file(file: UploadFile = File(), user: User = Depends(get_current_user)):
+    file_id = uuid4()
+
+    size = 0
+    # if file.file.name is None:
+    #     size = len(await file.read())
+    #     print("MemorySize", size)
+    # else:
+    #     f = file.file
+    #     a = file.file.fileno()
+    #     os.path.getsize(file.file.name)
+    #     print("DiskSize", size, "name:", file.file.name, "a:", file.file.tell(), "size")
+    file_obj = StorageItem(
+        id=file_id,
+        uploaded_at=datetime.now(),
+        mime_type=file.content_type,
+        hash=None,
+        user=user,
+        size=size,
+        deleted_at=None,
+        alt_text=None,
+    )
+    file_data = await file.read()
+    await storage.upload(file_name=file_id.hex, file_data=file_data)
+    await file_obj.save()
+    await arq.enqueue_job("calculate_hash", file_id.hex)