summaryrefslogtreecommitdiff
path: root/node_classes.py
diff options
context:
space:
mode:
Diffstat (limited to 'node_classes.py')
-rw-r--r--node_classes.py12
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]