def visitForBlock(self, element: Element) -> ResultType:
"""
Handle for loop block.
"""
Error.assertHasAttr(element=element, attr="value1")
Error.assertHasAttr(element=element, attr="iterable")
Error.assertHasSequence(element=element, sequence="nested")
value1 = element.getAttr("value1").value
value2 = element.getAttrValue("value2")
sequence = element.getNestedSequence(kind="nested")
assert sequence
block: ResultType = []
# Loop through the elements
iterable = self.resolveName(name=element.getAttr("iterable").value)
if value2 is None:
for value in iterable:
self.substitutions.register(element=element,
key=value1,
value=value)
block += self._visit(sequence=sequence)
self.substitutions.unregister(value1)
else:
iterablePair = iterable.items() if isinstance(
iterable, dict) else enumerate(iterable)
for key, value in iterablePair:
self.substitutions.register(element=element,
key=value1,
value=key)
self.substitutions.register(element=element,
key=value2,
value=value)
block += self._visit(sequence=sequence)
self.substitutions.unregister(value2)
self.substitutions.unregister(value1)
return block
def visitMacro(self, element: Element) -> None:
"""
Handle macro definition block.
"""
Error.assertHasSequence(element=element, sequence="argument")
Error.assertHasSequence(element=element, sequence="nested")
Error.assertHasAttr(element=element, attr="name")
name = element.getAttr("name").value
Error.assertTrue(
element=element,
attr="name",
condition=(name not in self.substitutions),
message=
"Name conflict with macro and an already existing name: '{}'.".
format(name))
# Register the macro
self.substitutions.register(
element=element,
key=name,
value=lambda *args: self.processMacro(element, *args))
def __init__(self, element: Element) -> None:
super().__init__(element, Role.Type)
Error.assertHasAttr(element=element, attr="name")
Error.assertHasSequence(element=element, sequence="values")
