summaryrefslogtreecommitdiff
path: root/rebuilder.py
diff options
context:
space:
mode:
Diffstat (limited to 'rebuilder.py')
-rw-r--r--rebuilder.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/rebuilder.py b/rebuilder.py
index d87eb09f..5378eb45 100644
--- a/rebuilder.py
+++ b/rebuilder.py
@@ -242,6 +242,14 @@ class RebuildVisitor(ASTVisitor):
const.parent = node
node.locals['__path__'] = [const]
+ def visit_const(self, node):
+ """visit an Name node to become astng"""
+ try:
+ cls, value = nodes.CONST_VALUE_TRANSFORMS[node.value]
+ node.__class__ = cls
+ except KeyError:
+ pass
+
def visit_name(self, node):
"""visit an Name node to become astng"""
try: