如何使用os.scandir()以递归方式在目录树上返回DirEntry对象？

Question

Python 3.5的os.scandir(path)函数返回轻量级DirEntry对象,这些对象对文件信息非常有帮助.但是,它只适用于传递给它的直接路径.有没有办法将它包装在递归函数中,以便它访问给定路径下的所有子目录？

Answer 1

您可以使用递归扫描os.walk(),或者如果您需要DirEntry对象或更多控件,请编写如下的递归函数scantree():

try:
    from os import scandir
except ImportError:
    from scandir import scandir  # use scandir PyPI module on Python < 3.5

def scantree(path):
    """Recursively yield DirEntry objects for given directory."""
    for entry in scandir(path):
        if entry.is_dir(follow_symlinks=False):
            yield from scantree(entry.path)  # see below for Python 2.x
        else:
            yield entry

if __name__ == '__main__':
    import sys
    for entry in scantree(sys.argv[1] if len(sys.argv) > 1 else '.'):
        print(entry.path)

笔记:

• 在PEP 471和os.scandir()文档中还有一些例子.
• 您还可以在for循环中添加各种逻辑,以跳过'.'以及类似内容开头的目录或文件.
• 您通常希望follow_symlinks=false在这样的is_dir()递归函数中调用,以避免符号链接循环.

• 在Python 2.x上,将yield from行替换为:

  for entry in scantree(entry.path):
      yield entry