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

开发背景：

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

程序设计的思路：

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

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

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

  遍历，判断子元素是文件,并且文件类型存在于searchTypeArr，如果真则存在返回路径

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

代码：

  

   

    

     

    
    

     
 # 根据配置好的文件，搜索文件夹
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import io
     
    
   

    

     

    
    

     

      import sys
     
    
   

    

     

    
    

     

      sys.stdout = io.TextIOWrapper(sys.stdout.buffer,encoding='utf8') # 处理vscode打印中文乱码问题
     
    
   

    

     

    
    

     

      # 主函数
     
    
   

    

     

    
    

     

      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文件路径列表

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

学无止境

这个小脚本，花费了我三个小时，除了关键字外，所有的api都是我一次一次百度所得，包括那个换行符，没办法，python写的少，但我相信:

虽然我走的慢，但每一步都是在前进，并且永不止步

文章来源: fizzz.blog.csdn.net，作者：拿我格子衫来，版权归原作者所有，如需转载，请联系作者。

原文链接：fizzz.blog.csdn.net/article/details/102619223