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