async_generator_library_yield_and_send_quickstart.py

1import asyncio
2from async_generator import async_generator, yield_, yield_from_
3
4@async_generator
5async def my_generator():
6    await yield_(1)
7    await yield_(2)
8    val = await yield_(3)
9    print(f"Generator received: {val}")
10
11async def main():
12    # Using async for to iterate
13    async for item in my_generator():
14        print(f"Generator yielded: {item}")
15
16    # Using send() to pass a value back in
17    gen = my_generator()
18    await gen.asend(None)  # Start the generator
19    await gen.asend(None)
20    await gen.asend(None)
21    try:
22        await gen.asend("Hello from outside!")
23    except StopAsyncIteration:
24        pass
25
26if __name__ == "__main__":
27    asyncio.run(main())