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

代码如下:

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

代码如下:

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"