我正在编写大量使用协同机制的代码,我希望在关机时有可靠的行为。
假设我有一个协同器和一个上下文管理器:
from contextlib import contextmanager
@contextmanager
def print_context_manager(text):
print("Enter", text)
yield
print("Exit", text)
def coro():
with print_context_manager("coro"):
while True:
print("Loop", (yield))我可以这样用:
c = coro()
next(c)
c.send("Hello ")
c.send("World!")
c.close()不幸的是,据我所知,没有办法在c.close()上执行我自己的代码。尤其是coroutine中的上下文管理器从不打印"Exit coro“。
协同中的上下文管理器有什么意义?我必须手动想出一种方法来指示流的结束吗?那么,close()有什么意义呢?
发布于 2017-10-10 23:58:52
您的上下文管理器有一个bug。纠正它,当协同线关闭时,它将自动执行清理。
关闭协同线的工作方式是在协同线被挂起的地方提高GeneratorExit。如果with中的代码引发异常,则@contextlib.contextmanager在yield点引发该异常。您的上下文管理器不处理这个问题,因此异常会阻止清理工作的运行。
如果您希望在异常情况下运行清理,则需要将yield包装为try--最后,并在finally中执行清理:
@contextmanager
def print_context_manager(text):
print("Enter", text)
try:
yield
finally:
print("Exit", text)https://stackoverflow.com/questions/46677432
复制相似问题