Question

我正在尝试提供一个统一的界面，用于从单个目录或目录列表中检索所有文件。

def get_files(dir_or_dirs):
    def helper(indir):
        file_list = glob.glob("*.txt")
        for file in file_list:
            yield file

    if type(dir_or_dirs) is list:
        # a list of source dirs
        for dir in dir_or_dirs:
            yield helper(dir)
    else:
        # a single source dir
        yield helper(dir_or_dirs)

def print_all_files(file_iter):
    for file in file_iter:
        print(file)        # error here!

问题：

错误说“文件”仍然是一个生成器，无论输入是单个目录还是列表。为什么它仍然是发电机？
是否可以在函数中包装或嵌入生成器？如果是这样，如何使这项工作？

Answer 1

你每次都屈服于helper()：

yield helper(dir)

但helper()本身就是一个生成器。

在Python 3.3及更高版本中，请改用yield from：

yield from helper(dir)

此将控制委托给另一台生成器。来自Yield expressions文档：

使用yield from <expr>时，它会将提供的表达式视为子实例。该子转换器生成的所有值都直接传递给当前生成器方法的调用者。

在旧的Python版本中，包括Python 2.x，使用另一个循环：

for file in helper(dir):
    yield file

有关yield from执行操作的详细信息，请参阅PEP 380 -- Syntax for Delegating to a Subgenerator。

并不是说你真的需要帮助函数，它只是循环遍历glob.glob()结果，你可以直接直接。

您还需要更正您的功能以实际使用indir;目前您忽略了该参数，因此您只能从当前工作目录中获取文本文件。

接下来，您希望使用glob.iglob()代替glob.glob()来对os.scandir()进行延迟评估，而不是一次将所有结果加载到内存中。我只是将非列表dir_or_dirs值转换为列表，然后只使用一个循环：

import glob
import os.path

def get_files(dirs):
    if not isinstance(dirs, list):
        # make it a list with one element
        dirs = [dirs]

    for dir in dirs:
        pattern = os.path.join(dir, '*.txt')
        yield from glob.iglob(pattern)

现在，我不是使用字符串或列表的单个参数，而是使用可变数量的参数，而使用*args参数语法：

def get_files(*dirs):
    for dir in dirs:
        pattern = os.path.join(dir, '*.txt')
        yield from glob.iglob(pattern)

可以使用0个或更多目录调用：

for file in get_files('/path/to/foo', '/path/to/bar'):
    # ...

如何包装或嵌入发电机？

1 个答案: