set atexit in mockupcamera

mxcube · Mar 3, 2025 · 37aca15 · 37aca15
1 parent 240212d
commit 37aca15
Show file tree

Hide file tree

Showing 3 changed files with 317 additions and 12 deletions.
diff --git a/mxcubecore/HardwareObjects/mockup/MDCameraMockup.py b/mxcubecore/HardwareObjects/mockup/MDCameraMockup.py
@@ -1,6 +1,9 @@
 """Class for cameras connected to framegrabbers run by Taco Device Servers"""
 
+import atexit
 import logging
+import os
+import signal
 import subprocess
 import time
 from typing import (
@@ -115,6 +118,10 @@ def get_stream_size(self) -> Tuple[int, int, float]:
         scale = float(width) / self.get_width()
         return (width, height, scale)
 
+    def clean_up(self):
+        logging.getLogger("HWR").info("Shutting down video_stream...")
+        os.kill(self._video_stream_process.pid, signal.SIGTERM)
+
     def start_video_stream_process(self) -> None:
         if (
             not self._video_stream_process
@@ -141,6 +148,8 @@ def start_video_stream_process(self) -> None:
                 close_fds=True,
             )
 
+            atexit.register(self.clean_up)
+
     def stop_streaming(self) -> None:
         if self._video_stream_process:
             try: