读取发送到 Flask 服务器的 zip 文件而不将其存储在磁盘上

Question

我想读取通过表单发布请求发送到烧瓶服务器的特定类型的 zip 文件中的所有文件，而不必将 zip 文件存储在磁盘上。

Answer 1

首先获取获取zip文件的代码

from flask import Flask, request
app = Flask(__name__)

@app.route("/",methods=["GET"])
def page_name_get(): 
    return """<form action="." method="post" enctype=multipart/form-data>
        <input type="file" accept="application/zip" name="data_zip_file" accept="application/zip" required>
         <button type="submit">Send zip file!</button>
        </form>"""
app.run()

这是 post 请求函数的样子

import zipfile

@app.route("/",methods=["POST"])
def page_name_post():
    file = request.files['data_zip_file']  
    file_like_object = file.stream._file  
    zipfile_ob = zipfile.ZipFile(file_like_object)
    file_names = zipfile_ob.namelist()
    # Filter names to only include the filetype that you want:
    file_names = [file_name for file_name in file_names if file_name.endswith(".txt")]
    files = [(zipfile_ob.open(name).read(),name) for name in file_names]
    return str(files)

现在我将逐行检查这一点

file = request.files['data_zip_file'] 首先，您需要从请求中获取文件对象，这是 werkzeug.datastructures.FileStorage该类的。

file_like_object = file.stream._file在这里，您首先取werkzeug.datastructures.FileStoragethis的 stream 属性，这是文件的输入流。这将返回tempfile.SpooledTemporaryFile用于临时文件的类的实例。从该实例中，您获取 ._file 属性。这将返回一个实例tempfile._TemporaryFileWrapperThis is enough like anio.BytesIO让 zipfile.ZipFile 类理解。

zipfile_ob = zipfile.ZipFile(file_like_object)在这里创建zipfile.Zipfile对象

现在，您应该可以使用 zip 完成几乎所有您想做的事情。要从 zip 中选择文件，请使用zipfile_ob.open()方法并传入要打开的文件的路径。

为了获得我们使用的那些路径 file_names = zipfile_ob.namelist()它将返回一个列表，其中包含指向 zip 中所有文件和目录的所有路径的字符串。

然后你可以过滤这些名字 file_names = [file_name for file_name in file_names if file_name.endswith(".txt")]

您想要的所有路径现在都在file_names. 然后您可以使用 open 函数提取这些文件的数据。

files = [(zipfile_ob.open(name).read(),name) for name in file_names]

在给出的示例中，我将文件路径保留在最终列表中，但如果您不想要，可以使用：

files = [zipfile_ob.open(name).read() for name in file_names]或使用其他方式查看文件。如果您想了解有关文件的更多信息infolist()，还可以使用该方法而不是namelist()这将返回ZipInfo 对象列表，而不是仅包含字符串的列表。这些对象包含有关所有文件的更多数据。