FastAPI上传Zip文件：SpooledTemporaryFile处理及读取异常问题

我之前在FastAPI处理上传文件时也踩过完全一样的坑！问题的核心其实都是文件指针位置不对——SpooledTemporaryFile在上传完成后，指针默认停在文件末尾，这时候直接读取或复制都会得到空内容，自然导致Zip文件无效、普通文件读出来是空字节。

一、修复Zip文件无法解压的问题 #

要处理上传的Zip文件，关键是先把文件指针移到开头，再进行保存或解压操作。这里给你一个完整的示例：

from fastapi import FastAPI, File, UploadFile
import shutil
import zipfile

app = FastAPI()

@app.post("/upload-and-extract-zip/")
async def process_zip(file: UploadFile = File(...)):
    # 第一步：将文件指针移到文件起始位置
    await file.seek(0)
    
    # 保存Zip文件到磁盘
    with open("received.zip", "wb") as dest_file:
        # 用shutil.copyfileobj高效复制文件内容
        shutil.copyfileobj(file.file, dest_file)
    
    # 尝试解压验证
    try:
        with zipfile.ZipFile("received.zip", "r") as zip_ref:
            zip_ref.extractall("extracted_files")
        return {"status": "success", "message": "Zip文件已成功解压"}
    except zipfile.BadZipFile:
        return {"status": "error", "message": "上传的文件不是有效的Zip格式"}

为什么之前失败？ #

你之前直接用zipfile处理时，文件指针在末尾，导致zipfile读取的是空内容，自然会被判定为无效Zip文件。调用seek(0)后，才能让文件对象从开头读取真实的Zip数据。

二、修复单个文件转BytesIO后读空的问题 #

同样是指针位置的问题，而且要注意两次指针重置：一次是在读取UploadFile之前，另一次是在写入BytesIO之后。示例代码如下：

from fastapi import FastAPI, File, UploadFile
import io

app = FastAPI()

@app.post("/upload-single-file/")
async def process_single_file(file: UploadFile = File(...)):
    # 第一步：重置UploadFile的指针到开头
    await file.seek(0)
    
    # 读取文件内容到BytesIO
    file_content = await file.read()
    bytes_io = io.BytesIO(file_content)
    
    # 第二步：重置BytesIO的指针到开头，否则读取会得到空
    bytes_io.seek(0)
    content = bytes_io.read()
    
    return {
        "filename": file.filename,
        "content_length": len(content),
        "content_sample": content.decode("utf-8")[:100] if content else "空内容"
    }

关键细节： #

• 调用await file.read()后，UploadFile的内部指针会移到末尾，如果后续还要读取，必须再次seek(0)。
• 将内容写入BytesIO后，BytesIO的指针也会停在内容末尾，所以要再次seek(0)才能读取到内容。

额外提示 #

如果你不需要保存到磁盘，而是想直接在内存中处理Zip文件，也可以这样做：

await file.seek(0)
zip_data = await file.read()
with zipfile.ZipFile(io.BytesIO(zip_data), "r") as zip_ref:
    # 直接在内存中操作Zip内容
    file_list = zip_ref.namelist()
    return {"files_in_zip": file_list}

这样既避免了磁盘IO，也能正确处理Zip文件。

内容的提问来源于stack exchange，提问作者Jesus