将压缩文件提取到内存？

Posted 2023-02-23

【中文标题】将压缩文件提取到内存？

【英文标题】：Extracting a zipfile to memory?

【发布时间】：2012-06-10 03:41:21

【问题描述】：

如何将 zip 提取到内存中？

我的尝试（在 .getvalue() 上返回 None）：

from zipfile import ZipFile
from StringIO import StringIO

def extract_zip(input_zip):
    return StringIO(ZipFile(input_zip).extractall())

【问题讨论】：

另见：***.com/questions/5710867

【参考方案1】：

extractall 提取到文件系统，所以你不会得到你想要的。要在内存中提取文件，请使用ZipFile.read() 方法。

如果你真的需要内存中的全部内容，你可以这样做：

def extract_zip(input_zip):
    input_zip=ZipFile(input_zip)
    return name: input_zip.read(name) for name in input_zip.namelist()

【参考方案2】：

可能的原因：

1.此模块目前不处理多磁盘 ZIP 文件。 (或) 2.检查 StringIO.getvalue() 天气 Unicode 错误即将到来。

【讨论】：

没有。 extractall 不返回任何内容（嗯，默认情况下是 None），这就是他得到的。

【参考方案3】：

如果您经常使用内存存档，我建议您制作一个工具。像这样的：

# Works in Python 2 and 3.

try:
    import BytesIO
except ImportError:
    from io import BytesIO  # Python 3
import zipfile


class InMemoryZip(object):
    def __init__(self):
        # Create the in-memory file-like object for working w/IMZ
        self.in_memory_zip = BytesIO()

    # Just zip it, zip it
    def append(self, filename_in_zip, file_contents):
        # Appends a file with name filename_in_zip and contents of
        # file_contents to the in-memory zip.
        # Get a handle to the in-memory zip in append mode
        zf = zipfile.ZipFile(self.in_memory_zip, "a", zipfile.ZIP_DEFLATED, False)

        # Write the file to the in-memory zip
        zf.writestr(filename_in_zip, file_contents)

        # Mark the files as having been created on Windows so that
        # Unix permissions are not inferred as 0000
        for zfile in zf.filelist:
            zfile.create_system = 0

        return self

    def read(self):
        # Returns a string with the contents of the in-memory zip.
        self.in_memory_zip.seek(0)
        return self.in_memory_zip.read()

    # Zip it, zip it, zip it
    def writetofile(self, filename):
        # Writes the in-memory zip to a physical file.
        with open(filename, "wb") as file:
            file.write(self.read())


if __name__ == "__main__":
    # Run a test
    imz = InMemoryZip()
    imz.append("testfile.txt", "Make a test").append("testfile2.txt", "And another one")
    imz.writetofile("testfile.zip")
    print("testfile.zip created")