How to iterate over an asynchronous iterator with a timeout?

Question

I think it's easier to understand in terms of code:

try:
    async for item in timeout(something(), timeout=60):
        await do_something_useful(item)
except asyncio.futures.TimeoutError:
    await refresh()

I want the async for to run at most 60 seconds.

Answer 1

I needed to do something like this to create a websocket(also an async iterator) which times out if it doesn't get a message after a certain duration. I settled on the following:

socket_iter = socket.__aiter__()
try:
    while True:
        message = await asyncio.wait_for(
            socket_iter.__anext__(),
            timeout=10
        )
except asyncio.futures.TimeoutError:
    # streaming is completed
    pass