[Python] AST(抽象構文木)からコードを生成する

pip install astor

import ast

input_source = """

def hoge(a,b=1):
  return a+b

a = 1

print(hoge(a))
"""

ast_object = ast.parse(input_source)

print(ast.dump(ast.object))

Module(
    body=[
        FunctionDef(
            name='hoge', 
            args=arguments(
                args=[
                    arg(arg='a', annotation=None),
                    arg(arg='b', annotation=None)
                ],
                vararg=None,
                kwonlyargs=[], 
                kw_defaults=[], 
                kwarg=None, 
                defaults=[Num(n=1)]
            ), 
            body=[Return(
                        value=BinOp(
                            left=Name(id='a', ctx=Load()), 
                            op=Add(), 
                            right=Name(id='b', ctx=Load())
                        )
                    )],
            decorator_list=[], 
            returns=None
        ),
        Assign(
            targets=[Name(id='a', ctx=Store())], 
            value=Num(n=1)
        ), 
        Expr(
            value=Call(
                func=Name(id='print', ctx=Load()), 
                args=[
                    Call(
                        func=Name(id='hoge', ctx=Load()), 
                        args=[Name(id='a', ctx=Load())], 
                        keywords=[]
                    )
                ], 
                keywords=[]
            )
        )
    ]
)

import astor

source = astor.to_source(ast_object)

print(source)

def hoge(a, b=1):
    return a + b


a = 1
print(hoge(a))