diff options
Diffstat (limited to 'node_classes.py')
| -rw-r--r-- | node_classes.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/node_classes.py b/node_classes.py index 2ddda5bf..3715a09d 100644 --- a/node_classes.py +++ b/node_classes.py @@ -200,6 +200,12 @@ class Compare(NodeNG): for _, comparator in self.ops: yield comparator # we don't want the 'op' + def last_child(self): + """override last_child""" + # XXX maybe if self.ops: + return self.ops[-1][1] + #return self.left + class Comprehension(NodeNG): """class representing a Comprehension node""" @@ -256,6 +262,12 @@ class Dict(NodeNG, Instance): yield key yield value + def last_child(self): + """override last_child""" + if self.items: + return self.items[-1][1] + return None + def itered(self): return self.items[::2] |
