Flask：send_file() 完成后从服务器删除文件

Posted 2023-03-14

【中文标题】Flask：send_file() 完成后从服务器删除文件

【英文标题】：Flask: delete file from server after send_file() is completed

【发布时间】：2020-10-10 12:14:59

【问题描述】：

我有一个 Flask 后端，它根据一些用户输入生成图像，并使用 Flask 的send_file() 函数将此图像发送到客户端。

这是 Python 服务器代码：

@app.route('/image',methods=['POST'])
def generate_image():
    cont = request.get_json()
    t=cont['text']
    print(cont['text'])
    name = pic.create_image(t) //A different function which generates the image
    time.sleep(0.5)
    return send_file(f"name.png",as_attachment=True,mimetype="image/png")

我想在将这张图片发送到客户端后从服务器中删除它。

如何实现？

【参考方案1】：

你可以有另一个函数 delete_image() 并在 generate_image() 函数的底部调用它

【讨论】：

return send_file(...) 语句退出了 generate_image 函数，所以不能在它下面添加任何东西

【参考方案2】：

好的，我解决了。我使用@app.after_request 并使用 if 条件检查端点，然后删除图像

@app.after_request
def delete_image(response):
    global image_name
    if request.endpoint=="generate_image": //this is the endpoint at which the image gets generated
        os.remove(image_name)
    return response