如何在 Flask 中通过 gzip 解压缩 FileStorage 对象

Question

我正在通过 HTTP 多部分/表单数据 POST 请求向我的烧瓶应用程序发送一个压缩文件（使用 gzip 压缩）。该请求正常工作，当我将密钥的值打印到控制台时，它会打印：<FileStorage: 'compressed.zip' ('application/zip')>。这是使用 Windows zip 压缩的压缩文件，但将使用 application/gzip 类型进行压缩。压缩的.zip 文件的密钥称为compressed。我的问题是如何解压缩 gzip 文件，然后将 FileStorage 对象存储到后端的特定路径？这是我目前拥有的代码：

from flask import request, Blueprint, make_response
import gzip
blueprint = Blueprint("blueprint", __name__)

@blueprint.route("/compressed-post", methods=["POST"])
def compressed_post():
    if request.method == "POST":
        file = request.files["compressed"]
        print(file)
        return make_response("", 200)
    else:
        return make_response("", 400)

Answer 1

鉴于您的代码 sn-p...

@blueprint.route("/compressed-post", methods=["POST"])
def compressed_post():
    if request.method == "POST":
        file = request.files["compressed"]

...您只需将file._file 传递给ZipFile 构造函数，因为ZipFile 需要一个类似文件的对象（或完整路径，我们这里没有，因为文件是尚未保存到磁盘）。

然后，您可以使用ZipFiles extractAll 方法将文件提取到您喜欢的路径。

另见

https://docs.python.org/3/library/zipfile.html#zipfile.ZipFile.extractall

所以，完整的代码应该是这样的：

from flask import request, Blueprint, make_response
import gzip
from zipfile import ZipFile
blueprint = Blueprint("blueprint", __name__)

@blueprint.route("/compressed-post", methods=["POST"])
def compressed_post():
    if request.method == "POST":
        file = request.files["compressed"]
        zip_handle = ZipFile(file._file)
        zip_handle.extractall(CHOOSE_YOUR_PATH)
        zip_handle.close()
        return make_response("", 200)
    else:
        return make_response("", 400)
        

ZipFile 也有一个上下文管理器，如果您更喜欢 with 语法。

请注意：file._file 中的_ 表示_file 不是公共接口的一部分，将来可能会更改。

Answer 2

尝试使用以下方法：

from flask import request, Blueprint, make_response
from zipfile import ZipFile
import gzip
blueprint = Blueprint("blueprint", __name__)

@blueprint.route("/compressed-post", methods=["POST"])
def compressed_post():
    if request.method == "POST":
        file = request.files["compressed"]
        print(file)
        file_name = file.filename
        with ZipFile(file_name, 'r') as zip: 
            # printing all the contents of the zip file 
            zip.printdir() 
  
            # extracting all the files 
            print('Extracting all the files now...') 
            zip.extractall(<path_to_destination_dir>) 
            print('Done!')
        
        return make_response("", 200)
    else:
        return make_response("", 400)

有关 zip 的更多详细信息，请参阅：ZipFile 和 extractall()

有关 FileStorage 对象的更多详细信息，请参阅此内容：FileStorage

如果您需要更多信息来存储 FileStorage 对象： 将文件保存到目标路径或文件对象。如果目标是文件对象，您必须在调用后自行关闭它。缓冲区大小是复制过程中内存中保存的字节数。默认为 16KB。

看到这个：save

希望对你有帮助。