Programmation PYTHON - Zenk - Security - Repository

from compiler.visitor import ASTVisitor
from compiler.ast import Stmt, Class, Function
class DisplayVisitor(ASTVisitor):
"""Visite l'AST"""
def __init__(self, name):
self.name = name
Discussion
Exercices corrigés
CHAPITRE 11
def _visit_node(self, node):
"""Appelle sur les nodes"""
if (isinstance(node, Stmt) or
(hasattr(node, 'name') and node.name != self.name)):
for subnode in node.getChildNodes():
self._visit_node(subnode)
return
if not (isinstance(node, Class) or isinstance(node, Function)):
return
print('Trouvé ligne %d' % node.lineno)
if node.doc is None:
print('\n\tAucun docstring\n')
else:
print('\n\t%s\n' % node.doc)
visitClass = _visit_node
visitFunction = _visit_node
visitStmt = _visit_node
def print_module(path, element, verbose=False):
"""Permet l'affichage d'un doctstring de classe ou de fonction"""
with open(path) as f:
ast = parse(f.read())
walk(ast, DisplayVisitor(element))
if __name__ == '__main__':
print_module(sys.argv[1], sys.argv[2])
Grâce à sa fonction parse, le paquet compiler construit un AST (Abstract Syntax
Tree) à partir de code Python. Très rapide, cette opération lit le contenu d’un module
sans avoir à l’importer.
walk offre ensuite la possibilité de traverser l’AST, en fournissant une classe qui dérive de
la classe compiler.visitor.ASTVisitor. Chaque nœud de l’AST est passé à la méthode
visitTypeNoeud de la classe si elle existe, où TypeNoeud est le type de nœud visité.
383