Python文件的压缩与解压_python_

pip install zipfile 

import zipfile import os from tqdm import tqdm def zip_file(src_dir, save_name='default'): ''' 压缩文件夹下所有文件及文件夹 默认压缩文件名：文件夹名 默认压缩文件路径：文件夹上层目录 ''' if save_name == 'default': zip_name = src_dir + '.zip' else: if save_name is None or save_name == '': zip_name = src_dir + '.zip' else: zip_name = save_name + '.zip' 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) z.close() return True 

def zip_file_by_extension(src_dir, extension='dll', save_name='default'): ''' 根据文件扩展名筛选需要压缩的文件到压缩包 注：针对单层目录情况，文件夹下的子文件不会被压缩 ''' if save_name == 'default': zip_name = src_dir + '.zip' else: if save_name is None or save_name == '': zip_name = src_dir + '.zip' else: zip_name = save_name + '.zip' z = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED) file_list = os.listdir(src_dir) for filename in tqdm(file_list): if not os.path.isfile(os.path.join(src_dir, filename)): continue if filename.split('.')[-1] == extension: print(r'正在压缩文件：', filename, end='') z.write(os.path.join(src_dir, filename), filename) z.close() return True 

def get_zip_file_list(src_dir, extension='dll'): ''' 获取需要压缩的文件列表 ''' file_list = os.listdir(src_dir) zip_list = [] for filename in file_list: if not os.path.isfile(os.path.join(src_dir, filename)): continue if filename.split('.')[-1] == extension: zip_list.append(filename) print(zip_list) 

def unzip_file(zip_src, dst_dir): ''' 解压缩 ''' r = zipfile.is_zipfile(zip_src) if r: fz = zipfile.ZipFile(zip_src, 'r') for file in fz.namelist(): fz.extract(file, dst_dir) else: print('This is not zip') return False return True