pyglaze 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

pyglaze/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"

pyglaze/device/ampcom.py

@@ -32,6 +32,13 @@ if TYPE_CHECKING:
     from pyglaze.helpers.types import FloatArray
 
 
+class DeviceComError(Exception):
+    """Raised when an error occurs in the communication with the device."""
+
+    def __init__(self: DeviceComError, message: str) -> None:
+        super().__init__(message)
+
+
 @dataclass
 class _ForceAmpCom:
     config: ForceDeviceConfiguration
@@ -272,12 +279,12 @@ class _LeAmpCom:
         self._raw_byte_send_ints(
             [self.scanning_points, self.config.integration_periods, self.config.use_ema]
         )
-        return self._get_response()
+        return self._get_response(self.SEND_SETTINGS_COMMAND)
 
     def write_list(self: _LeAmpCom) -> str:
         self._encode_send_response(self.SEND_LIST_COMMAND)
         self._raw_byte_send_floats(self.scanning_list)
-        return self._get_response()
+        return self._get_response(self.SEND_LIST_COMMAND)
 
     def start_scan(self: _LeAmpCom) -> tuple[str, np.ndarray, np.ndarray, np.ndarray]:
         self._encode_send_response(self.START_COMMAND)
@@ -301,7 +308,7 @@ class _LeAmpCom:
 
     def _encode_send_response(self: _LeAmpCom, command: str) -> str:
         self._encode_and_send(command)
-        return self._get_response()
+        return self._get_response(command)
 
     def _encode_and_send(self: _LeAmpCom, command: str) -> None:
         self.__ser.write(command.encode(self.ENCODING))
@@ -326,9 +333,19 @@ class _LeAmpCom:
             time.sleep(self.config._sweep_length_ms * 1e-3 * 0.01)  # noqa: SLF001, access to private attribute for backwards compatibility
             status = self._get_status()
 
-    @_BackoffRetry(backoff_base=0.05, logger=logging.getLogger(LOGGER_NAME))
-    def _get_response(self: _LeAmpCom) -> str:
-        return self.__ser.read_until().decode(self.ENCODING).strip()
+    @_BackoffRetry(
+        backoff_base=1e-2, max_tries=3, logger=logging.getLogger(LOGGER_NAME)
+    )
+    def _get_response(self: _LeAmpCom, command: str) -> str:
+        response = self.__ser.read_until().decode(self.ENCODING).strip()
+
+        if len(response) == 0:
+            msg = f"Command: '{command}'. Empty response received"
+            raise serialutil.SerialException(msg)
+        if response[: len(self.OK_RESPONSE)] != self.OK_RESPONSE:
+            msg = f"Command: '{command}'. Expected response '{self.OK_RESPONSE}', received: '{response}'"
+            raise DeviceComError(msg)
+        return response
 
     @_BackoffRetry(
         backoff_base=1e-2, max_tries=5, logger=logging.getLogger(LOGGER_NAME)

pyglaze/devtools/mock_device.py

@@ -26,6 +26,8 @@ class MockDevice(ABC):
         self: MockDevice,
         fail_after: float = np.inf,
         n_fails: float = np.inf,
+        *,
+        empty_responses: bool = False,
     ) -> None:
         pass
 
@@ -183,7 +185,11 @@ class LeMockDevice(MockDevice):
     DAC_BITWIDTH = 2**12
 
     def __init__(
-        self: LeMockDevice, fail_after: float = np.inf, n_fails: float = np.inf
+        self: LeMockDevice,
+        fail_after: float = np.inf,
+        n_fails: float = np.inf,
+        *,
+        empty_responses: bool = False,
     ) -> None:
         self.fail_after = fail_after
         self.fails_wanted = n_fails
@@ -197,6 +203,7 @@ class LeMockDevice(MockDevice):
         self.use_ema: bool | None = None
         self.scanning_list: list[float] | None = None
         self._scan_start_time: float | None = None
+        self.empty_responses = empty_responses
 
     def write(self: LeMockDevice, input_bytes: bytes) -> None:
         """Mock-write to the serial connection."""
@@ -217,12 +224,16 @@ class LeMockDevice(MockDevice):
 
     def read(self: LeMockDevice, size: int) -> bytes:
         """Mock-read from the serial connection."""
+        if self.empty_responses:
+            return self._create_scan_bytes(n_bytes=0)
         if self.state == _LeMockState.IDLE:
             return self._create_scan_bytes(n_bytes=size)
         raise NotImplementedError
 
     def read_until(self: LeMockDevice, _: bytes = b"\r") -> bytes:  # noqa: PLR0911
         """Mock-read_until from the serial connection."""
+        if self.empty_responses:
+            return "".encode(self.ENCODING)
         if self.state == _LeMockState.WAITING_FOR_SETTINGS:
             return "ACK: Ready to receive settings.".encode(self.ENCODING)
         if self.state == _LeMockState.RECEIVED_SETTINGS:
@@ -353,6 +364,7 @@ def list_mock_devices() -> list[str]:
         "mock_device",
         "mock_device_scan_should_fail",
         "mock_device_fail_first_scan",
+        "mock_device_empty_responses",
     ]
 
 
@@ -364,6 +376,8 @@ def _mock_device_factory(config: DeviceConfiguration) -> MockDevice:
         return mock_class()
     if config.amp_port == "mock_device_fail_first_scan":
         return mock_class(fail_after=0, n_fails=1)
+    if config.amp_port == "mock_device_empty_responses":
+        return mock_class(empty_responses=True)
 
     msg = f"Unknown mock device requested: {config.amp_port}. Valid options are: {list_mock_devices()}"
     raise ValueError(msg)

pyglaze/scanning/_asyncscanner.py

@@ -95,13 +95,15 @@ class _AsyncScanner:
     def _get_scan(self: _AsyncScanner) -> _TimestampedWaveform:
         try:
             return self._shared_mem.get(timeout=self._SCAN_TIMEOUT)
-        except Empty as err:
+        except Exception as err:
+            scanner_err: Exception | None = None
             if self._scanner_conn.poll(timeout=self.startup_timeout):
                 msg: _ScannerHealth = self._scanner_conn.recv()
-                if not msg.is_alive:
-                    self.is_scanning = False
                 if msg.error:
-                    raise msg.error from err
+                    scanner_err = msg.error
+            self.stop_scan()
+            if scanner_err:
+                raise scanner_err from err
             raise
 
     @staticmethod
@@ -121,7 +123,7 @@ class _AsyncScanner:
         while not stop_signal.is_set():
             try:
                 waveform = _TimestampedWaveform(datetime.now(), scanner.scan())  # noqa: DTZ005
-            except (serialutil.SerialException, TimeoutError) as e:
+            except Exception as e:  # noqa: BLE001
                 parent_conn.send(
                     _ScannerHealth(is_alive=False, is_healthy=False, error=e)
                 )

