run_with_wait doesn't cancel but wait for all tasks to be completed when the main task is completed, unlike asyncio.run()

run_with_wait.py

from asyncio import coroutines, events, tasks


def run_with_wait(main, *, debug=None):
    """
    Difference from asyncio.run():
    - asyncio.run() cancels all tasks when `main` is completed.
      any tasks created using `asyncio.create_task()` will hence be cancelled.
    - run_with_wait() waits for all tasks to be completed using
      `run_until_complete()` and `asyncio.tasks.gather()`
    """
    if events._get_running_loop() is not None:
        raise RuntimeError(
            "asyncio.run() cannot be called from a running event loop")

    if not coroutines.iscoroutine(main):
        raise ValueError("a coroutine was expected, got {!r}".format(main))

    loop = events.new_event_loop()
    try:
        events.set_event_loop(loop)
        if debug is not None:
            loop.set_debug(debug)
        return loop.run_until_complete(main)
    finally:
        try:
            _wait_for_all_tasks(loop)
            loop.run_until_complete(loop.shutdown_asyncgens())
            loop.run_until_complete(loop.shutdown_default_executor())
        finally:
            events.set_event_loop(None)
            loop.close()


def _wait_for_all_tasks(loop):
    all_tasks = tasks.all_tasks(loop)
    loop.run_until_complete(tasks.gather(*all_tasks, return_exceptions=True))

    for task in all_tasks:
        if task.exception() is not None:
            loop.call_exception_handler({
                'message': 'unhandled exception during asyncio.run() shutdown',
                'exception': task.exception(),
                'task': task,
            })