1  import ast
2  from astmonkey import visitors, transformers
3  
4  node = ast.parse('def foo(x):\n\treturn x + 1')
5  node = transformers.ParentChildNodeTransformer().visit(node)
6  visitor = visitors.GraphNodeVisitor()
7  visitor.visit(node)
8  
9  visitor.graph.write_png('graph.png')