usb-tools/amaranth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lib.data: improve diagnostics for field access on array layout view.

Browse source 
Fixes #837.
This commit is contained in:
Catherine 2023-07-18 14:03:17 +00:00
parent ea36c80663
commit 385b10d743
2 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
amaranth/lib/data.py
View file

@ -662,6 +662,10 @@ class View(ValueCastable):
If ``ShapeCastable.__call__`` does not return a value or a value-castable object.
"""
if isinstance(self.__layout, ArrayLayout):
if not isinstance(key, (int, Value, ValueCastable)):
raise TypeError("Views with array layout may only be indexed with an integer "
"or a value, not {!r}"
.format(key))
shape = self.__layout.elem_shape
value = self.__target.word_select(key, Shape.cast(self.__layout.elem_shape).width)
else:
@ -697,6 +701,9 @@ class View(ValueCastable):
If the layout does not define a field called ``name``, or if ``name`` starts with
an underscore.
"""
if isinstance(self.__layout, ArrayLayout):
raise AttributeError("View of {!r} with an array layout does not have fields"
.format(self.__target))
try:
item = self[name]
except KeyError: