[cytation] atexit.register(self._stop_camera)

Author: rickwierenga

pylabrobot/plate_reading/biotek_backend.py

@@ -1,4 +1,5 @@
 import asyncio
+import atexit
 import enum
 import logging
 import math
@@ -176,6 +177,8 @@ async def setup(self, use_cam: bool = False) -> None:
         raise
 
   async def _set_up_camera(self) -> None:
+    atexit.register(self._stop_camera)
+
     if not USE_PYSPIN:
       raise RuntimeError(
         "PySpin is not installed. Please follow the imaging setup instructions. "
@@ -447,19 +450,25 @@ async def stop(self) -> None:
     await self.stop_shaking()
     await self.io.stop()
 
-    if hasattr(self, "cam") and self.cam is not None:
-      await self._reset_trigger()
-
-      self.cam.DeInit()
-      del self.cam
-    if hasattr(self, "spinnaker_system") and self.spinnaker_system is not None:
-      self.spinnaker_system.ReleaseInstance()
+    self._stop_camera()
 
     self._objectives = None
     self._filters = None
     self._slow_mode = None
 
-  async def _reset_trigger(self):
+  def _stop_camera(self) -> None:
+    if self.cam is not None:
+      if self._acquiring:
+        self.stop_acquisition()
+
+      self._reset_trigger()
+
+      self.cam.DeInit()
+      self.cam = None
+    if self.spinnaker_system is not None:
+      self.spinnaker_system.ReleaseInstance()
+
+  def _reset_trigger(self):
     if self.cam is None:
       return