Recursive, max depth file walk, with filtering

get_filenames.py

always = lambda f: True


def get_filenames(root_path, max_depth=1, predicate=None):

    predicate = predicate or always
    path = os.path.normpath(root_path)
    filenames = []

    for root, dirs, files in os.walk(path, topdown=True):

        depth = root[len(path):].count(os.path.sep)

        if max_depth < 0 or depth < max_depth:
            filenames += [os.path.join(root, f) for f in files if predicate(f)]
        else:
            break

    return filenames