2012-04-06 3 views
3

Возможно ли провести обход после экземпляра ast.NodeVisitor в Python, просто манипулируя ast.NodeVisitor.generic_visit()? Я сделал это:Постоперационный обход на ast.nodevisitor в Python

class ExpParser(ast.NodeVisitor): 

    def generic_visit(self, node): 
     for x in ast.iter_child_nodes(node): 
      ast.NodeVisitor.generic_visit(self, x) 
     ast.NodeVisitor.generic_visit(self, node) 

    def visit_BinOp(self, node): 
     print type(node.op).__name__ 

    def visit_Name(self, node): 
     print node.id 

if __name__ == '__main__': 
    node = ast.parse("T1+T2*T3") 
    v = ExpParser() 
    v.visit(node) 

это дало мне:

T1 
T2 
T3 
Mult 
Add 

Я хочу, чтобы дать мне:

T2 
T3 
Mult 
T1 
Add 

Как я могу это сделать? пожалуйста, я застрял.

+0

Я заметил, что устаревший пакет 'compiler.visitor' имел это средство, но похоже, что в пакете' ast' нет эквивалента. – snim2

ответ

3

попробовать что-то вроде этого

import ast 

class ExpParser(ast.NodeVisitor): 

    def generic_visit(self, node): 
     for field, value in reversed(list(ast.iter_fields(node))): 
      if isinstance(value, list): 
       for item in value: 
        if isinstance(item, ast.AST): 
         self.visit(item) 
      elif isinstance(value, ast.AST): 
       self.visit(value) 

    def visit_BinOp(self, node): 
     self.generic_visit(node) 
     print type(node.op).__name__ 

    def visit_Name(self, node): 
     self.generic_visit(node) 
     print node.id 

if __name__ == '__main__': 
    node = ast.parse("T1+T2*T3") 
    v = ExpParser() 
    v.visit(node) 

не забудьте позвонить generic_visit по каждому методу посещения, или в качестве альтернативы переописать метод посещения, чтобы заботиться о нем автоматически.