Question

Python asynchronous context managers很有用，但是do not work与asyncio.wait_for Timeouts一起使用。

将超时添加到异步上下文管理器的最佳方法是什么？

Answer 1

将超时添加到异步上下文管理器的最佳方法是什么？

您不能将wait_for应用于异步上下文管理器，但可以将其应用于使用它的协程。因此，要将超时添加到上下文管理器，请在异步函数中使用它，然后将超时应用到它。例如：

async def download(url, session):
    async with session.get(url) as resp:
        return await resp.text()

async def print_google(session):
    try:
        text = await asyncio.wait_for(download('http://www.google.com', session), 1)
    except asyncio.TimeoutError:
        text = None
    print(text)

Answer 2

您尝试过async-timeout吗？只需用async with timeout()包装任何异步代码（包括异步上下文管理器的用法）：

import asyncio
from contextlib import asynccontextmanager

from async_timeout import timeout


@asynccontextmanager
async def my_acm():
    print('before')
    yield
    print('after')


async def main():
    async with timeout(1):
        async with my_acm():
            await asyncio.sleep(1.5)


asyncio.run(main())

如果仅要将超时应用于异步上下文管理器，则可以创建利用async-timeout的新上下文管理器：

import asyncio
from contextlib import asynccontextmanager

from async_timeout import timeout


@asynccontextmanager
async def my_acm():
    print('before')
    yield
    print('after')


@asynccontextmanager
async def my_acm_plus_timeout(time):
    async with timeout(time):
        async with my_acm():
            yield


async def main():
    async with my_acm_plus_timeout(1):
        await asyncio.sleep(1.5)


asyncio.run(main())

将超时添加到异步上下文管理器

2 个答案: