Skip to content

Commit 593baa9

Browse files
committed
Fix enumeration of device register attributes
1 parent 62b562c commit 593baa9

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

harp/reader.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,12 @@ def __init__(self, model: Model, registers: dict[str, RegisterReader]) -> None:
2525
self.model = model
2626
self.registers = registers
2727

28+
def __dir__(self) -> Iterable[str]:
29+
return self.registers.keys()
30+
31+
def __getattr__(self, __name: str) -> RegisterReader:
32+
return self.registers[__name]
33+
2834

2935
def compose(f, g):
3036
return lambda *a, **kw: f(g(*a, **kw))
@@ -44,7 +50,7 @@ def create_bitreader(mask):
4450

4551
def create_reader(device: Model):
4652
reg_readers = {
47-
create_register_reader(device, name) for name in device.registers.keys()
53+
name: create_register_reader(device, name) for name in device.registers.keys()
4854
}
4955
return DeviceReader(device, reg_readers)
5056

0 commit comments

Comments
 (0)