如何修改Python源代码以添加新的AST节点?

时间:2012-02-08 04:26:23

标签: python grammar abstract-syntax-tree asdl

例如,如果我要创建一个名为“Something”的节点,我想在Python AST树中使用,那么为了做到这一点,我必须在哪里和哪些更改添加到Python源代码?

我知道我应该从Python.asdl开始,其中定义了AST语法。然后我应该进入ast.c.

不幸的是,我不确定我必须在ast.c文件中进行更改才能实现节点。

另外,为简单起见,我只想说节点作为占位符,这意味着除了能够将自身插入树中之外,它不应该做任何事情。

1 个答案:

答案 0 :(得分:1)

无需潜入Python.asdl!然后你必须去寻找ASDL解析器然后你必须弄清楚如何处理它(尽管这本身就是一个值得做的练习)。由于令人惊叹的ast模块,你可以留在Python中。

请务必查看Eli Bendersky的article on the matter

以下是他提供构建AST并从头开始eval'的示例!

import ast

node = ast.Expression(ast.BinOp(
                ast.Str('xy'),
                ast.Mult(),
                ast.Num(3)))

fixed = ast.fix_missing_locations(node)

codeobj = compile(fixed, '<string>', 'eval')
print eval(codeobj)

ast.NodeTransformer

如果要转换现有AST,请查看ast.NodeTransformer类。

以上是上述博客文章中的一个示例,它将字符串值修改为 以str:为前缀。

class MyTransformer(ast.NodeTransformer):
    def visit_Str(self, node):
        return ast.Str('str: ' + node.s)