#!/usr/bin/env.python
# -*- coding: utf-8 -*-
import zlib

'''
壓縮infile檔案內容，寫入dst檔案中
'''
def compress(infile, dst, level=9):
    infile = open(infile, 'rb')
    dst = open(dst, 'wb')
    compress = zlib.compressobj(level)
    data = infile.read(1024)
    while data:
        dst.write(compress.compress(data))
        data = infile.read(1024
 
)
    dst.write(compress.flush())

'''
讀取infile中的壓縮內容，解壓寫入dst檔案中
'''
def decompress(infile, dst):
    infile = open(infile, 'rb')
    dst = open(dst, 'wb')
    decompress = zlib.decompressobj()
    data = infile.read(1024)
    while data:
        dst.write(decompress.decompress(data))
        data = infile.read(1024
 
)
    dst.write(decompress.flush())

if __name__ == '__main__':
    # compress("F:/zdl/ptd/MaliciousFile/test2.txt","F:/zdl/ptd/MaliciousFile/test.txt")
decompress("F:/zdl/ptd/MaliciousFile/test.txt","F:/zdl/ptd/MaliciousFile/test3.txt")