处理提前 __iter__ 终止答案

【问题标题】：Handling early __iter__ termination处理提前 __iter__ 终止
【发布时间】：2020-08-25 12:33:41
【问题描述】：

我有一个实现__iter__ 的类，我有以下情况：

for i in MyClass():
    if i == 'some condition':
        break
    # do other stuff

我需要能够处理迭代提前终止的情况。这可以在类实现中实现吗？

【问题讨论】：

@Tomer 该类还定义了__enter__ 和__exit__，它目前可以用作上下文管理器、迭代器或两者兼而有之。如果__iter__ 已启动但未用尽，则需要进行一些“清理”。
是否必须立即进行清理？如果它可以等待，你可以把它贴在__del__里面。
@0x5453 与其线程相关，所以最好尽快防止线程缠绵。

标签： python python-3.x iteration

【解决方案1】：

你可以试试看GeneratorExit:

class A:
    def __iter__(self):
        # You need to adapt this to your actual __iter__ implementation:
        i = 0
        try:
            while i < 5:
                yield i
                i += 1
        except GeneratorExit:
            print("GeneratorExit")

如果您不存储生成器，这将起作用，例如在

# If you break in any of these two loops, you'll get GeneratorExit printed:
a = A()
for i in a: ...
for i in A(): ...

如果你存储了生成器，只有当生成器被垃圾回收而没有耗尽时，你才会得到GeneratorExit：

a = A()
it = iter(a)
for i in it:
    if i == 2:
        break  # Does not print GeneratorExit.

it = iter(a)  # Prints GeneratorExit.

无论如何，如果生成器还活着，你可能不想清理，否则这会变得很奇怪：

it = iter(A())
for i in it:
    if i == 2:
        break

for i in it:  # What happens here if you cleaned-up?
    pass

【讨论】：

谢谢。 __iter__ 实际上确实使用了yield，因此GeneratorExit 非常适合确保正确处理资源。