python实现在目录中查找指定文件的方法

python实现在目录中查找指定文件的方法

本文实例讲述了python实现在目录中查找指定文件的方法。分享给大家供大家参考。具体实现方法如下：

1. 模糊查找

代码如下:

import os
from glob import glob #用到了这个模块
def search_file(pattern, search_path=os.environ['PATH'], pathsep=os.pathsep):
      for path in search_path.split(os.pathsep):
              for match in glob(os.path.join(path, pattern)):
                      yield match
if __name__ == '__main__':
      import sys
      if len(sys.argv)<2  or sys.argv[1].startswith('-'):#sys.argv[0]是当前路径,1开始是后面的参数
              print 'Use: %s ' % sys.argv[0]
              sys.exit(1)
      if len(sys.argv)>2:
              matchs = list(search_file(sys.argv[1],sys.argv[2]))
      else:
              matchs = list(search_file(sys.argv[1]))
      print '%d match' % len(matchs)
      for match in matchs:
              print match

2. 指定的文件名精确查找

代码如下:

import os,optparse
#1:精确查找
def search_file(filename, search_path=os.environ['PATH'], pathsep=os.pathsep):#os.pathsep是分隔符';'
      for path in search_path.split(os.pathsep):
              candidate = os.path.join(path, filename)#预选路径
              if os.path.isfile(candidate):
                      yield os.path.abspath(candidate) #用生成器可以方便控制返回的数据.可以使用.next()等方法只返回下一个子项
def parse_args():#帮助提示
      usage = u'''这是一个查找文件夹路径中是否有文件指定文件的脚本,
第一个参数是要找的文件名，第二个是路径'''
      parser = optparse.OptionParser(usage)
      help = u'要查找的文件名字'
      parser.add_option('--filename', help=help)#type='int',
      help = u'查找的路径多个路径以;分隔'
      parser.add_option('--path', help=help, default='e:')
      options, args = parser.parse_args()
      return options, args
if __name__ == '__main__':
      options, args = parse_args()
      find_file = list(search_file(args[0], args[1]))
      if find_file:
              for file in find_file:
                      print "Found File at %s" % file
      else:
              print "Not Found"

例子：在e:/py和e:/phpwww目录下找以a到d开头的.php的文件
E:py>python_cook [a-d]*.php e:/py;e:/phpwww
2 match
e:/phpwwwcurl.php
e:/phpwwwduoxiancheng.php

希望本文所述对大家的Python程序设计有所帮助。