🐛 Fix S3 upload not working

This commit is contained in:
Mawoka
2024-06-15 23:16:02 +02:00
parent dcf3a1cee3
commit 0e29a22141
+2 -5
View File
@@ -118,9 +118,7 @@ async def download_file_head(file_name: str) -> Response:
@router.post("/") @router.post("/")
async def upload_file( async def upload_file(file: UploadFile = File(), user: User = Depends(get_current_user)) -> PublicStorageItem:
request: Request, file: UploadFile = File(), user: User = Depends(get_current_user)
) -> PublicStorageItem:
if file.content_type not in ALLOWED_MIME_TYPES: if file.content_type not in ALLOWED_MIME_TYPES:
raise HTTPException(status_code=422, detail="Unsupported") raise HTTPException(status_code=422, detail="Unsupported")
if user.storage_used > settings.free_storage_limit: if user.storage_used > settings.free_storage_limit:
@@ -137,12 +135,11 @@ async def upload_file(
deleted_at=None, deleted_at=None,
alt_text=None, alt_text=None,
) )
print("File Size:", request.headers.get("Content-Length"))
await storage.upload( await storage.upload(
file_name=file_id.hex, file_name=file_id.hex,
file_data=file.file, file_data=file.file,
mime_type=file.content_type, mime_type=file.content_type,
size=request.headers.get("Content-Length"), size=size,
) )
await file_obj.save() await file_obj.save()
await arq.enqueue_job("calculate_hash", file_id.hex) await arq.enqueue_job("calculate_hash", file_id.hex)