{pyglaze-0.2.0.dist-info → pyglaze-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyglaze
-Version: 0.2.0
+Version: 0.2.1
 Summary: Pyglaze is a library used to operate the devices of Glaze Technologies
 Author: GLAZE Technologies ApS
 License: BSD 3-Clause License

{pyglaze-0.2.0.dist-info → pyglaze-0.2.1.dist-info}/RECORD

@@ -1,13 +1,13 @@
-pyglaze/__init__.py,sha256=Zn1KFblwuFHiDRdRAiRnDBRkbPttWh44jKa5zG2ov0E,22
+pyglaze/__init__.py,sha256=HfjVOrpTnmZ-xVFCYSVmX50EXaBQeJteUHG-PD6iQs8,22
 pyglaze/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyglaze/datamodels/__init__.py,sha256=DJLByl2C7pC4RM4Uh6PW-McM5RIGBjcopzGywCKhSlI,111
 pyglaze/datamodels/pulse.py,sha256=BoW_GDvkwEpn_UYUFPoUfi5E_oCnvv8cU4nZKG7Ph2U,20911
 pyglaze/datamodels/waveform.py,sha256=n31DhJHFeBNZ3hHqQUiCGXssm5Dc8wV6tGPkhmFYB4Q,5809
 pyglaze/device/__init__.py,sha256=5RjCHuFKMi9g2KLUkxixO9hNpAgkUBcOURNTuhAdoUk,177
-pyglaze/device/ampcom.py,sha256=mtsprV51UeRwqkvbbLRbO2WeRd2XdFcU4_Q8AMul28g,16523
+pyglaze/device/ampcom.py,sha256=6JXl7PojYr4F1_pdVItO-XbGCp8mpy4wpwaaUn8Loog,17214
 pyglaze/device/configuration.py,sha256=gh_eerX8TdXx3LnFxHieJqOpfDfE9cV6Xgm5WYVnvO0,7994
 pyglaze/devtools/__init__.py,sha256=9EW20idoaZv_5GuSgDmfpTPjfCZ-Rl27EV3oJebmwnQ,90
-pyglaze/devtools/mock_device.py,sha256=XJ1jwOOYPDqynX3QDYfEJk9n3sTMmefNPAe9N3M1T5Q,13522
+pyglaze/devtools/mock_device.py,sha256=3RMa-JAehbbRCmOO74aZzGPdYp2qsP-XE8ulzLXDP6Q,13994
 pyglaze/devtools/thz_pulse.py,sha256=xp-T9psdOrUMtSUFu8HEwQJVu_aMixJdZHtg_BCVu_k,923
 pyglaze/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyglaze/helpers/types.py,sha256=p9xSAP5Trr1FcCWl7ynCWqDOUZKgMQYzMUXSwDpAKHg,599
@@ -15,11 +15,11 @@ pyglaze/helpers/utilities.py,sha256=n_x9Tqm305MUorS29O6CoJM8Mi4apo2bsN_odrRaVAw,
 pyglaze/interpolation/__init__.py,sha256=WCxHPsiI7zvJykp-jfytoEbO4Tla-YIF6A7fjDfcDvU,72
 pyglaze/interpolation/interpolation.py,sha256=rQWzPD7W8TXETps7VZI0gcfAOCWO8pGL1HhhBnyxaMw,735
 pyglaze/scanning/__init__.py,sha256=uCBaeDTufOrC9KWf30ICqcmvFg_YT85olb3M9jkvZRg,99
-pyglaze/scanning/_asyncscanner.py,sha256=n4PhRBSZLFSSeGgBkgVwFsdY4rnduXJnQo6TQ0ujVkw,5224
+pyglaze/scanning/_asyncscanner.py,sha256=SldM7XCavfugsDstxMVI_WLL6GUJutLYN0iU__mfY5A,5277
 pyglaze/scanning/client.py,sha256=3qrQStkeLQzCeu4yMHJ_ENLGQ7E5GMc4CP9J55rk-ug,1817
 pyglaze/scanning/scanner.py,sha256=PSjXVpSpHpYIl-sW34pIThocbM9GSHJ_E4gGcsePeTw,8139
-pyglaze-0.2.0.dist-info/LICENSE,sha256=LCP3sGBX7LxuQopcjeug1fW4tngWCHF4zB7QCgB28xM,1504
-pyglaze-0.2.0.dist-info/METADATA,sha256=ZWjcLSzB9CD-_MmxsXdaX2tEcQbia3zLTH2kp5cnp1E,3498
-pyglaze-0.2.0.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
-pyglaze-0.2.0.dist-info/top_level.txt,sha256=X7d5rqVVuWNmtK4-Uh4sgOLlqye8vaHZOr5RYba0REo,8
-pyglaze-0.2.0.dist-info/RECORD,,
+pyglaze-0.2.1.dist-info/LICENSE,sha256=LCP3sGBX7LxuQopcjeug1fW4tngWCHF4zB7QCgB28xM,1504
+pyglaze-0.2.1.dist-info/METADATA,sha256=WETmnm6CF1_eyojd-ahfIFLJ_VCxV2jshcrbSoPLg3Y,3498
+pyglaze-0.2.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+pyglaze-0.2.1.dist-info/top_level.txt,sha256=X7d5rqVVuWNmtK4-Uh4sgOLlqye8vaHZOr5RYba0REo,8
+pyglaze-0.2.1.dist-info/RECORD,,

{pyglaze-0.2.0.dist-info → pyglaze-0.2.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.2.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any