将zip文件解压缩到内存?

时间:2012-06-06 05:59:10

标签: python memory zip zipfile stringio

如何将zip解压缩到内存?

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

from zipfile import ZipFile
from StringIO import StringIO

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

3 个答案:

答案 0 :(得分:54)

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 :(得分:13)

经常使用存档我会建议制作一个工具,以便舒适地使用内存存档。像这样的东西:

import zipfile
import StringIO

class InMemoryZip(object):

   def __init__(self):
       # Create the in-memory file-like object for working w/imz
       self.in_memory_zip = StringIO.StringIO()

   # 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 file.
       f = file(filename, "wb")
       f.write(self.read())
       f.close()

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

import zipfile import StringIO class InMemoryZip(object): def __init__(self): # Create the in-memory file-like object for working w/imz self.in_memory_zip = StringIO.StringIO() # 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 file. f = file(filename, "wb") f.write(self.read()) f.close() 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")

就像一个魅力

答案 2 :(得分:-4)

可能的原因:

1.此模块当前不处理多磁盘ZIP文件。       (OR)
2.检查StringIO.getvalue()天气Unicode错误即将出现。