convert AST node to python code

Question

Suppose I have the following string:

code = """
if 1 == 1 and 2 == 2 and 3 == 3:
    test = 1
"""

The following code converts that string in a AST.

ast.parse(code)

Then I have a tree like:

Module(body=[<_ast.If object at 0x100747358>])
  If(test=BoolOp(op=And(), values=[<_ast.Compare object at 0x100747438>, <_ast.Compare object at 0x100747a90>, <_ast.Compare object at 0x100747d68>]), body=[<_ast.Assign object at 0x100747e48>], orelse=[])

I want to know if there is a way to convert the object at.If into the string if 1 == 1 and 2 == 2 and 3 == 3:

I know it can be done traversing the children node, but it's getting too complicated that way.

traversing the children node, <- that's the only way I know how to do it — inspectorG4dget
– inspectorG4dget, Commented Oct 13, 2016 at 20:06
There are 3rd party tools that make a credible attempt, e.g. astor — AChampion
– AChampion, Commented Sep 18, 2017 at 2:44

Kalle · Accepted Answer · 2021-06-23 15:01:11Z

18

Python 3.9 introduced ast.unparse which does exactly this, i.e. it reverses ast.parse. Using your example:

import ast

code = """
if 1 == 1 and 2 == 2 and 3 == 3:
    test = 1
"""

tree = ast.parse(code)
print(ast.unparse(tree))

This will print out:

if 1 == 1 and 2 == 2 and (3 == 3):
    test = 1

Note that there may be slight differences to the original input.

answered Jun 23, 2021 at 15:01

Kalle

1,4631 gold badge12 silver badges7 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Blane · Accepted Answer · 2020-06-28 15:53:00Z

12

ast.get_source_segment was added in python 3.8:

>>> import ast

>>> code = """
>>> if 1 == 1 and 2 == 2 and 3 == 3:
>>>     test = 1
>>> """
>>> node = ast.parse(code)
>>> ast.get_source_segment(code, node.body[0])
'if 1 == 1 and 2 == 2 and 3 == 3:\n    test = 1'

answered Jun 28, 2020 at 15:53

Blane

6738 silver badges13 bronze badges

Comments

Anentropic · Accepted Answer · 2020-02-24 14:13:43Z

7

You can use the astunparse library which was basically just code from the core, repackaged individually.

First, install the library:

pip install astunparse

Then, run your AST module through it to get the source right back. So running:

import ast
import astunparse

code = """
if 1 == 1 and 2 == 2 and 3 == 3:
    test = 1
"""

node = ast.parse(code)

astunparse.unparse(node)

Will output:

'\nif ((1 == 1) and (2 == 2) and (3 == 3)):\n    test = 1\n'

edited Feb 24, 2020 at 14:13

Anentropic

34.2k13 gold badges111 silver badges170 bronze badges

answered Sep 18, 2017 at 2:44

noamt

7,8854 gold badges44 silver badges63 bronze badges

Collectives™ on Stack Overflow

convert AST node to python code

3 Answers 3

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related