一个可以忽略目录的Python walker

Question

我需要一个文件系统遍历程序，我可以指示它忽略遍历的目录，包括该分支下的所有子目录。os.walk和os.path.walk就是不这样做。

Answer 1

所以我做了这个家庭角色遍历函数：

import os
from os.path import join, isdir, islink, isfile

def mywalk(top, topdown=True, onerror=None, ignore_list=('.ignore',)):
    try:
        # Note that listdir and error are globals in this module due
        # to earlier import-*.
        names = os.listdir(top)
    except Exception, err:
        if onerror is not None:
            onerror(err)
        return
    if len([1 for x in names if x in ignore_list]):
        return 
    dirs, nondirs = [], []
    for name in names:
        if isdir(join(top, name)):
            dirs.append(name)
        else:
            nondirs.append(name)

    if topdown:
        yield top, dirs, nondirs
    for name in dirs:
        path = join(top, name)
        if not islink(path): 
            for x in mywalk(path, topdown, onerror, ignore_list):
                yield x
    if not topdown:
        yield top, dirs, nondirs

Answer 2

实际上，os.walk可能会做您想要做的事情。假设我在ignore中有一个要忽略的目录列表(可能是一组)。那么这应该是可行的：

def my_walk(top_dir, ignore):
    for dirpath, dirnames, filenames in os.walk(top_dir):
        dirnames[:] = [ 
            dn for dn in dirnames 
            if os.path.join(dirpath, dn) not in ignore ]
        yield dirpath, dirnames, filenames

Answer 3

可以就地修改os.walk返回值的第二个元素：

...调用者可以就地修改目录名称列表(可能使用del或slice赋值)，walk()将只递归到其名称保留在目录名称中的子目录中；这可以用于修剪搜索...

def fwalk(root, predicate):
    for dirpath, dirnames, filenames in os.walk(root):
        dirnames[:] = [d for d in dirnames if predicate(r, d)]
        yield dirpath, dirnames, filenames

现在，您只需为子目录提交一个谓词：

>>> ignore_list = [...]
>>> list(fwalk("some/root", lambda r, d: d not in ignore_list))