Flask 如何将动态生成的zip文件发送给客户端

在本文中，我们将介绍如何使用Flask框架将动态生成的zip文件发送给客户端。通常情况下，当我们需要在服务器上动态生成zip文件时，可以使用Flask提供的send_file函数将其发送给客户端。

阅读更多：Flask 教程

生成动态zip文件

首先，我们需要使用Python内置的zipfile模块来生成zip文件。下面是一个简单的示例代码，展示了如何将多个文件压缩成一个zip文件：

import zipfile

def generate_zip():
    file_paths = ['/path/to/file1.txt', '/path/to/file2.txt', '/path/to/file3.txt']
    zip_file_path = '/path/to/output.zip'

    with zipfile.ZipFile(zip_file_path, 'w') as zip_file:
        for file_path in file_paths:
            zip_file.write(file_path)

在上述代码中，我们使用zipfile.ZipFile类创建了一个zip文件对象，并通过write方法将每个文件添加到zip文件中。最后，我们可以将生成的zip文件保存在指定路径。

使用Flask发送zip文件

接下来，我们可以使用Flask框架的send_file函数将动态生成的zip文件发送给客户端。下面是一个简单的Flask应用示例：

from flask import Flask, send_file

app = Flask(__name__)

@app.route('/download')
def download_zip():
    zip_file_path = '/path/to/output.zip'

    return send_file(zip_file_path, as_attachment=True)

在上述代码中，我们创建了一个Flask应用，并定义了一个/download路由。当用户访问该路由时，Flask会调用download_zip函数来发送zip文件。

在download_zip函数中，我们使用send_file函数来发送zip_file_path指定的zip文件。设置as_attachment参数为True，可以指定浏览器将文件下载为附件。

例子

假设我们正在构建一个文档管理系统，用户可以选择多个文件，并将它们压缩成一个zip文件进行下载。下面是一个完整的示例代码：

from flask import Flask, request, send_file
import zipfile

app = Flask(__name__)

@app.route('/')
def index():
    return '''
    <form method="POST" action="/download">
        <input type="checkbox" name="files" value="/path/to/file1.txt"> File 1<br>
        <input type="checkbox" name="files" value="/path/to/file2.txt"> File 2<br>
        <input type="checkbox" name="files" value="/path/to/file3.txt"> File 3<br>
        <input type="submit" value="Download">
    </form>
    '''

@app.route('/download', methods=['POST'])
def download_zip():
    selected_files = request.form.getlist('files')
    zip_file_path = '/path/to/output.zip'

    with zipfile.ZipFile(zip_file_path, 'w') as zip_file:
        for file_path in selected_files:
            zip_file.write(file_path)

    return send_file(zip_file_path, as_attachment=True)

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

在这个示例中，我们创建了一个根路由用于显示文件的选择框。当用户选择文件并点击下载按钮时，Flask会调用download_zip函数来生成并发送zip文件。

用户选择的文件会通过checkbox的name属性传递到download_zip函数中，我们可以使用request.form.getlist方法获取所有选中的文件路径。然后，我们将这些文件添加到zip文件中，并使用send_file函数发送zip文件到客户端。