python批量解压

CODE

import os
import shutil
import zipfile
import rarfile
 
# 首先引入需要的工具包
# shutil为后期移动文件所需
 
 
# 路径改这里！
parent_path = input('请输入要解压的文件路径：')
 
# 文件类型选择
file_flag = '.' + input('请输入一种需要解压的压缩类型（例：zip或rar）解压后会删除原有压缩文件，请注意备份：')
 
 
# 删除已解压过的文件
# 一定要先测试，不然很麻烦
def del_old_zip(file_path):
    os.remove(file_path)
 
 
# 解压
def zip_decompress(file_path, root):
    # 开始
    # zipfile打开zip文件
    z = zipfile.ZipFile(f'{file_path}', 'r')
 
    # 解压
    z.extractall(path=f"{root}")  # path为解压路径，解包后位于该路径下
 
    # 判断是否需要重复解包
    for names in z.namelist():
        if names.endswith(file_flag):
            z.close()
            return 1
 
    # 结束
    z.close()
 
    return 0
 
 
def rar_decompress(file_path, root):
    # 开始
    # rarfile打开rar文件
    z = rarfile.RarFile(f'{file_path}', 'r')
 
    # 解压
    z.extractall(path=f"{root}")  # path为解压路径，解包后位于该路径下
 
    # 判断是否需要重复解包
    for names in z.namelist():
        if names.endswith(file_flag):
            z.close()
            return 1
 
    # 结束
    z.close()
 
    return 0
 
 
decompress = None
if file_flag == '.zip':
    decompress = zip_decompress
elif file_flag == '.rar':
    decompress = rar_decompress
else:
    print('格式输入错误或不支持当前格式')
    os.system('pause')
    exit(0)
 
 
# 因为我在使用过程中发现有些文件解包后会混在一起
# 在平时大家手动解压时可能也会遇到提示是否覆盖的问题
# 下面的两个函数解决这一问题
 
# 开始要先创建一个大文件夹  与压缩包名字相同
# 避免后期混乱和麻烦
def start_dir_make(root, dirname):
    os.chdir(root)
    os.mkdir(dirname)
    return os.path.join(root, dirname)
 
 
# 去除多余文件夹
def rem_dir_extra(root, father_dir_name):
    # 递归要注意信息的正常处理  搞不好上一个调用已经改变了东西  而下面的调用还是使用之前的数据
 
    try:
 
        # 判断文件夹重名  开始
        for item in os.listdir(os.path.join(root, father_dir_name)):
 
            # 第一步判断是不是一个文件夹，如果不是则跳过本次循环
            if not os.path.isdir(os.path.join(root, father_dir_name, item)):
                continue
 
            # 判断是否要脱掉一层目录结构
            # 文件夹名字要相同，且子目录中只有单独的一个文件夹
            if item == father_dir_name and len(
                    os.listdir(os.path.join(root, father_dir_name))) == 1:
 
                # 改变工作目录
                os.chdir(root)
 
                # 将无用文件夹重命名，因为直接移动会有重名错误
                os.rename(father_dir_name, father_dir_name + '-old')
 
                # 移动文件后删除空文件夹
                shutil.move(os.path.join(root, father_dir_name + '-old', item), os.path.join(root))
                os.rmdir(os.path.join(root, father_dir_name + '-old'))
 
                # 将去掉一层目录结构后的文件夹继续作为父本递归处理下去
                # 这里要注意，上面已经发生过数据的改动，所以下面递归传参一定要正确！
                rem_dir_extra(root, item)
 
            else:
 
                # 处理那些不满足上面条件的文件夹
                rem_dir_extra(os.path.join(root, father_dir_name), item)
 
    except Exception as e:
 
        # 打印错误信息
        print("清除文件夹出错" + str(e))
 
 
# 入口
if __name__ == '__main__':
 
    flag = 1
 
    while flag:
 
        #  循环遍历文件夹
        for root, dirs, files in os.walk(parent_path):
 
            # 读取文件名
            for name in files:
 
                if name.endswith(file_flag):
                    # 创建文件夹
                    new_ws = start_dir_make(root, name.replace(file_flag, ''))
 
                    # zip文件地址
                    zip_path = os.path.join(root, name)
 
                    # 解压
                    flag = decompress(zip_path, new_ws)
 
 
                    # 一定要备份或先测试，不然可能会凉，自己选择修改
                    del_old_zip(zip_path)
 
                    # 去掉多余的文件结构
                    rem_dir_extra(root, name.replace(file_flag, ''))
 
                    print(f'{root}\\{name}'.join(['文件：', '\n解压完成\n']))
 
    # 由于解压可能解了好几次 所以可能会有已经解压好的父级目录重名无法处理 这里要再处理一次
    rem_dir_extra(os.path.split(parent_path)[0], os.path.split(parent_path)[1])
 
    print("解压完成啦，记得检查有没有{}格式之外的呀!\n\n其他格式需要自己改一下了".format(file_flag))
 
    os.system('pause')

在全量代码审计项目中经常会遇到开发把所有目录压缩的情况，这时候有这个就非常方便了，但是要注意一下解压之后会把压缩文件给删了，所以注意备份，用法

python unzip.py

原文件

效果