From b328f064f2aa860e5a96a3c1a94a6bf43f80e2bf Mon Sep 17 00:00:00 2001 From: appel_c Date: Thu, 5 Jun 2025 14:35:57 +0200 Subject: [PATCH] fix(psi-device-base): add on_destroy hook to psi-device-base --- ophyd_devices/interfaces/base_classes/psi_device_base.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/ophyd_devices/interfaces/base_classes/psi_device_base.py b/ophyd_devices/interfaces/base_classes/psi_device_base.py index def17ea..8fa6b46 100644 --- a/ophyd_devices/interfaces/base_classes/psi_device_base.py +++ b/ophyd_devices/interfaces/base_classes/psi_device_base.py @@ -156,6 +156,11 @@ class PSIDeviceBase(Device): self.stopped = True # Set stopped flag to True, in case a custom stop method listens to stopped property super().stop(success=success) + def destroy(self): + """Destroy the device.""" + self.on_destroy() # Call the on_destroy method + return super().destroy() + ######################################## # Utility Method to wait for signals # ######################################## @@ -234,3 +239,6 @@ class PSIDeviceBase(Device): def on_stop(self) -> None: """Called when the device is stopped.""" + + def on_destroy(self) -> None: + """Called when the device is destroyed. Cleanup resources here."""