python-3.x - 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

好的，我解决了。我使用了 @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