Python Cookbook-2.19 根据指定的搜索路径和模式寻找文件

任务

给定一个搜索路径(一个描述目录信息的字符串)，需要在此目录中找出所有符合匹配模式的文件。

解决方案

基本上，需要循环路径中的所有目录。这个循环最好被封装成一个生成器:

import glob，os
def all_files(pattern,search_path,pathsep=os.pathsep):
'''给定搜索路径，找出所有满足匹配条件的文件'''
	for path in search_path.split(pathsep):
		for match in glob.glob(os.path.join(path, pattern)):
			yield match

讨论

生成器的好处是，可以很容易地获取第一个子项，或者所有子项，再或者其中任意一个子项。比如，打印出你的环境变量 PATH 中第一个符合"*.pye"模式的文件:

print all_files('*.pye',os.environ['PATH']).next( )`

打印所有这种文件，一行一个:

for match in all_files('*.pye',os.environ['PATH']):
	print match

以列表形式一次全部打印出来:

print list(all_files('*.pye',os.environ['PATH']))

给 all_files函数提供了一个主脚本，以方便打印出的 PATH 中所有符合匹配模式的文件。因此，不仅能够看到根据指定名字将被执行的那个文件(第一个)，还能看到被第一个文件“屏蔽”掉的其他同名文件:

if __name__ == '__main__':
	import sys
	if len(sys.argv) != 2 or sys.argv[1].startswith('_'):
		print 'Use: %s <pattern>' % sys.arqv[0]
		sys.exit(1)
	matches = list(all_files(sys.argv[1], os.environ['PATH']))
	print '%d match:' % len(matches)
	for match in matches:
		print match