Add quiet start/stop with automatic stop atexit.

Author: ayjayt

src/py/kaleido/__init__.py

@@ -15,19 +15,37 @@
 _global_server = _sync_server.GlobalKaleidoServer()
 
 
+def safe_start_sync_server(*args, **kwargs):
+    """
+    Start a kaleido server which will process all sync generation requests.
+
+    The kaleido server is a singleton, so it can't be opened twice. This
+    function will simply return if the server is already running.
+
+    This wrapper function takes the exact same arguments as kaleido.Kaleido().
+    """
+    _global_server.ensure_opened(*args, **kwargs)
+
+
 def start_sync_server(*args, **kwargs):
     """
     Start a kaleido server which will process all sync generation requests.
 
-    Only one server can be started at a time.
+    The kaleido server is a singleton, so it can't be opened twice. This
+    function will warn you if the server is already running.
 
     This wrapper function takes the exact same arguments as kaleido.Kaleido().
     """
     _global_server.open(*args, **kwargs)
 
 
-def stop_sync_server():
+def safe_stop_sync_server():
     """Stop the kaleido server. It can be restarted."""
+    _global_server.ensure_closed()
+
+
+def stop_sync_server():
+    """Stop the kaleido server. It can be restarted. Warns if not started."""
     _global_server.close()
 
 

src/py/kaleido/_sync_server.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import asyncio
+import atexit
 import warnings
 from queue import Queue
 from threading import Thread
@@ -53,6 +54,15 @@ def __new__(cls):
     def is_running(self):
         return self._initialized
 
+    def ensure_opened(self, *args: Any, **kwargs: Any) -> None:
+        with warnings.catch_warnings():
+            warnings.filterwarnings(
+                "ignore",
+                message=r"Server already open.",
+                category=RuntimeWarning,
+            )
+            self.open(*args, **kwargs)
+
     def open(self, *args: Any, **kwargs: Any) -> None:
         """Initialize the singleton with three values."""
         if self.is_running():
@@ -72,6 +82,16 @@ def open(self, *args: Any, **kwargs: Any) -> None:
         self._return_queue: Queue[Any] = Queue()
         self._thread.start()
         self._initialized = True
+        atexit.register(self.ensure_closed)
+
+    def ensure_closed(self):
+        with warnings.catch_warnings():
+            warnings.filterwarnings(
+                "ignore",
+                message=r"Server already closed.",
+                category=RuntimeWarning,
+            )
+            self.close()
 
     def close(self):
         """Reset the singleton back to an uninitialized state."""