python包装os.walk遍历目录树

#********************************************
# filename: all_files.py
# Author:   xiyang
# email:    sdlgxxy@gmail.com
# date:     2012-7-2
# description:
#   包装os.walk实现目录树的遍历，提供更加完善的功能，
#   比如文件过滤、单层遍历等
#*********************************************
import os,fnmatch

def all_files(root,patterns='*',single_level=False,yield_folders=False):
patterns = patterns.split(';')
print root
for path,subdirs,files in os.walk(root):
if yield_folders:
files.extend(subdirs)
files.sort()
for name in files:
for pattern in patterns:
if fnmatch.fnmatch(name,pattern):
yield os.path.join(path,name)
break
if single_level:
break

if __name__ == '__main__':
for path in all_files('f:/workspace/python/','*.txt;*.log'):
print path