Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/gast/astn.py: 88%

1import ast 

2import gast 

3 

4 

5def _generate_translators(to): 

6 

7 class Translator(ast.NodeTransformer): 

8 

9 def _visit(self, node): 

10 if isinstance(node, ast.AST): 

11 return self.visit(node) 

12 elif isinstance(node, list): 

13 return [self._visit(n) for n in node] 

14 else: 

15 return node 

16 

17 def generic_visit(self, node): 

18 class_name = type(node).__name__ 

19 if not hasattr(to, class_name): 

20 # handle nodes that are not part of the AST 

21 return 

22 cls = getattr(to, class_name) 

23 new_node = cls( 

24 **{ 

25 field: self._visit(getattr(node, field)) 

26 for field in node._fields 

27 } 

28 ) 

29 

30 for attr in node._attributes: 

31 try: 

32 setattr(new_node, attr, getattr(node, attr)) 

33 except AttributeError: 

34 pass 

35 return new_node 

36 

37 return Translator 

38 

39 

40AstToGAst = _generate_translators(gast) 

41 

42GAstToAst = _generate_translators(ast)