Async Iteration with Context Managers

Snippet

Async Iteration with Context Managers

This demonstrates combining asynchronous context managers (__aenter__/__aexit__) with asynchronous iterators (__aiter__). It allows for safe resource management in concurrent environments while processing streams of data asynchronously using 'async with' and 'async for'.

snippet.py
python
import asyncio
 
class AsyncBuffer:
    async def __aenter__(self):
        print('Opening...')
        return self
 
    async def __aexit__(self, *args):
        print('Closing...')
 
    async def __aiter__(self):
        for i in range(3):
            await asyncio.sleep(0.1)
            yield i
 
async def main():
    async with AsyncBuffer() as buf:
        async for item in buf:
            print(item)

Breakdown

async def __aenter__(self):

Asynchronous setup method for the 'async with' block.

async def __aiter__(self):

Defines the object as an asynchronous iterator.

yield i

Asynchronously produces a value for the 'async for' loop.

Previous snippet Next snippet

From your library

Async Iteration with Context Managers

Related