From 8fca6bed113e0ea62190310d05e2e34c1e81b62e Mon Sep 17 00:00:00 2001 From: appel_c Date: Wed, 22 Apr 2026 17:17:35 +0200 Subject: [PATCH] fix: Add logs for simulated DDG --- csaxs_bec/devices/sim/simulated_beamline_devices.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/csaxs_bec/devices/sim/simulated_beamline_devices.py b/csaxs_bec/devices/sim/simulated_beamline_devices.py index 0e5e479..d359d6f 100644 --- a/csaxs_bec/devices/sim/simulated_beamline_devices.py +++ b/csaxs_bec/devices/sim/simulated_beamline_devices.py @@ -2,6 +2,7 @@ from __future__ import annotations +from bec_lib.logger import bec_logger from ophyd import Device from ophyd_devices import StatusBase from ophyd_devices.tests.utils import patched_device @@ -9,6 +10,8 @@ from ophyd_devices.tests.utils import patched_device from csaxs_bec.devices.epics.delay_generator_csaxs.ddg_1 import DDG1 from csaxs_bec.devices.epics.fast_shutter import cSAXSFastEpicsShutter +logger = bec_logger.logger + class cSAXSSimulatedFastShutter(Device): """ @@ -32,6 +35,12 @@ class DDG1WithPatchedStage(DDG1): We inherit here to be able to overwrite the on_stage method if needed, without compromising the rest of the device's functionality. """ + def on_stage(self): + logger.warning( + f"Staging of {self.name}. This is a simulated device of {self.__class__.__name__}." + ) + super().on_stage() + def on_trigger(self): """ We overwrite and patch the on_trigger method. This resolves immediately now and @@ -46,5 +55,5 @@ class DDG1WithPatchedStage(DDG1): class SimulatedDDG1(Device): def __new__(cls, *args, **kwargs): - with patched_device(DDG1, *args, **kwargs) as ddg1: + with patched_device(DDG1WithPatchedStage, *args, **kwargs) as ddg1: return ddg1