Flask返回文件后删除文件

Flask返回文件后删除文件

在web开发中，有时候我们需要让用户能够下载一些文件，比如生成的报表、图片或者其他类型的文件。而在下载完成后，我们可能需要将这些文件删除以节省存储空间或保护用户隐私。在本文中，我们将探讨如何使用Flask返回文件给用户下载，并在用户下载完成后删除这些文件的方法。

Flask返回文件

Flask是一个轻量级的Web框架，它非常适合用来快速开发web应用。在Flask中，要返回一个文件给用户下载，可以使用send_file方法。

from flask import Flask, send_file

app = Flask(__name)

@app.route('/download')
def download_file():
    # 假设要下载的文件名为example.txt
    filename = 'example.txt'

    return send_file(filename, as_attachment=True)

if __name__ == '__main__':
    app.run()

在上面的示例中，我们定义了一个路由/download，当用户访问这个路由时，Flask会将example.txt文件返回给用户进行下载。as_attachment=True表示以附件的形式下载文件，否则浏览器会尝试直接展示文本文件。

删除文件

一旦用户完成了下载，我们可能希望将文件删除。在Python中，可以使用os模块来进行文件操作。

import os

# 删除文件
def delete_file(filename):
    os.remove(filename)
    print(f"{filename} has been deleted.")

上面的代码定义了一个delete_file函数，接受文件名作为参数，使用os.remove方法删除文件，并在删除后打印一条消息。

结合Flask和删除文件

现在我们来将Flask返回文件和删除文件结合起来。当用户下载文件后，我们通过调用删除文件的函数来删除相应的文件。为了实现这个功能，我们需要修改原来的download_file函数。

from flask import Flask, send_file
import os

app = Flask(__name)

# 删除文件
def delete_file(filename):
    os.remove(filename)
    print(f"{filename} has been deleted.")

@app.route('/download')
def download_file():
    # 假设要下载的文件名为example.txt
    filename = 'example.txt'

    try:
        delete_file(filename)
        return send_file(filename, as_attachment=True)
    except Exception as e:
        return str(e)

if __name__ == '__main__':
    app.run()

在上面的代码中，我们在download_file函数中先调用delete_file函数删除文件，然后再使用send_file方法返回文件给用户下载。如果删除文件出现异常，我们捕获异常并返回异常信息。

运行示例

为了演示上面的代码，我们先创建一个名为example.txt的文本文件，内容为Hello, world!。然后运行Flask应用，访问/download路由，查看文件下载和删除的效果。

$ echo "Hello, world!" > example.txt
$ python app.py

浏览器访问http://127.0.0.1:5000/download，点击下载文件。下载完成后，控制台会输出文件已被删除的信息。

总结

在本文中，我们使用Flask返回文件给用户下载，并在用户下载完成后自动删除文件的方法。通过结合send_file方法和自定义的删除文件函数，我们可以实现文件下载和删除的功能。这种方法可以帮助我们更好地管理文件和保护用户隐私。