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

Answer 2

您可以使用另一个函数delete_image（）并在generate_image（）函数的底部调用它

Flask：在send_file（）完成后从服务器删除文件

2 个答案: