More robust wait for exit condition during .serve() (#1939)

* More robust wait for exit condition during .serve() * lint
rerun-io · Apr 20, 2023 · b0d986a · b0d986a
1 parent 1af5080
commit b0d986a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/rerun_py/rerun_sdk/rerun/script_helpers.py b/rerun_py/rerun_sdk/rerun/script_helpers.py
@@ -91,10 +91,10 @@ def script_teardown(args: Namespace) -> None:
 
     """
     if args.serve:
-        import signal
-        from threading import Event
+        import time
 
-        exit = Event()
-        signal.signal(signal.SIGINT, lambda sig, frame: exit.set())
-        print("Sleeping while serving the web viewer. Abort with Ctrl-C")
-        exit.wait()
+        try:
+            while True:
+                time.sleep(1)
+        except KeyboardInterrupt:
+            print("Ctrl-C received. Exiting.")