diff options
Diffstat (limited to 'rebuilder.py')
-rw-r--r-- | rebuilder.py | 8 |
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: |