python压缩文件

#coding=utf-8
#压缩文件

import os,os.path
import zipfile

#压缩：传路径，文件名
def zip_compression(dirname,zipfilename):
    filelist = []
    if os.path.isfile(dirname):
        filelist.append(dirname)
    else :
        for root, dirs, files in os.walk(dirname):
            for name in files:
                filelist.append(os.path.join(root, name))

    zf = zipfile.ZipFile(zipfilename, "w", zipfile.zlib.DEFLATED)
    for tar in filelist:
        arcname = tar[len(dirname):]
        #print arcname
        zf.write(tar,arcname)
    zf.close()

if __name__ == '__main__':
    zip_compression(r'E:/python/learning',r'E:/python/learning/zip.zip')