version 0.1.3

Author: vincentberenz

nightskyrunner/runner.py

@@ -70,21 +70,21 @@ def wait(self):
         """
         if self._previous is None:
             with self._lock:
-                self._previous = time.time()
+                self._previous = time.monotonic()
         while True:
             try:
                 with self._lock:
-                    if time.time() - self._previous > self._period:
+                    if time.monotonic() - self._previous > self._period:
                         break
                 for interrupt in self._interrupts:
                     if interrupt():
-                        self._previous = time.time()
+                        self._previous = time.monotonic()
                         return
                 time.sleep(1.0 / self._core_frequency)
             except KeyboardInterrupt:
                 self._keyboard_interrupted = True
                 break
-        self._previous = time.time()
+        self._previous = time.monotonic()
 
 
 def _clearer_error_message(e: Exception) -> str:
@@ -335,7 +335,10 @@ def _frequency_iterate(self) -> None:
             self._frequency_iterate_error(e, True)
         else:
             self._status.state(State.running)
-        self._status.activity("sleep")
+        try:
+            self._status.activity("sleep")
+        except Exception as e:
+            self._frequency_iterate_error(e, False)
         try:
             self.wait()
         except Exception as e:
@@ -388,9 +391,12 @@ def stop(self, blocking: bool = False) -> None:
         self._monitor_stop(self._on_stop, blocking)
 
     def alive(self) -> bool:
-        if self._thread is None or not self._thread.is_alive():
+        try:
+            if self._thread is None or not self._thread.is_alive():
+                return False
+            return True
+        except Exception:
             return False
-        return True
 
     def revive(self):
         while self.alive():
@@ -449,7 +455,7 @@ def start(self):
         self._starting = True
         self._running.value = True
         with self._manage_starting():
-            if self._status.state != State.error:
+            if self._status.get_state() != State.error:
                 self._status.state(State.starting)
             self._process = Process(
                 target=self.run,
@@ -470,8 +476,11 @@ def stop(self, blocking: bool = False) -> None:
     def alive(self) -> bool:
         if self._process is None:
             return False
-        self._process.join(timeout=0.1)
-        return self._process.is_alive()
+        try:
+            self._process.join(timeout=0.1)
+            return self._process.is_alive()
+        except Exception:
+            return False
 
     def revive(self):
         self._starting = True
@@ -486,6 +495,9 @@ def revive(self):
 
     def run(self, memories: Dict[str, DictProxy], running: MpValue) -> None:
         SharedMemory.set_all(memories)
+        # Reinitialize the _Sleeper lock in the child process to avoid
+        # potential deadlock from inheriting a locked copy after fork.
+        self._lock = threading.Lock()
         # note: running.value set to True by start and revive
         while running.value:  # type: ignore
             try:

nightskyrunner/shared_memory.py

@@ -3,6 +3,7 @@
 """
 
 import multiprocessing as mp
+import os
 from contextlib import contextmanager
 from multiprocessing import managers, sharedctypes
 from threading import Lock
@@ -125,6 +126,15 @@ def stop(cls) -> None:
                 cls._manager = None
 
 
+def _reinit_shared_memory_lock():
+    """Reinitialize SharedMemory._lock in child processes after fork."""
+    SharedMemory._lock = Lock()
+
+
+if hasattr(os, 'register_at_fork'):
+    os.register_at_fork(after_in_child=_reinit_shared_memory_lock)
+
+
 @contextmanager
 def clean_shared_memory() -> Generator[None, None, None]:
     """

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nightskyrunner"
-version = "0.1.2"
+version = "0.1.3"
 description = ""
 authors = ["Vincent Berenz"]
 readme = "README.md"