Python文件的压缩与解压

前言：

Python在人工智能，后台服务等领域中得到了广泛应用。由于python有着大量的第三方开源包以及其易用的语言特性，这门语言在数据处理、批处理等方法具有强大的优势，使用起来相当便捷。本文为大家带来的是使用Python进行文件、文件夹压缩与解压，用到的是zipfile的第三方依赖库。根据不同应用场景封装了几个方法，方便根据实际需要使用相应的方法，能够筛选出需要压缩的文件。

安装：

pip install zipfile

代码：

1、压缩整个文件夹

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

2、压缩指定扩展名文件

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

3、获取指定类型的待压缩文件列表

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)

4、解压文件

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

来源：https://juejin.cn/post/7085611772018688008