文件下载接口响应后删除临时文件

问题: FastApi文件下载接口响应后删除临时文件

在我们写文件下载接口的时候,肯定会遇到文件响应给接口调用者后,会清理掉产生的临时文件,但是当我们响应之前肯定无法删除文件,但是在响应完成后也不能删除文件.这时我们怎么办呢?

示例代码:

from fastapi import FastAPI
from starlette.responses import FileResponse


app = FastAPI()


@app.get("/download")
def download():
    file_name = 'd:/test.csv'
    response = FileResponse(file_name, filename=file_name)
    return response

解决方案:

例如FastApi自带的BackgroundTasks后台任务来解决这个问题非常OK.

from fastapi import FastAPI
from starlette.responses import FileResponse
from starlette.background import BackgroundTasks


app = FastAPI()


def del_file(file_name):
    """
    下载完成后删除文件
    :param file_name:
    :return:
    """
    os.remove(file_name)

@app.get("/download")
def download():
    file_name = 'd:/test.csv'
    response = FileResponse(file_name, filename=file_name)
    task = BackgroundTasks()
    task.add_task(del_file, file_name)
    return response

这样在响应完成之后会自动启动一个后台任务去删除刚才生成的临时文件.就这样完美的解决了接口响应后删除临时文件的问题.