nominal-streaming 0.7.7__cp310-abi3-musllinux_1_2_aarch64.whl → 0.7.12__cp310-abi3-musllinux_1_2_aarch64.whl

nominal_streaming/nominal_dataset_stream.py

@@ -34,6 +34,7 @@ import datetime
 import logging
 import pathlib
 import signal
+import threading
 from types import TracebackType
 from typing import Mapping, Sequence, Type
 
@@ -78,6 +79,7 @@ class NominalDatasetStream:
         self._opts = opts
         self._impl = PyNominalDatasetStream(self._opts)
         self._old_sigint = None
+        self._opened = False
 
     @classmethod
     def create(
@@ -181,25 +183,35 @@ class NominalDatasetStream:
         """Create the stream as a context manager.
 
         NOTE: installs a sigint handler to enable more graceful shutdown.
-              This is restored upon exit.
+              This is restored upon exit. The handler is only installed when
+              running in the main thread; in worker threads, the stream still
+              functions but Ctrl+C will not trigger graceful cancellation.
         """
-        if self._old_sigint is not None:
+        if self._opened:
             raise RuntimeError("Stream already opened!")
 
         logger.info("Opening underlying stream")
         self._impl.open()
+        self._opened = True
+
+        if threading.current_thread() is threading.main_thread():
+            # Map Ctrl+C → fast cancel; keep handler tiny and re-raise KeyboardInterrupt.
+            def _on_sigint(signum, frame):  # type: ignore[no-untyped-def]
+                logger.debug("Cancelling underlying stream")
+                try:
+                    self._impl.cancel()
+                finally:
+                    raise KeyboardInterrupt
+
+            logger.info("Installing sigint handler")
+            self._old_sigint = signal.getsignal(signal.SIGINT)  # type: ignore[assignment]
+            signal.signal(signal.SIGINT, _on_sigint)
+        else:
+            logger.info(
+                "Stream opened from worker thread; Ctrl+C will not trigger fast cancellation. "
+                "Stream will close gracefully when the context manager exits."
+            )
 
-        # Map Ctrl+C → fast cancel; keep handler tiny and re-raise KeyboardInterrupt.
-        def _on_sigint(signum, frame):  # type: ignore[no-untyped-def]
-            logger.debug("Cancelling underlying stream")
-            try:
-                self._impl.cancel()
-            finally:
-                raise KeyboardInterrupt
-
-        logger.info("Installing sigint handler")
-        self._old_sigint = signal.getsignal(signal.SIGINT)  # type: ignore[assignment]
-        signal.signal(signal.SIGINT, _on_sigint)
         return self
 
     def __enter__(self) -> Self:
@@ -227,6 +239,7 @@ class NominalDatasetStream:
                 logger.info("Restoring original sigint handler")
                 signal.signal(signal.SIGINT, self._old_sigint)
                 self._old_sigint = None
+            self._opened = False
 
     def __exit__(
         self,

{nominal_streaming-0.7.7.dist-info → nominal_streaming-0.7.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nominal-streaming
-Version: 0.7.7
+Version: 0.7.12
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Rust
 Classifier: License :: OSI Approved :: MIT License

nominal_streaming-0.7.12.dist-info/RECORD

@@ -0,0 +1,9 @@
+nominal_streaming-0.7.12.dist-info/METADATA,sha256=3eBnTdIuigzCrrQ95lSTguA9FcqFKBMP2J8DoHUKb_A,2815
+nominal_streaming-0.7.12.dist-info/WHEEL,sha256=EYTFw83l-1Egr0yK8_y3A2oK6lLpVyAvyCNXIbM92ys,107
+nominal_streaming.libs/libgcc_s-0bf60adc.so.1,sha256=9fhX5YpiwtHdsiLMLz77Pw9jjIEGriteV7UOUrrwvpk,527681
+nominal_streaming/__init__.py,sha256=7isEI7NFTEnTNEmxjtDgehQQPPRM-ByrH903VBWrulQ,202
+nominal_streaming/_nominal_streaming.abi3.so,sha256=px2fDVEjTHnBLsLm_WdwvQ6xn2gNOHQzARJwKzEAs2c,7736865
+nominal_streaming/_nominal_streaming.pyi,sha256=hZ5TpAFFMZzajSu13yh-klEh81ItvF5C5f_3i0rMfkM,12093
+nominal_streaming/nominal_dataset_stream.py,sha256=egiSbsd6q_yL8ub8Rc6YukPauwEDivICQd1dSTlOfEc,12121
+nominal_streaming/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nominal_streaming-0.7.12.dist-info/RECORD,,

nominal_streaming-0.7.7.dist-info/RECORD

@@ -1,9 +0,0 @@
-nominal_streaming-0.7.7.dist-info/METADATA,sha256=FFVgUr8QBa6CIJORR6G_r9v-cokNL4U40IX5iP7ZHqk,2814
-nominal_streaming-0.7.7.dist-info/WHEEL,sha256=EYTFw83l-1Egr0yK8_y3A2oK6lLpVyAvyCNXIbM92ys,107
-nominal_streaming.libs/libgcc_s-39080030.so.1,sha256=fIO6GHOh8Ft9CR0Geu7wSUb9Xnl122iTtrxQQ9TAkTQ,789673
-nominal_streaming/__init__.py,sha256=7isEI7NFTEnTNEmxjtDgehQQPPRM-ByrH903VBWrulQ,202
-nominal_streaming/_nominal_streaming.abi3.so,sha256=COb2kv29-z3yisfbtHMofFWAtg3nHl7nrfvdPNvaJTw,7867921
-nominal_streaming/_nominal_streaming.pyi,sha256=hZ5TpAFFMZzajSu13yh-klEh81ItvF5C5f_3i0rMfkM,12093
-nominal_streaming/nominal_dataset_stream.py,sha256=0OXLiGDhyT9Q4z0QJlXaV7H3Nd3OWMPi2bjQJKcM5eE,11507
-nominal_streaming/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nominal_streaming-0.7.7.dist-info/RECORD,,