Python AST,ast.NodeTransformer,TypeError：stmt中缺少必填字段“ lineno”

我会很感激地学习一些有用的东西,因为到目前为止,我一直在盲目地前进.
所以问题出在python的ast.NodeTransformer中.我想知道是否可以使用这种方法将功能添加到现有类中,而不会发疯.

到目前为止,这就是我的处理方式.

import ast, inspect, cla # cla is a name of class to which we want to add a new function

klass = inspect.getsource(cla)
tree = ast.parse(klass)
st = '''def function(): return 1'''
Foo = ast.parse(st)

class AddFunc(ast.NodeTransformer):
      def visit_ClassDef(self, node):
          return node, node.body + Foo.body
          self.generic_visit(node)

inst = AddFunc()
stuff = i.visit(tree)

# now the trouble begins, a compiling..
co = compile(stuff, filename='<ast>', mode='exec')

# i get TypeError: required "lineno" missing from stmt

我已经尝试过(可能无法猜到),使用
ast库帮助程序函数ast.fix_missing_locations(),
和ast.copy_locaction(),但在大多数情况下,我最终都在猜测或
通过在AddFunc类内部的元组面对AttributeError.
有任何想法,如何处理？

解决方法:

kolko的答案是正确的,在这种情况下,您需要返回一个AST节点.在这种情况下,Foo是一个模块,它的主体应该是构造该函数的语句,可以将其简单地拼接到类定义中.

有助于了解ast.NodeTransformer非常特别：

>它通过特定的类名调用访问方法,而忽略层次结构
>它使用isinstance(AST)和isinstance(list)评估返回值,其他任何内容都将被忽略！

即使使用ast.fix_missing_locations添加位置数据,您也可能会收到此错误或紧密相关的“ expr中缺少lineno”.

如果发生这种情况,那是因为您的AST具有该位置不允许的元素.使用astunparse.dump仔细检查结构.在模块主体中,只能有语句.在函数def中,只能有语句.然后,您可以在特定的地方有一个表达式.

为了解决这个问题,写出您期望生成的python,对其进行解析,转储并对照您生成的内容进行检查.

标签：abstract-syntax-tree,python
来源： https://codeday.me/bug/20191029/1960676.html