2012-04-06 2 views
3

Est-il possible de faire une traversée de post-commande sur une instance de ast.NodeVisitor en Python simplement en manipulant le ast.NodeVisitor.generic_visit()? Je l'ai fait:Parcours de traversier sur un ast.nodevisitor en 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) 

cela m'a donné:

T1 
T2 
T3 
Mult 
Add 

Je veux me donner:

T2 
T3 
Mult 
T1 
Add 

Comment puis-je faire? S'il te plaît, je suis coincé.

+0

Je remarque que le paquet 'compiler.visitor' déprécié avait cette fonctionnalité mais il ne semble pas y avoir d'équivalent dans le paquet' ast'. – snim2

Répondre

3

essayer quelque chose comme ça

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) 

ne pas oublier d'appeler generic_visit sur chaque méthode de visite, ou encore réimplémentez méthode de visite pour prendre soin de automatiquement.

Questions connexes