diff --git a/src/aare/common/exception_handler.py b/src/aare/common/exception_handler.py index e786cc83..f454c357 100644 --- a/src/aare/common/exception_handler.py +++ b/src/aare/common/exception_handler.py @@ -17,6 +17,24 @@ class TransformationInvalidException(Exception): return self.message +class DataCollectionException(Exception): + def __init__(self, message: str = "Data collection failed"): + super().__init__(message) + self.message = message + + def __str__(self) -> str: + return self.message + + +class RasterScanException(Exception): + def __init__(self, message: str = "Data collection failed"): + super().__init__(message) + self.message = message + + def __str__(self) -> str: + return self.message + + class LoopCenteringFailed(Exception): def __init__(self, message: str = "Loop Centering did not detect a sample"): super().__init__(message)