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

      

  • 相关阅读:
    一些零碎小知识点积累随笔
    STM32学习笔记——新建工程模板步骤(向原子哥学习)
    记一次电信反射xss的挖掘与利用
    mysql字符串操作相关函数用法总结
    mysql学习笔记
    sigmoid function的直观解释
    多变量线性回归时使用梯度下降(Gradient Descent)求最小值的注意事项
    SAE中Python无法创建多线程的解决方案
    BCNF/3NF 数据库设计范式简介
    web.py下获取get参数
  • 原文地址:https://www.cnblogs.com/sherlockhomles/p/7356618.html
Copyright © 2011-2022 走看看