Merge pull request #6109 from JoeyShapiro/dev

fix: disallow empty files
This commit is contained in:
Timothy Jaeryang Baek
2024-10-20 20:41:56 -07:00
committed by GitHub
2 changed files with 11 additions and 0 deletions

View File

@@ -47,6 +47,12 @@ def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
file_path = f"{UPLOAD_DIR}/{filename}"
contents = file.file.read()
if len(contents) == 0:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.EMPTY_CONTENT,
)
with open(file_path, "wb") as f:
f.write(contents)
f.close()