diff --git a/src/aare/common/exception_handler.py b/src/aare/common/exception_handler.py index 7012fc78..08abf5bd 100644 --- a/src/aare/common/exception_handler.py +++ b/src/aare/common/exception_handler.py @@ -296,5 +296,17 @@ class AerotechCommunicationError(Exception): }, ) + def __str__(self) -> str: + return self.message + +class MagnetPositionSensorErorr(Exception): + def __init__(self, message: str = "Magnet position sensor error"): + super().__init__(message) + self.message = message + logger.error( + message, + extra={"device": "magnet_position_sensor"}, + ) + def __str__(self) -> str: return self.message \ No newline at end of file