将 zipfile 解压缩到内存中？

如何将压缩文件解压缩到内存中？

我的尝试(返回 .getvalue()上的 None) :

from zipfile import ZipFile
from StringIO import StringIO


def extract_zip(input_zip):
return StringIO(ZipFile(input_zip).extractall())
66968 次浏览
小开
最佳答案

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()}
小开

可能的原因:

1. 此模块当前不处理多磁盘 ZIP 文件。 (或)
2. 检查 StringIO.getvalue ()是否出现 Unicode 错误。

小开

如果你经常使用内存存档，我建议你制作一个工具，比如:

# 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")

提交答案