Python asyncio中含await的__aexit__未完整执行问题咨询

问题复现 #

以下为问题复现代码的简化版本：
main是第二次迭代后即终止的协程，get_numbers是运行于异步上下文管理器内、用于产出数值的异步生成器。

import asyncio


class MyContextManager:
    async def __aenter__(self):
        print("Enter to the Context Manager...")
        return self

    async def __aexit__(self, exc_type, exc_value, exc_tb):
        print(exc_type)
        print("Exit from the Context Manager...")
        await asyncio.sleep(1)
        print("This line is not executed")  # <-------------------
        await asyncio.sleep(1)


async def get_numbers():
    async with MyContextManager():
        for i in range(30):
            yield i


async def main():
    async for i in get_numbers():
        print(i)
        if i == 1:
            break


asyncio.run(main())

上述代码的运行输出如下：

Enter to the Context Manager...
0
1
<class 'asyncio.exceptions.CancelledError'>
Exit from the Context Manager...

待解答问题 #

• 根据常规理解，AsyncIO会在事件循环的下一轮周期调度待执行的Task，给予__aexit__方法执行机会，但示例中print("This line is not executed")语句始终未执行，原因是什么？是否可以认为如果__aexit__内部存在await语句，await之后的代码完全不会执行，不能依赖这部分代码实现资源清理逻辑？
• 对异步生成器执行help()查询得到的说明显示：

|  aclose(...)
 |      aclose() -> raise GeneratorExit inside generator.

但为什么示例中__aexit__内接收到的异常是<class 'asyncio.exceptions.CancelledError'>？

运行环境：Python 3.10.4

问题解答 #

关于__aexit__中await后代码不执行的问题 #

这个现象是Python异步生成器默认回收机制的特性，和__aexit__本身逻辑无关：

• 当async for循环执行break提前退出时，会触发异步生成器的aclose()流程，这个流程默认是事件循环隐式调度的终结任务，事件循环不会等这类隐式终结任务完全跑完——只要任务执行到第一个await点让出控制权，事件循环判断主业务逻辑已经结束，就会直接启动关闭流程，取消所有剩余的未完成任务。
• 示例代码中__aexit__运行到第一个await asyncio.sleep(1)时交出控制权，此时事件循环发现主协程main已经执行完毕，直接抛出CancelledError取消正在运行的__aexit__协程，所以await后面的代码根本没机会执行。
• 绝对不要依赖这种隐式触发的异步生成器清理逻辑做资源释放。要保证异步上下文管理器退出逻辑完整跑完，要么显式调用异步生成器的aclose()并等待执行完成，要么在Python 3.12+调整事件循环的异步生成器关闭策略为强制等待，确保清理逻辑有足够的运行时间。

关于__aexit__接收到CancelledError而非GeneratorExit的问题 #

aclose()的设计确实是先向生成器内部抛入GeneratorExit，你看到CancelledError是因为这个异常被事件循环的关闭逻辑覆盖了：

• 异步生成器触发关闭时，首先会在生成器调用栈内抛出GeneratorExit，触发async with块调用__aexit__。
• 但这个关闭流程是程序退出阶段事件循环隐式调度的任务，当__aexit__执行到第一个await点挂起时，事件循环的关闭逻辑会给所有未完成任务（包括正在运行__aexit__的任务）发送CancelledError，这个异常会替换原本的GeneratorExit沿调用栈传递，所以你在__aexit__里拿到的异常类型是CancelledError。
• 如果你不依赖事件循环隐式回收，而是显式执行await 生成器对象.aclose()等待关闭完成，__aexit__里收到的就会是预期的GeneratorExit，只要没有额外的取消操作，__aexit__内await之后的代码也能正常执行。

内容的提问来源于stack exchange，提问作者S.B