modify_ast.py

modify_ast.py

import ast

code = '''
def add(n1, n2):
    return n1 + n2
print(add(3, 4))
'''

class CrazyTransformer(ast.NodeTransformer):
    def visit_BinOp(self, node):
        node.op = ast.Mult()
        return node

node = ast.parse(code)
modified = CrazyTransformer().visit(node)
exec(compile(modified, '<string>', 'exec'))