Python 3.11 asyncio concurrency execution example

conrurrency.py

import asyncio


async def run_task():
    print('run task!')
    await asyncio.sleep(3)
    print('long task finished!')


async def run_task_error():
    print('run task error')
    await asyncio.sleep(1)
    raise RuntimeError("shit happens")


async def gather_version():
    coro = run_task()
    coro2 = run_task_error()

    try:
        res, res2 = await asyncio.gather(coro, coro2)
        print(res)
        print(res2)
    except Exception as e:
        print(e)

    await asyncio.sleep(3)


async def wait_version():
    coro = run_task()
    coro2 = run_task_error()

    aws = [
        asyncio.ensure_future(coro),
        asyncio.ensure_future(coro2)
    ]

    done, pending = await asyncio.wait(aws, timeout=5, return_when=asyncio.FIRST_EXCEPTION)
    print(done)
    print(pending)
    for p in pending:
        p.cancel()

async def task_group_version():
    coro = run_task()
    coro2 = run_task_error()
    coro3 = run_task_error()

    try:
        async with asyncio.TaskGroup() as tg:
            tg.create_task(coro)
            tg.create_task(coro2)
            tg.create_task(coro3)
    except Exception as error:
        print(error)
    await asyncio.sleep(3)


async def main():
    await gather_version()
    print()
    await wait_version()
    print()
    await task_group_version()

asyncio.run(main())