Add lookup expressions

This commit is contained in:
James Westman 2022-01-29 21:21:04 -06:00
parent c094743e84
commit 4fefa0bd73
No known key found for this signature in database
GPG key ID: CE2DBA0ADB654EA6
8 changed files with 177 additions and 5 deletions

View file

@ -33,7 +33,10 @@ class Children:
def __iter__(self):
return iter(self._children)
def __getitem__(self, key):
return [child for child in self._children if isinstance(child, key)]
if isinstance(key, int):
return self._children[key]
else:
return [child for child in self._children if isinstance(child, key)]
class AstNode: