def main(infilename, outfilename):
node = ast.parse(open(infilename).read())
node = transformers.ParentChildNodeTransformer().visit(node)
visitor = visitors.GraphNodeVisitor()
visitor.visit(node)
visitor.graph.write_png(outfilename)
os.system('open ' + outfilename)
def ast_playing(source_code):
node = ast.parse(source_code)
node = transformers.ParentChildNodeTransformer().visit(node)
visitor = visitors.GraphNodeVisitor()
visitor.visit(node)
for n in ast.walk(node):
print(type(n))
visitor.graph.write_png('graph_1.png')
def __tests():
import ast
with open('tests/function_def_test.py') as test_code:
test_node = ast.parse(test_code.read())
node = transformers.ParentNodeTransformer().visit(test_node)
visitor = visitors.GraphNodeVisitor()
visitor.visit(node)
visitor.graph.write_png('graph.png')
def dump_ast(mod, name):
if IMPORT_ASTMONKEY:
mod = deepcopy(mod)
mod = transformers.ParentChildNodeTransformer().visit(deepcopy(mod))
visitor = visitors.GraphNodeVisitor()
visitor.visit(mod)
visitor.graph.write_png(name + '.png')
print("\033[1;32;40mAST visualization saved as \033[94m%s.png\033[0m" % name)
else:
print("\033[93mInstall astmonkey for visualization.\033[0m")
def main():
code = ''
for line in sys.stdin:
code += line
print(line)
node = ast.parse(code)
node = transformers.ParentChildNodeTransformer().visit(node)
visitor = visitors.GraphNodeVisitor()
visitor.visit(node)
visitor.graph.write_png('src/main/resources/graph.png')
def setUp(self):
self.visitor = visitors.GraphNodeVisitor()
def visitor(self):
return visitors.GraphNodeVisitor()
import ast
from astmonkey import visitors, transformers
node = ast.parse('def foo(x):\n\treturn x + 1')
node = transformers.ParentNodeTransformer().visit(node)
visitor = visitors.GraphNodeVisitor()
visitor.visit(node)
visitor.graph.write_png('graph.png')