Visit Python Abstract Syntax Tree

maxme (Maxime Biais) — Wed, 24 Jan 2007 13:03:54 GMT

Simplest AST visitor. More on this blog post :
http://www.biais.org/blog/index.php/2007/01/10/9-visit-python-abstract-syntax-tree



import compiler

 

class CodePrinter:

    def __init__(self):

        self.src = ''

 

    def visitName(self,t):

        self.src += t.name

 

    def visitConst(self,t):

        self.src += str(t.value)

 

    def visitStmt(self, t):

        for i in t:

            a = pretty_print(i)

            self.src += a + "\n"

 

    def visitAssName(self, t):

        self.src += t.name + " = "

 

def pretty_print(node):

    myvisitor = CodePrinter()

    # compiler.walk return the visitor instance : 2nd arg

    return compiler.walk(node, myvisitor).src

DZone Snippets: abstract code

Visit Python Abstract Syntax Tree