Python批量压缩文件夹

import os
import zipfile

def zip_file(src_dir, zip_name):

    z = zipfile.ZipFile(zip_name,'w',zipfile.ZIP_DEFLATED)

    for dirpath, dirnames, filenames in os.walk(src_dir):

        fpath = dirpath.replace(src_dir,'')

        fpath = fpath and fpath + os.sep or ''

        for filename in filenames:

            z.write(os.path.join(dirpath, filename),fpath+filename)

            print ('==压缩成功==')

    z.close()


file_root = 'train_oriresize_split'
save_root = 'train_oriresize_split_zip'
if not os.path.exists(save_root):
    os.makedirs(save_root)

file_name_list = os.listdir(file_root)

for file_name in file_name_list:
    filr_path_name = os.path.join(file_root, file_name)
    zip_name = os.path.join(save_root, file_name.split('.')[0]+'.zip')
    zip_file(filr_path_name, zip_name)