refactored expose method

Author: thusser

pyobs_flipro/fliprocamera.py

@@ -128,15 +128,8 @@ async def _expose(self, exposure_time: float, open_shutter: bool, abort_event: a
         # start exposure
         self._driver.start_exposure()
 
-        # wait for exposure to finish
-        while not self._driver.is_available():
-            # aborted?
-            if abort_event.is_set():
-                await self._change_exposure_status(ExposureStatus.IDLE)
-                raise InterruptedError("Aborted exposure.")
-
-            # sleep a little
-            await asyncio.sleep(0.01)
+        # wait exposure
+        await self._wait_exposure(abort_event, exposure_time)
 
         # readout
         log.info("Exposure finished, reading out...")
@@ -177,6 +170,23 @@ async def _expose(self, exposure_time: float, open_shutter: bool, abort_event: a
         log.info("Readout finished.")
         return image
 
+    async def _wait_exposure(self, abort_event: asyncio.Event, exposure_time: float) -> None:
+        """Wait for exposure to finish.
+
+        Params:
+            abort_event: Event that aborts the exposure.
+            exposure_time: Exp time in sec.
+        """
+        # wait for exposure to finish
+        while not self._driver.is_available():
+            # aborted?
+            if abort_event.is_set():
+                await self._change_exposure_status(ExposureStatus.IDLE)
+                raise InterruptedError("Aborted exposure.")
+
+            # sleep a little
+            await asyncio.sleep(0.01)
+
     async def _abort_exposure(self) -> None:
         """Abort the running exposure. Should be implemented by derived class.