Python递归遍历文件夹搜索文件 脚本

开发背景：

电脑的E盘里有很多电子书，以前对个技术比较感兴趣就去下载很多电子书，有些看了，有些没看，这些电子书没有在一个地方，于是我准备写一个脚本，将这个电子书书搜索出来，进行整理一下。

程序设计的思路：

定义一个搜索的根目录baseDir，一个不搜索的文件夹列表notSearhFolderArr，一个搜索的文件类型列表searchTypeArr，

判断根目录baseDir是有效的，并且不存在于notSearhFolderArr数组中，

获取文件夹下的所有文件及文件夹，

遍历，判断子元素是文件就，判断文件类型是否存在于searchTypeArr，如果存在返回路径

判断子元素，是文件夹并且不属于notSearhFolderArr数组中， 执行第一步，进行递归搜索

代码：

 # 根据配置好的文件，搜索文件夹import osimport ioimport sys
sys.stdout = io.TextIOWrapper(sys.stdout.buffer,encoding='utf8')# 主函数baseDir = "E:\\Pang\\for_search" # 搜索的根目录notSearchFolderArr = ['node_modules'] # 不搜索的目录searchFileTypeArr = ['.pdf','.PDF'] # 搜索的文件类型def searhMain():
 allResArr = searchFolder(baseDir)
 print('\n'.join(allResArr))# 搜索一个文件目录 传入一个文件目录路径def searchFolder(folderPath):
 folderName = os.path.split(folderPath)[-1]
 searFilePathArr = [] if os.path.exists(folderPath) and (folderName not in notSearchFolderArr):
 fileArr = os.listdir(folderPath) for item in fileArr:
 currentPath = folderPath+'\\'+item
 (fileName,fileType) = os.path.splitext(item) if os.path.isfile(currentPath) and (fileType in searchFileTypeArr):
 searFilePathArr.append(currentPath) if os.path.isdir(currentPath) and (item not in notSearchFolderArr):
 innerFileArr = searchFolder(currentPath)
 searFilePathArr.extend(innerFileArr) return searFilePathArr
searhMain()

主要用到的模块和api：

模块 os： 操作文件的模块

主要api：

os.path.split ： 分割路径
os.path.exists： 路径是否存在
os.listdir： 路径是否是文件夹
os.path.splitext：拆分路径中的文件扩展名于其他
os.path.isfile： 路径是否是文件
append： 向数组中追加一个元素
extend: 向数组追加一个数组

运行结果：

程序返回的事根目录下所有的pdf文件路径列表

这个脚本稍作修改就可以查询多个文件类型，对文件进行分类，获取要查询的文件路径后，还可以对文件，进行批量备份到一个文件夹下，或者将搜索结果直接打包压缩。现在怎么玩都可以。