Question

我会感激地学习一些有用的东西，就像现在一样，我一直在盲目地移动。所以问题出现在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()，但在大多数情况下，我最终猜测或由AttributeError类内的元组面对AddFunc。有什么想法，如何管理？

Answer 1

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

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

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

即使您使用ast.fix_missing_locations添加位置数据，也可能会出现此错误或与expr密切相关的“lineno”。

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

要解决此问题，请写出您希望生成的python，解析它，转储它并根据您生成的内容进行检查。

Answer 2

我在twitter上找到了答案： https://twitter.com/mitsuhiko/status/91169383254200320 mitsuhiko：＆＃34; TypeError：必填字段＆＃34; lineno＆＃34; stmt＆＃34; - 不，你的意思是＆＃34;元组不是一个声明＆＃39;。 #helpfulpython

你回到这里元组，但你必须返回

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

Python AST，ast.NodeTransformer，TypeError：必填字段＆＃39; lineno＆＃39;从stmt失踪

2 个答案: