Python 压缩文件

#encoding=utf-8
import tarfile, zipfile

# tar压缩
def make_targz(source_dir, output_filename):                                                        
    with tarfile.open(output_filename, "w:gz") as tar:
        tar.add(source_dir, arcname=os.path.basename(source_dir))
        
# zip压缩
def make_zip(source_dir, output_filename):
    zipf = zipfile.ZipFile(output_filename, 'w')
    pre_len = len(os.path.dirname(source_dir))
    for parent, dirnames, filenames in os.walk(source_dir):
        for filename in filenames:
            #print(filename)
            pathfile = os.path.join(parent, filename)
            arcname = pathfile[pre_len:].strip(os.path.sep)   #相对路径
            zipf.write(pathfile, arcname)
    zipf.close()
    
if __name__ == "__main__":
    make_zip(zip_path, dir_name+".zip")
    make_tar(tar_path, dir_name+".tar.gz")