python 内存文件_python2/python3 内存中打包/压缩文件

python2：(包含压缩选项，如果只打包，可以调整zipfile.ZIP_DEFLATED)

import zipfile

import StringIO

class InMemoryZip(object):

def __init__(self):

# Create the in-memory file-like object

self.in_memory_zip = StringIO.StringIO()

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

def writetofile(self, filename):

'''Writes the in-memory zip to a file.'''

f = file(filename, "w")

f.write(self.read())

f.close()

if __name__ == "__main__":

# Run a test

imz = InMemoryZip()

imz.append("/home/test/1.jpg",imagebuf)

imz.writetofile("test.zip")

python3：(包含打包选项，不压缩)

import zipfile

import io

class InMemoryZip(object):

def __init__(self):

self.in_memory_zip = io.BytesIO()

def append(self, filename_in_zip, file_contents):

zf = zipfile.ZipFile(self.in_memory_zip, "a", zipfile.ZIP_STORED, False)

zf.writestr(filename_in_zip, file_contents)

for zfile in zf.filelist:

zfile.create_system = 0

return self

def read1(self):

self.in_memory_zip.seek(0)

return self.in_memory_zip.read()

def writetofile(self, filename):

f = open(filename, "wb")

f.write(self.read1())

f.close()

if __name__ == "__main__":

imz = InMemoryZip()

f1 = open('/home/yangbing/jpg/1.jpg','rb').read()

imz.append("1.jpg", f1)

f2 = open('/home/yangbing/jpg/2.jpg','rb').read()

imz.append("2.jpg",f2)

imz.writetofile("test.zip")