2

假设我们有这个递归函数（生成器）

data = [1,[21, 22, [231, 232, 233]], [31, 32, 33]]

def listgen(data):
    for each in data:
        if type(each) is int:
            yield str(each)
        elif type(each) is list:
            g = listgen(each)
            for i in g:
                yield i
        else:
            continue
gen = listgen(data)
print ' '.join(gen)

输出是：

1 21 22 231 232 233 31 32 33

现在......我怎么能把它写成一个类，因为我需要为那个生成器定义 .close() 方法？

4

1 回答 1

2

当您调用close生成器时，它会GeneratorExit在当前暂停执行的位置引发 a。您可以在重新引发异常之前捕获该异常并在except块中进行清理（不允许您忽略它）。

递归生成器稍微复杂一些，因为您需要手动将异常向上传播到递归堆栈。但即使这样也不是太难。这是您的生成器的一个版本，当它关闭时，会在其执行的每个级别上打印“清理”。如果在它从示例数据集中产生 231 后立即关闭它，您将获得 3 次，列表中的每一层嵌套一次。在您的实际用例中，您可以将其替换为关闭数据库连接的代码，或者执行您必须执行的任何其他清理工作。

def closable_listgen(data):
    try:
        for each in data:
            if type(each) is int:
                yield str(each)
            elif type(each) is list:
                g = closable_listgen(each)
                try:
                    for i in g:
                        yield i
                except GeneratorExit:
                    g.close()
                    raise
            else:
                continue
    except GeneratorExit:
        print("cleaning up")
        raise

Python 3.3 使这更简单一些，因为新yield from语法会自动为我们将调用传播到close生成器链的上游：

def closable_listgen33(data):
    try:
        for each in data:
            if type(each) is int:
                yield str(each)
            elif type(each) is list:
                g = closable_listgen33(each)
                yield from g
            else:
                continue
    except GeneratorExit:
        print("cleaning up")
        raise

为了更好地衡量，以下是您如何将生成器实现为一个类（这是您最初要求的）：

class list_gen_class(object):
    def __init__(self, data):
        self.iterator = iter(data)
        self.child = None

    def __iter__(self):
        return self

    def __next__(self):
        while True:
            if self.child:
                try:
                    return next(self.child)
                except StopIteration:
                    self.child = None

            value = next(self.iterator)
            if isinstance(value, int):
                return str(value)
            elif isinstance(value, list):
                self.child = list_gen_class(value)

    def close(self):
        if self.child:
            self.child.close()

        print("cleaning up")
于 2012-11-22T12:35:38.343 回答