astmonkey/examples/parent_node_transformer.py

9 lines
260 B
Python

import ast
from astmonkey import transformers
node = ast.parse('x = 1')
node = transformers.ParentChildNodeTransformer().visit(node)
assert(node == node.body[0].parent)
assert(node.body[0].parent_field == 'body')
assert(node.body[0].parent_field_index == 0)