#! python3 #chapter09-test01- 遍历目录树,查找特定扩展名的文件不论这些文件的位置在哪里,都将他们 #拷贝到一个新的文件夹中 import os,shutil,pprint #寻找指定文件夹内的某种类型的文件 #返回匹配文件的绝对地址的一个列表 #folder-指定的文件夹地址 #fileType-要查找类型的扩展名 def findFileType(fileType,folder): matchFileAbs=[] #用于存放查到的文件绝对地址 #使用walk()遍历目录树 for foldername,subfolders,filenames in os.walk(path): #遍历这个文件名列表,然后匹配扩展名 for filename in filenames: extensionName=os.path.splitext(filename) if extensionName[1]==fileType: #获取绝对路径,然后拷贝 fileAbsPath=os.path.join(foldername,filename) fileAbsPath=os.path.abspath(fileAbsPath) matchFileAbs.append(fileAbsPath) return matchFileAbs #拷贝文件 #srcfileAbs-一个包含文件地址的列表 #desFolder-指定用来存放的文件夹 def copyFile(srcfileAbs,desFolder): for srcPath in srcfileAbs: shutil.copy(srcPath,desFolder) #为了避免出现目标文件夹不存在而copy()又不会创建文件夹造成复制失败的情况 #定义一个函数,他能让指定的地址存在 def presence(path): if not os.path.exists(path): os.makedirs(path) desFolder=r'.forTest oSave2' path=r'.forTest' matchFileAbs=findFileType('.txt',path) presence(desFolder) copyFile(matchFileAbs,desFolder)
关于遍历文件和获取文件名怎么解决
首先想的是使用 os.walk() 还是用 os.listdir() 的组合,使用 os.walk() 返回数据当中的当前文件夹,然后对其使用 os.listdir() 获取文件名;写完后又发现这样太臃肿了,明明 os.walk() 就能返回当前文件夹下的文件名,遂改回来了
遇到的问题
1)忘记了 os.path.splitext(path) 返回的是一个包含文件名与扩展名元组,还以为他只返回一个扩展名的字符串
2)在函数内部对全局变量操作时,我想到了这样是不行的,并进行了测试。完事后有个想法,能不能把这种改错方式迁移到修改Bug的过程中,既能够想到,又能够做到在该BUG的时候去溯源
3)虽然这个程序很简单,但我这次编写非常顺利,且逻辑清晰,还是离不开“关注主要的”这个思想
4)有的可以用相对地址,有的能用绝对地址 不严谨
5)发现 shutil.copy(src,des) 假如 des 不存在,他不会新建文件夹,而是直接将所有的文件复制过去,然后你得到一个没有扩展名的文件