不知道有没有小伙伴跟我一样犯这样的懒,一些简单重复的工作,总是嫌麻烦,懒得一步步去做,还每次都重复一样的操作。比如解压zip或rar的包,也许你会说,不就解压嘛,有啥的。问题来了,如果只是一个简单的压缩包也是算了,关键是压缩包里还有压缩包,也不知道多少层,要一层层去解压,这就比较恶心了吧?如下图:
学习了Python的递归调用后,心想,为什么我不写一个脚本自动解压所有的压缩包呢?emmm....心动不如行动,说写就写,谁让我是这么懒的一个人呢,连解压都得自动化,哈哈哈。
那就来吧,先从最简单的开始,搞个zip包用命令行解压看看:
#-*- coding: utf-8 -*-
importzipfileimportos
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
importzipfileimportosdefunzip_file(path):'''解压zip包'''
ifos.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 inzip_list:
new_path=os.path.join(unzip_path,zip_file)
unzip_file(new_path)
z.close()elifos.path.isdir(path):for file_name inos.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
像这样的一个路径,打印出来的类型就是一个普通的,整行路径去转换格式是不行的!!只有把乱码的部分切割出来,再单独转换就可以了。哦,天。就这样,一个小小的问题,卡了我好久。泪奔~
再加上编码转换:
#-*- coding: utf-8 -*-#2019/8/13 14:57
importzipfileimportosdefunzip_file(path):'''解压zip包'''
ifos.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 inzip_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)ifos.path.exists(old_path):
os.renames(old_path, new_path)
unzip_file(new_path)
z.close()elifos.path.isdir(path):for file_name inos.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
importzipfileimportrarfileimportosimportsysdefchengeChar(path):'''处理乱码'''
if not os.path.exists(path): returnpath
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!!')returnpath2defdel_zip(path):'''删除解压出来的zip包'''path=chengeChar(path)if path.endswith('.zip') or path.endswith('.rar'):
os.remove(path)elifos.path.isdir(path):for i inos.listdir(path):
file_path=os.path.join(path, i)
del_zip(file_path)#递归调用,先把所有的文件删除
defunzip_file(z, unzip_path):'''解压zip包'''z.extractall(path=unzip_path)
zip_list= z.namelist() #返回解压后的所有文件夹和文件list
z.close()for zip_file inzip_list:
path=os.path.join(unzip_path, zip_file)ifos.path.exists(path): main(path)defmain(path):'''主逻辑函数'''path=chengeChar(path)ifos.path.exists(path):
unzip_path= os.path.splitext(path)[0] #解压至当前目录
if path.endswith('.zip') andzipfile.is_zipfile(path):
z= zipfile.ZipFile(path, 'r')
unzip_file(z, unzip_path)elif path.endswith('.rar'):
r=rarfile.RarFile(path)
unzip_file(r, unzip_path)elifos.path.isdir(path):for file_name inos.listdir(path):
path=os.path.join(path, file_name)ifos.path.exists(path): main(path)else:print(path)else:print('the path is not exist!!!')print(path)if __name__ == '__main__':#zip_path = r'C:\Users\ES-IT-PC-193\Desktop\aa\HighLevel\HighLevel'
zip_path = r'C:\Users\ES-IT-PC-193\Desktop\aa\HighLevel.zip'
#zip_path = sys.argv[1] # 接收传入的路径参数
ifos.path.isdir(zip_path):for file_name inos.listdir(zip_path):
path=os.path.join(zip_path, file_name)
main(path)else:
main(zip_path)if zipfile.is_zipfile(zip_path): #删除解压出来的压缩包
del_zip(os.path.splitext(zip_path)[0]) #以后缀名切割