zoukankan      html  css  js  c++  java
  • 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")
    

      

  • 相关阅读:
    COJ979 WZJ的数据结构(负二十一)
    COJ980 WZJ的数据结构(负二十)
    奇怪的错误
    COJ883 工艺品
    COJ885 LCS???
    COJ559 回文
    hdu1505(dp求最大子矩阵)
    hdu1506(dp求最大子矩阵)
    hdu2569(递推dp)
    hdu1081(最大子矩阵)
  • 原文地址:https://www.cnblogs.com/sherlockhomles/p/7356618.html
Copyright © 2011-2022 走看看