Fastapi 请求结束后删除临时文件

最新推荐文章于 2024-09-15 16:54:15 发布

chunheli

最新推荐文章于 2024-09-15 16:54:15 发布

阅读量1.1k

点赞数 3

分类专栏：后端文章标签：后端 python

本文链接：https://blog.csdn.net/chunhe_li/article/details/121645317

版权

在使用 Fastapi 构建文件下载接口时，遇到临时文件删除问题。由于 Fastapi 的异步处理，直接删除文件会导致文件在请求结束后被清理。本文介绍了通过 Starlette 背景任务在请求完成后优雅地删除临时 ZIP 文件的方法，避免了维护定时器的复杂性。

摘要由CSDN通过智能技术生成

背景

最近要用 Fastapi 框架简单实现一个文件下载接口，需要临时创建一个目录，将需要下载的文件打包为 zip 文件，返回给前端后，需要把这个 zip 文件删除。一开始我想到了两个思路：1. 维护一个定时器，定时去删除。2. 创建临时文件夹，接口返回后自动清理掉这个文件夹。

对比当然选择第二个方案，因为实现比较简单。

简单 demo 实现如下：

import os
import tempfile
import zipfile

import uvicorn
from fastapi import FastAPI
from fastapi.responses import FileResponse

app = FastAPI()


@app.get("/file/download")
async def file_download():
    base_dir = os.path.dirname(__file__)

    # Create temp dir
    tmp_dir_obj = tempfile.TemporaryDirectory(dir=base_dir)
    tmp_dirname = tmp_dir_obj.name