What’s the correct way to clean up after an interrupted event loop?

by

When you CTRL+C, the event loop gets stopped, so your calls to t.cancel() don’t actually take effect. For the tasks to be cancelled, you need to start the loop back up again.

Here’s how you can handle it:

import asyncio

@asyncio.coroutine
def shleepy_time(seconds):
    print("Shleeping for {s} seconds...".format(s=seconds))
    yield from asyncio.sleep(seconds)


if __name__ == '__main__':
    loop = asyncio.get_event_loop()

    # Side note: Apparently, async() will be deprecated in 3.4.4.
    # See: https://docs.python.org/3.4/library/asyncio-task.html#asyncio.async
    tasks = asyncio.gather(
        asyncio.async(shleepy_time(seconds=5)),
        asyncio.async(shleepy_time(seconds=10))
    )

    try:
        loop.run_until_complete(tasks)
    except KeyboardInterrupt as e:
        print("Caught keyboard interrupt. Canceling tasks...")
        tasks.cancel()
        loop.run_forever()
        tasks.exception()
    finally:
        loop.close()

Once we catch KeyboardInterrupt, we call tasks.cancel() and then start the loop up again. run_forever will actually exit as soon as tasks gets cancelled (note that cancelling the Future returned by asyncio.gather also cancels all the Futures inside of it), because the interrupted loop.run_until_complete call added a done_callback to tasks that stops the loop. So, when we cancel tasks, that callback fires, and the loop stops. At that point we call tasks.exception, just to avoid getting a warning about not fetching the exception from the _GatheringFuture.

More Related Contents:

  1. How to properly create and run concurrent tasks using python’s asyncio module?
  2. Why does the asyncio’s event loop suppress the KeyboardInterrupt on Windows?
  3. How to use asyncio with threading in Python is Method threading.Timer
  4. When to use and when not to use Python 3.5 `await` ?
  5. How could I use requests in asyncio?
  6. How to limit concurrency with Python asyncio?
  7. asyncio.sleep() vs time.sleep()
  8. How to use asyncio with existing blocking library?
  9. How can I use functools.singledispatch with instance methods?
  10. Python 3.4.0 with MySQL database
  11. How to install PyGame on Python 3.4?
  12. Simplest async/await example possible in Python
  13. Can an asyncio event loop run in the background without suspending the Python interpreter?
  14. How do I run Python asyncio code in a Jupyter notebook?
  15. How to test Python 3.4 asyncio code?
  16. How to use ‘yield’ inside async function?
  17. Getting value of enum on string conversion
  18. urllib HTTPS request:
  19. asyncio: Is it possible to cancel a future been run by an Executor?
  20. How to install PyGame on Python?
  21. Why can I only use the await keyword inside of async function?
  22. How to do multiprocessing in FastAPI
  23. How to set class attribute with await in __init__
  24. What are __signature__ and __text_signature__ used for in Python 3.4
  25. Make a Python asyncio call from a Flask route
  26. Why am I getting NotImplementedError with async and await on Windows?
  27. Create generator that yields coroutine results as the coroutines finish
  28. Is it possible to run only a single step of the asyncio event loop
  29. Handling Timeouts with asyncio
  30. How to combine Celery with asyncio?

Leave a Comment