kgriffs · November 2, 2022 01:52 · Nov 2, 2022 · Nov 2, 2022
diff --git a/example_asyncio_signal_shutdown.py b/example_asyncio_signal_shutdown.py
@@ -18,7 +18,7 @@ async def worker(shutdown_event):
 async def do_shutdown(task):
     # Here we could await join() or whatever
 
-    # Then also cancel one more more tasks
+    # Then also cancel one or more tasks
     task.cancel()
 
 

diff --git a/example_asyncio_signal_shutdown.py b/example_asyncio_signal_shutdown.py
@@ -0,0 +1,40 @@
+import asyncio
+import functools
+import signal
+import sys
+import time
+
+
+async def worker(shutdown_event):
+    try:
+        while True:
+            await asyncio.sleep(1)
+    except asyncio.CancelledError:  # catch it so it does not get logged by the asyncio library
+        pass
+
+    print('Shutting down...')
+
+
+async def do_shutdown(task):
+    # Here we could await join() or whatever
+
+    # Then also cancel one more more tasks
+    task.cancel()
+
+
+async def listener():
+    shutdown_event = asyncio.Event()
+
+    task = asyncio.create_task(worker(shutdown_event))
+    asyncio.get_running_loop().add_signal_handler(
+        signal.SIGINT,
+        functools.partial(asyncio.create_task, do_shutdown(task))
+    )
+
+    await task
+
+    # Not really necessary in this example since asyncio.run() will return
+    #   once listener() returns and then the script ends.
+    sys.exit(0)
+
+asyncio.run(listener())