使用 fastapi UploadFile 验证文件类型和扩展名

Question

我目前正在做一个小项目，其中涉及创建一个fastapi 服务器，允许用户上传一个jar 文件。

基本上我有这条路线：

@app.post("/upload")
async def upload(jar_file: UploadFile = File(...)):

我真的很想检查并验证该文件是否真的是jar 文件。

我可以自己实现它，但我很好奇fastapi 或任何其他包是否提供此功能。

Answer 1

您可以检查 MIME 类型 (https://fastapi.tiangolo.com/tutorial/request-files/#uploadfile)。

@app.post("/upload")
async def upload(jar_file: UploadFile = File(...)):
    if jar_file.content_type != "application/java-archive":
        raise HTTPException(400, detail="Invalid document type")
    return {"filename": "jar_file.filename"}