为什么_uaexit_u_u_______________________________？

Question

这是我代码的简化版本：

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...

其实我有两个问题：

1. 据我所知，AsyncIO将在事件循环的下一个周期中很快调用一个任务，并为执行提供了机会。但打印行（“该行未执行”）未执行。为什么？假设如果我们在\uuuaexit\uuuu0>中有一个await语句，那么该行后面的代码根本不会执行，我们不应该依赖它来清理，这是正确的吗

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

那么为什么我得到

_{*我正在使用Python 3.10.4}




Answer 1

匿名用户

回答第一个问题：

如果我们在\uuaexit\uuuuu>中有一个await语句，那么该行后面的代码根本不会执行，这是正确的吗？

我会说不，情况并非总是如此。只要main有足够的时间并且可以再次将控制传递回事件循环，__aexit__中的代码就可以执行。我尝试了这个：

async def main():
    async for i in get_numbers():
        print(i)
        if i == 1:
            break
    await asyncio.sleep(4)   #  <---- New

<代码>。run（）只关心传递给它并最终运行的协同路由，而不关心其他协同路由，包括\uu aexit\uu。因此，如果它没有足够的时间或没有将控制传递给事件循环，我就不能依赖于第一个wait语句之后的下一行。

可能有助于的其他信息：

这里是base_events。py/run_forever方法（由.run（）调用），我发现了自己_asyncgen\u finalizer\u钩子传递给系统。设置异步钩子。asyncgen\u finalizer\u钩子的主体是：

    def _asyncgen_finalizer_hook(self, agen):
        self._asyncgens.discard(agen)
        if not self.is_closed():
            self.call_soon_threadsafe(self.create_task, agen.aclose())

但是call_soon_threadsafe的实现是空的。

_{我稍后会清理这个答案并消除这些猜测。}

Answer 2

答案很简单：解释器会在一秒钟后继续执行__aexit__，但是main函数完成了，没有指向上下文管理器的指针。

你提到的第一个明显的解决方案是在main函数之后等待足够长的时间：

async def main():
    async for i in get_numbers():
        print(i)
        if i == 1:
            break
    await asyncio.sleep(4)   #  <---- New

另一种方法是使用try/finally：

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

Answer 3

我不确定发生了什么，但发布我的发现，以防它被证明对决定调查的其他人有用。当我们将对get_numbers（）的引用存储在main（）之外时，输出会更改为预期。我会说get_numbers（）似乎是垃圾收集到早期，但是禁用gc没有帮助，所以我的猜测可能是错误的。

import asyncio


test = None


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")  # <-- Executed now
        await asyncio.sleep(1)


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


async def main():
    global test
    test = get_numbers()

    async for i in test:
        print(i)
        if i == 1:
            break


asyncio.run(main())