Handle Ctrl+C gracefully in WebAgg example

QuLogic · QuLogic · commit 370aac03143d · 2022-08-03T01:25:18.000-04:00
diff --git a/examples/user_interfaces/embedding_webagg_sgskip.py b/examples/user_interfaces/embedding_webagg_sgskip.py
@@ -16,6 +16,7 @@
 import json
 import mimetypes
 from pathlib import Path
+import signal
 import socket
 
 try:
@@ -259,4 +260,17 @@ def __init__(self, figure):
         print(f"Listening on http://{addr}:{port}/")
     print("Press Ctrl+C to quit")
 
-    tornado.ioloop.IOLoop.instance().start()
+    ioloop = tornado.ioloop.IOLoop.instance()
+
+    def shutdown():
+        ioloop.stop()
+        print("Server stopped")
+
+    old_handler = signal.signal(
+        signal.SIGINT,
+        lambda sig, frame: ioloop.add_callback_from_signal(shutdown))
+
+    try:
+        ioloop.start()
+    finally:
+        signal.signal(signal.SIGINT, old_handler)
