我尝试使用AST在源代码中的每个for循环之后添加一个print语句。但问题是,print语句没有添加到新行,而是添加到同一行,就像for循环一样。添加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()
答案 0 :(得分:0)
这个问题似乎因为我遇到了类似的问题而放弃了,我终于找到了一个解决方案,所以我写下来以防它对某人有用。
首先,请注意ASTOR不依赖lineno
也不依赖col_offset
因此使用ast.fix_missing_locations(node)
,increment_lineno(node, n=1)
或new_node = ast.copy_location(new_node, node)
不会产生任何影响在输出代码上。
这就是说,问题是Call
语句不是一个独立的操作,因此,ASTOR将它应用到前一个节点(因为它是同一个操作的一部分,但你错过了写入节点& #39; s lineno
)。
然后,解决方案是使用Call
语句使用void调用包装Expr
语句:
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()