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

Question

我有一个 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")

这张图片发送给客户端后，我想从服务器上删除它。

如何实现？

Answer 1

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

Answer 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

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

Flask: delete file from server after send_file() is completed

delete-file

sendfile

flask

python-3.x