Skip to content

Commit

Permalink
cleaned up after rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
evalott100 committed Apr 17, 2024
1 parent c4d8e04 commit 293cf31
Showing 1 changed file with 5 additions and 8 deletions.
13 changes: 5 additions & 8 deletions src/ophyd_async/epics/pvi/pvi.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ def _split_subscript(tp: T) -> Union[Tuple[Any, Tuple[Any]], Tuple[T, None]]:


def _strip_union(field: Union[Union[T], T]) -> T:
is_optional = None
if get_origin(field) is Union:
args = get_args(field)
is_optional = type(None) in args
Expand Down Expand Up @@ -130,7 +131,7 @@ def _parse_type(
):
if common_device_type:
# pre-defined type
device_cls = _strip_union(common_device_type)
device_cls, _ = _strip_union(common_device_type)
is_device_vector, device_cls = _strip_device_vector(device_cls)
device_cls, device_args = _split_subscript(device_cls)
assert issubclass(device_cls, Device)
Expand Down Expand Up @@ -164,7 +165,7 @@ def _sim_common_blocks(device: Device, stripped_type: Optional[Type] = None):
)

for device_name, device_cls in sub_devices:
device_cls = _strip_union(device_cls)
device_cls, _ = _strip_union(device_cls)
is_device_vector, device_cls = _strip_device_vector(device_cls)
device_cls, device_args = _split_subscript(device_cls)
assert issubclass(device_cls, Device)
Expand All @@ -189,8 +190,7 @@ def _sim_common_blocks(device: Device, stripped_type: Optional[Type] = None):
if is_signal:
sub_device = device_cls(SimSignalBackend(signal_dtype, device_name))
else:
sub_device = device_cls()

sub_device = getattr(device, device_name, device_cls())
_sim_common_blocks(sub_device, stripped_type=device_cls)

setattr(device, device_name, sub_device)
Expand Down Expand Up @@ -225,10 +225,7 @@ async def _get_pvi_entries(entry: PVIEntry, timeout=DEFAULT_TIMEOUT):
if is_signal:
device = _pvi_mapping[frozenset(pva_entries.keys())](signal_dtype, *pvs)
else:
if hasattr(entry.device, sub_name):
device = getattr(entry.device, sub_name)
else:
device = device_type()
device = getattr(entry.device, sub_name, device_type())

sub_entry = PVIEntry(
device=device, common_device_type=device_type, sub_entries={}
Expand Down

0 comments on commit 293cf31

Please sign in to comment.