代码改变世界

python2/python3 内存中打包/压缩文件

2017-08-14 10:56  夏洛克·福尔摩斯  阅读(922)  评论(0编辑  收藏  举报

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