不知道有没有小伙伴跟我一样犯这样的懒,一些简单重复的工作,总是嫌麻烦,懒得一步步去做,还每次都重复一样的操作。比如解压zip或rar的包,也许你会说,不就解压嘛,有啥的。问题来了,如果只是一个简单的压缩包也是算了,关键是压缩包里还有压缩包,也不知道多少层,要一层层去解压,这就比较恶心了吧?如下图:
学习了Python的递归调用后,心想,为什么我不写一个脚本自动解压所有的压缩包呢?emmm....心动不如行动,说写就写,谁让我是这么懒的一个人呢,连解压都得自动化,哈哈哈。
那就来吧,先从最简单的开始,搞个zip包用命令行解压看看:
# -*- coding: utf-8 -*- import zipfile import os path = r'C:\Users\ES-IT-PC-193\Desktop\aa\A.zip' z = zipfile.ZipFile(path, 'r') unzip_path = os.path.split(path)[0] #解压路径 z.extractall(path=unzip_path)
嗯,试了试,没问题,能正常解压。那接下来就是解决如果把压缩包里包含压缩包的也一块解压了,那我得把解压的命令封装成函数,然后循环解压就好了。这里正好用上了刚学不久的递归调用。
# -*- coding: utf-8 -*- # 2019/8/13 14:57 import zipfile import os def unzip_file(path): '''解压zip包''' if os.path.exists(path): if path.endswith('.zip'): z = zipfile.ZipFile(path, 'r') unzip_path = os.path.split(path)[0] z.extractall(path=unzip_path) zip_list = z.namelist() # 返回解压后的所有文件夹和文件 for zip_file in zip_list: new_path = os.path.join(unzip_path,zip_file) unzip_file(new_path) z.close() elif os.path.isdir(path): for file_name in os.listdir(path): unzip_file(os.path.join(path, file_name)) else: print('the path is not exist!!!') if __name__ == '__main__': zip_path = r'C:\Users\ES-IT-PC-193\Desktop\aa\A.zip' unzip_file(zip_path)
嗯,效果还不错,能把所有的压缩包都递归解压开了,然而,问题来了,解压开的文件名称怎么都成乱码了?!!
于是开始寻找解决乱码问题的方法,然后是各种百度。哎,编码问题一直困扰了我很久,总也搞不明白,这里推荐Alex写的一篇博客,写的很详细,反正我看完后就只知道,不管什么编码格式,都统统给它转成Unicode,然后再转成utf-8或者gbk的格式。哈哈哈。
Alex的博客链接:https://www.cnblogs.com/alex3714/articles/7550940.html
然后我就接着搞搞,各种尝试,还是不行,就是转换不过来,奇怪了。搞了好久还是没找到什么原因。彻底蒙圈了,郁闷~~,晚上睡觉我都能梦见我在解决编码问题,哈哈哈。今天早上脑袋还算比较清醒,再继续找bug。一行行代码调试。果然啊,功夫不负有心人。还是被我找到是什么原因了。
C:\Users\ES-IT-PC-193\Desktop\aa\A\│┬┤·└╓\C:\Users\ES-IT-PC-193\Desktop\aa\A\╤Θ╓ñ┬δ.py
像这样的一个路径,打印出来的类型就是一个普通的<type 'str'>,整行路径去转换格式是不行的!!只有把乱码的部分切割出来,再单独转换就可以了。哦,天。就这样,一个小小的问题,卡了我好久。泪奔~
再加上编码转换:
# -*- coding: utf-8 -*- # 2019/8/13 14:57 import zipfile import os def unzip_file(path): '''解压zip包''' if os.path.exists(path): if path.endswith('.zip'): z = zipfile.ZipFile(path, 'r') unzip_path = os.path.split(path)[0] z.extractall(path=unzip_path) zip_list = z.namelist() # 返回解压后的所有文件夹和文件 for zip_file in zip_list: try: zip_file2 = zip_file.encode('cp437').decode('gbk') except: zip_file2 = zip_file.encode('utf-8').decode('utf-8') old_path = os.path.join(unzip_path,zip_file) new_path = os.path.join(unzip_path,zip_file2) if os.path.exists(old_path): os.renames(old_path, new_path) unzip_file(new_path) z.close() elif os.path.isdir(path): for file_name in os.listdir(path): unzip_file(os.path.join(path, file_name)) else: print('the path is not exist!!!') if __name__ == '__main__': zip_path = r'C:\Users\ES-IT-PC-193\Desktop\aa\A.zip' unzip_file(zip_path)
作为有强迫症患者的我,看着这个代码真是乱啊,不爽,需要优化一下。
于是做了一下几点优化:
1、加上rar压缩包的解压,这里需要注意的是,需要下载一个UnRAR.exe文件放在脚本的同级目录。
2、把编码转换再单独写一个函数,传入一个带有乱码的路径,返回转换好的路径就行了。
3、把解压单独写一个函数,这样解压zip和rar的包就不用写段重复的代码了。
4、解压后,目录里有一堆压缩包,我只想要解压后的文件,于是再写了一个删除解压后不用的压缩包的函数。
最终版的代码如下,欢迎各位小伙伴指导。
# -*- coding: utf-8 -*- # 2019/8/13 14:57 import zipfile import rarfile import os import sys def chengeChar(path): '''处理乱码''' if not os.path.exists(path): return path path = path.rstrip('/').rstrip('\\') # 去除路径最右边的/ file_name = os.path.split(path)[-1] # 获取最后一段字符,准备转换 file_path = os.path.split(path)[0] # 获取前面的路径,为rename做准备 try: # 将最后一段有乱码的字符串转换,尝试过整个路径转换,不生效,估计是无法获取整个路径的编码格式吧。 new_name = file_name.encode('cp437').decode('gbk') except: # 先转换成Unicode再转换回gbk或utf-8 new_name = file_name.encode('utf-8').decode('utf-8') path2 = os.path.join(file_path, new_name) # 将转换完成的字符串组合成新的路径 try: os.renames(path, path2) # 重命名文件 except: print('renames error!!') return path2 def del_zip(path): '''删除解压出来的zip包''' path = chengeChar(path) if path.endswith('.zip') or path.endswith('.rar'): os.remove(path) elif os.path.isdir(path): for i in os.listdir(path): file_path = os.path.join(path, i) del_zip(file_path) # 递归调用,先把所有的文件删除 def unzip_file(z, unzip_path): '''解压zip包''' z.extractall(path=unzip_path) zip_list = z.namelist() # 返回解压后的所有文件夹和文件list z.close() for zip_file in zip_list: path = os.path.join(unzip_path, zip_file) if os.path.exists(path): main(path) def main(path): '''主逻辑函数''' path = chengeChar(path) if os.path.exists(path): unzip_path = os.path.split(path)[0] # 解压至当前目录 if path.endswith('.zip'): z = zipfile.ZipFile(path, 'r') unzip_file(z, unzip_path) elif path.endswith('.rar'): r = rarfile.RarFile(path) unzip_file(r, unzip_path) elif os.path.isdir(path): for file_name in os.listdir(path): path = os.path.join(path, file_name) if os.path.exists(path): main(path) else: print(path) else: print('the path is not exist!!!') if __name__ == '__main__': # zip_path = r'C:\Users\ES-IT-PC-193\Desktop\aa\A.zip' zip_path = sys.argv[1] # 接收传入的路径参数 main(zip_path) if zipfile.is_zipfile(zip_path): # 删除解压出来的压缩包 del_zip(os.path.splitext(zip_path)[0]) # 以后缀名切割