Python基于zlib库实现二进制文件的压缩与读取

# coding=utf-8
import zlib


def saveData(content, save_path):
    f = open(save_path, 'wb')
    for i in range(len(content)):
        f.write(content[i])
    f.close()


def saveAndCompressData(content, save_path, level=9):
    f = open(save_path, 'wb')  # 打开文件用于写入
    compress = zlib.compressobj(level)  # 根据压缩等级构造对象
    for i in range(len(content)):
        f.write(compress.compress(content[i]))
    f.write(compress.flush())
    f.close()


def readAndDecompressData(file_path):
    parts = []
    infile = open(file_path, 'rb')
    decompress = zlib.decompressobj()
    data = infile.read(1024)  # 每次读取1024个字节
    # 简单解释下这里为什么读1024个字节
    # 这里的1024其实写成其它数字都ok
    # 由于在保存的时候是按行保存的因此读完之后根据换行符进行分割即可
    while data:
        res = decompress.decompress(data)
        parts.append(res)
        data = infile.read(1024)
    tmp_str = ""
    for i in range(len(parts)):
        tmp_str += parts[i]  # 把读取的每一段都拼起来
    lines = tmp_str.split("\n")[:-1]  # 根据换行符进行分割
    return lines


if __name__ == '__main__':
    content = []
    for i in range(10000):
        content.append("zhaoxuhui " + i.__str__().zfill(5) + "\n")
    saveData(content, "test.txt")
    saveAndCompressData(content, "test.zxh")
    lines = readAndDecompressData("test.zxh")
    print lines