在生成器内返回参数

Question

我知道已经有人问过类似的问题，但没有回答我的需要，因为我的有点不同。

我的代码：

def tFileRead(fileName, JSON=False):
    with open(fileName) as f:
        if JSON:
            return json.load(f)
        for line in f:
            yield line.rstrip('\n')

我想做什么： 如果JSON 为真，则表示它是从json 文件中读取的，我想返回json.load(f)，否则，我想将文件的行生成到生成器中。

我尝试了将生成器转换为 json 的替代方法，但它变得非常混乱、非常快，而且效果不佳。

Answer 1

我想到的第一个解决方案是显式返回一个生成器对象，该对象将提供您尝试实现的确切行为。

问题是：如果你明确地返回了一个像 return (line.rstrip('\n') for line in f) 这样的生成器对象，文件将在返回后关闭，并且从文件中进一步读取将引发异常。

您应该在这里编写两个函数：一个读取 json 文件，一个读取普通文件。然后，您可以编写一个包装器，该包装器基于要调用这两个函数中的哪一个的参数。

或者只是将迭代部分移动到另一个函数中，如下所示：

def iterate_file(file_name):
    with open(file_name) as fin:
        for line in fin:
            yield line.rstrip("\n")


def file_read(file_name, as_json=False):
    if as_json:
        with open(file_name) as fin:
            return json.load(fin)
    else:
        return iterate_file(file_name)

Answer 2

您可以 yield from 加载 JSON 的字典，从而遍历字典中的键值对，但这不是您想要的行为。

def tFileRead(fileName, JSON=False):
    with open(fileName) as f:
        if JSON:
            yield from json.load(f).items()  # works, but differently
        for line in f:
            yield line.rstrip('\n')

如果你可以只返回一个生成器会很好，但这不会起作用，因为使用with，文件会在函数返回后立即关闭，即在生成器被消耗之前。

def tFileRead(fileName, JSON=False):
    with open(fileName) as f:
        if JSON:
            return json.load(f)
        else:
            return (line.rstrip('\n') for line in f)  # won't work

或者，您可以定义另一个函数，仅用于从文件中产生行并在生成器中使用它：

def tFileRead(fileName, JSON=False):
    if JSON:
        with open(fileName) as f:
            return json.load(f)
    else:
        def withopen(fileName):
            with open(fileName) as f:
                yield from f
        return (line.rstrip('\n') for line in withopen(fileName))

但是一旦你在那里，你真的可以只使用两个单独的函数来读取文件 en-block 作为 JSON 或迭代行......