find files per pattern

find.py

import os
import fnmatch

def gen_find(filepath, loc):
    for path, diralist, filelist in os.walk(loc):
        for name in fnmatch.filter(filelist, filepath):
            yield os.path.join(path, name)

pyfiles = gen_find('*.py', '.')

for file in pyfiles:
    print file