编程语言
首页 > 编程语言> > 如何修改Python源代码以添加新的AST节点?

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

作者:互联网

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

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

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

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

解决方法:

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

一定要看看Eli Bendersky的article on the matter.

这是他提供的一个例子,其中构建了AST并从头开始评估!

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)

标签:python,grammar,abstract-syntax-tree,asdl
来源: https://codeday.me/bug/20190709/1416597.html