FilesExpand file tree

 master

/

transformers.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

35 lines (29 loc) · 1.1 KB

 master

/

transformers.py

Copy path

Top

File metadata and controls

• Code
• Blame

35 lines (29 loc) · 1.1 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

import ast

class ParentChildNodeTransformer(object):

def visit(self, node):

self._prepare_node(node)

for field, value in ast.iter_fields(node):

self._process_field(node, field, value)

return node

@staticmethod

def _prepare_node(node):

if not hasattr(node, 'parent'):

node.parent = None

if not hasattr(node, 'parents'):

node.parents = []

if not hasattr(node, 'children'):

node.children = []

def _process_field(self, node, field, value):

if isinstance(value, list):

for index, item in enumerate(value):

if isinstance(item, ast.AST):

self._process_child(item, node, field, index)

elif isinstance(value, ast.AST):

self._process_child(value, node, field)

def _process_child(self, child, parent, field_name, index=None):

self.visit(child)

child.parent = parent

child.parents.append(parent)

child.parent_field = field_name

child.parent_field_index = index

child.parent.children.append(child)