def decorate_ast_with_source(_ast, code): class MyVisitor(ast.NodeVisitor): def visit(self, node): self.generic_visit(node) node.source_code = code MyVisitor().visit(_ast)