python-批量解压zip、rar文件

时间:2023-03-09 19:11:50
python-批量解压zip、rar文件

有bug,因为测试用的小文件--没啥问题,但实际是要解压一些学习资料(比较大,出错可能也看不出来),后来就发现可能有问题了。

自己平时又不用--不改了,所以代码仅供参考

这是一个用python写解压大量zip脚本的说明,本人新手一个,希望能对各位有所启发。

GitHub:https://github.com/lgf133214/Windows-python3-

首先要注意的,在运行自己的脚本之前一定先备份或者复制出一些样本进行测试,不然出错会很麻烦;

之后我用到的是解压zip文件的扩展包zipfile,可以直接pip安装或者在IDE里安装,需要特别注意的是这个包的文件名解码方式需要我们去修改,先去查看源文件,直接搜索“cp437”(一个编码方式),找到后全部替换为“gbk”,即可解决中文显示问题。

代码:

import os
import shutil
import zipfile
import rarfile # 首先引入需要的工具包
# shutil为后期移动文件所需 # 路径改这里!
parent_path = input('请输入要解压的文件路径:') # 文件类型选择
file_flag = '.' + input('请输入一种需要的压缩类型(例:zip):') # 删除已解压过的文件
# 一定要先测试,不然很麻烦
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')

  

  

好了,以上仅是个人所遇到的问题的解决方案,希望能对各位有所帮助,如果解决不了问题还请耐心查阅其他资料,将过程理顺之后再去实践,一定程度可以让自己保持清醒,提高效率。