如何解决处理提早__iter__终止
for i in MyClass():
if i == 'some condition':
break
# do other stuff
我需要能够处理早期终止迭代的时间。可以在类实现中实现吗?
解决方法
您可以尝试捕获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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。