在 Python 中将节点添加到 AST 时，函数未添加到新行

Question

在Python中，我尝试使用AST在源代码中添加一个循环后的打印语句。但是，问题在于，打印说明未添加到新线路中，而是在与循环的同一行中添加。添加fix_missing_locations()和increment_lineno()的各种组合无济于事。我在做什么错？

import astor
import ast
class CodeInstrumentator(ast.NodeTransformer):
    def get_print_stmt(self, lineno):
        return ast.Call(
            func=ast.Name(id='print', ctx=ast.Load()),
            args=[ast.Num(n=lineno)],
            keywords=[]
            )
    def insert_print(self, node):
        node.body.insert(0, self.get_print_stmt(node.lineno))
    def visit_For(self, node):
        self.insert_print(node)
        self.generic_visit(node)
        return node
def main():
    input_file = 'source.py'
    try:
        myAST = astor.parsefile(input_file)
    except Exception as e:
        raise e
    CodeInstrumentator().visit(myAST)
    instru_source = astor.to_source(myAST)
    source_file = open('test.py', 'w')
    source_file.write(instru_source)
if __name__ == "__main__":
    main()

Answer 1

这个问题似乎被我面临类似的问题所抛弃，我终于找到了解决方案，所以我写下来，以防万一对某人有用。

首先，请注意，ASTOR不依赖也不依赖lineno和col_offset，因此使用ast.fix_missing_locations(node)，increment_lineno(node, n=1)或new_node = ast.copy_location(new_node, node)不会对输出代码产生任何影响。

这样说，问题是Call语句不是独立操作，因此，Astor将其应用于上一个节点(因为它是同一操作的一部分，但您错过了写入节点的lineno(。

然后，解决方案是使用Expr语句用void调用包装Call语句：

def get_print_stmt(self, lineno):
    return ast.Expr(value=ast.Call(
        func=ast.Name(id='print', ctx=ast.Load()),
        args=[ast.Num(n=lineno)],
        keywords=[]
        ))

如果将包含void调用的代码写入函数，您会注意到其AST表示已包含Expr节点：

test_file.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# MAIN
#
my_func()

process_file.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
from __future__ import print_function
def main():
    tree = astor.code_to_ast.parse_file("test_file.py")
    print("DUMP TREE")
    print(astor.dump_tree(tree))
    print("SOURCE")
    print(astor.to_source(tree))
#
# MAIN
#
if __name__ == '__main__':
    main()

输出

$ python process_file.py
DUMP TREE
Module(
    body=[
        Expr(value=Call(func=Name(id='my_func'), args=[], keywords=[], starargs=None, kwargs=None))])
SOURCE
my_func()