Flask返回文件后删除文件|极客笔记

Flask返回文件后删除文件

在开发Web应用程序时，有时候我们需要让用户通过网页上传文件，然后再下载这些文件。在这种情况下，我们可能需要在返回文件给用户后将文件从服务器中删除，以节省空间或者保护用户隐私。

本文以Flask框架为例，介绍如何在返回文件给用户后删除文件的操作。首先我们会讲解如何实现文件上传和下载的功能，然后再展示如何在返回文件给用户后删除文件。

文件上传和下载

首先，我们需要在Flask应用中添加文件上传和下载的功能。下面是一个简单的示例代码：

from flask import Flask, request, send_file

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    file = request.files['file']
    file.save('uploads/' + file.filename)
    return 'File uploaded successfully!'

@app.route('/download/<filename>', methods=['GET'])
def download_file(filename):
    return send_file('uploads/' + filename, as_attachment=True)

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

在上面的代码中，我们定义了两个路由：/upload用于接收文件上传请求，并保存文件到服务器的uploads文件夹中；/download/<filename>用于下载指定文件。

返回文件后删除文件

接下来，我们需要在返回文件给用户后删除文件。为了实现这个功能，我们可以使用Python的os模块来操作文件系统。下面是修改后的代码：

import os
from flask import Flask, request, send_file

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    file = request.files['file']
    file.save('uploads/' + file.filename)
    return 'File uploaded successfully!'

@app.route('/download/<filename>', methods=['GET'])
def download_file(filename):
    filepath = 'uploads/' + filename
    return send_file(filepath, as_attachment=True), filepath

@app.after_request
def delete_file(response):
    try:
        filepath = response.headers.get('filepath')
        os.remove(filepath)
    except Exception as e:
        print(f"Failed to delete file: {e}")
    return response

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

在修改后的代码中，我们增加了一个after_request装饰器，每次请求返回响应后会执行delete_file函数。在这个函数中，我们获取了要删除的文件路径，并使用os.remove方法删除文件。需要注意的是，为了在返回文件后删除文件，我们需要将文件路径作为响应头信息返回给客户端。