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

墨蓝 2022-09-15 12:59 312阅读 0赞

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

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBATGktYm9zcw_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center]

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

示例代码:

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

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

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBATGktYm9zcw_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center]: /images/20220828/432b55a75bf54916973bad747161d5c2.png