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")
技术成就现在,眼光着看未来。