bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2025-06-06 03:40:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor(psi-device-base): remove device_manager from kwargs, improve Status return on stage/unstage

Browse Source
This commit is contained in:
appel_c 2025-05-12 19:23:36 +02:00
parent 673be87a2d
commit a7bffb8e3a
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ophyd_devices/interfaces/base_classes/psi_device_base.py
View File

@ -4,6 +4,7 @@ Base class for all PSI ophyd device integration to ensure consistent configurati
from __future__ import annotations
import inspect
import time
from typing import Any, Callable
@ -44,6 +45,12 @@ class PSIDeviceBase(Device):
name (str) : Name of the device
scan_info (ScanInfo): The scan info to use.
"""
# Make sure device_manager is not passed to super().__init__ if not specified
# This is to avoid issues with ophyd.OphydObject.__init__ when the parent is ophyd.Device
# and the device_manager is passed to it. This will cause a TypeError.
sig = inspect.signature(super().__init__)
if "device_manager" not in sig.parameters:
kwargs.pop("device_manager", None)
super().__init__(prefix=prefix, name=name, **kwargs)
self._stopped = False
self.task_handler = TaskHandler(parent=self)
@ -87,7 +94,7 @@ class PSIDeviceBase(Device):
self.stopped = False
super_staged = super().stage()
status = self.on_stage() # pylint: disable=assignment-from-no-return
if isinstance(status, DeviceStatus):
if isinstance(status, StatusBase):
return status
return super_staged
@ -95,7 +102,7 @@ class PSIDeviceBase(Device):
"""Unstage the device."""
super_unstage = super().unstage()
status = self.on_unstage() # pylint: disable=assignment-from-no-return
if isinstance(status, DeviceStatus):
if isinstance(status, StatusBase):
return status
return super_unstage