python - 如何在尚未迭代的生成器中捕获 GeneratorExit？

Question

例如，

In [33]: def gener():
    ...:   try:
    ...:     print('hi')
    ...:     yield 1
    ...:     print('hi 2')
    ...:     yield 2
    ...:   except:
    ...:     print('Closed: {!r}'.format(sys.exc_info()[1]))
    ...:

In [34]: gener().close()

# ↑ No output!
In [35]: g = gener()

In [36]: next(g)
hi
Out[36]: 1

In [37]: g.close()
Closed: GeneratorExit()

如果您没有进入 Python 中的生成器，是否不可能注意到生成器的 close() 调用？

(在我的实际用例中，生成器的参数之一是返回生成器的函数传递给生成器的资源，包括所有权。即，

def do_it():
    resource = acquire()
    try:
        return a_generator(resource)()  # pass ownership of the resource
    except:
        resource.close()

如果您在第一次迭代之前关闭生成器，并且生成器无法捕获该事实，则资源将被泄漏。)

Answer 1

你不能。如果生成器尚未被 nexted，则主体内的任何 try 子句都不会被输入。 close 将在函数开始时抛出 GeneratorExit，然后才能设置任何异常处理。

您必须更改前提 - 例如，通过引入虚拟 yield 并提前 next 生成器:

def gen_wrapper():
    gen_iter = _gen()
    next(gen_iter)
    return gen_iter

def _gen():
    try:
        yield
        ...
    finally:
        perform_cleanup()

# Generator returned from gen_wrapper will perform cleanup even if closed immediately.