压缩多层级文件夹下的所有文件

￡神魔★判官ぃ 2023-07-18 04:47 15阅读 0赞

该方法通用性更广，可以将文件夹下的子文件夹中的数据也都压缩。

1.  先遍历获取所有的文件
2.  创建一个压缩文件，将1中所有文件压缩进去
3.  在压缩的时候注意压缩文件中的名称要包含目录，即压缩文件中的文件名arcname= file.replace(fold\_path,’’).lstrip(’/’)

import zipfile
    import glob,os,time,shutil
    
    def ZipFoldWithChildren(fold_path,is_remove=True):
        ''' fold_path:要压缩的文件夹 is_remove:压缩后是否要把源文件删除，默认是删除的 '''
        t0 = time.time()
        # 判断压缩文件是否存在
        fold_path = fold_path.replace('\\','/')
        zipfiles = glob.glob(fold_path+'*.zip')
        if len(zipfiles) >0:
            zipfilename = fold_path + '-{}.zip'.format(len(zipfiles))
        else:
            zipfilename = fold_path + '.zip'
        files = []
        for root,fold,filenames in os.walk(fold_path):
            root = root.replace('\\','/')
            if len(filenames)>0:
                tmp_files = ['/'.join([root,filename]) for filename in filenames]
                files.extend(tmp_files)
        badfiles = []
        isuccess = 0
        ifail = 0
        with zipfile.ZipFile(zipfilename, 'w') as z:
            for file in set(files):
                try:
                    zip_name = file.replace(fold_path,'').lstrip('/')
                    z.write(file, arcname=zip_name, compress_type=zipfile.ZIP_DEFLATED)
                    isuccess += 1
                    if is_remove:
                        os.remove(file)
                except:
                    badfiles.append(file)
                    ifail += 1
        # 如果文件都被压缩了，并且源文件都被删除了，那么这个空文件夹也就没存在的意义了，故删除
        if len(badfiles)==0 and is_remove:# len 比 not badfiles更好理解
            shutil.rmtree(fold_path)
        print(f'文件夹{fold_path}压缩成功，共压缩文件成功{isuccess}个,失败{ifail}个')
        print('耗时{:.2f}s'.format(time.time()-t0))
        return badfiles