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

发布时间：2020-12-20 13:26:31 所属栏目：Python 来源：网络整理

导读：例如,如果我要创建一个名为“Something”的节点,我想在 Python AST树中使用,我必须在哪里和哪些更改添加到Python源代码才能做到这一点？我知道我应该从Python.asdl开始,其中定义了AST语法.然后我应该进入ast.c. 不幸的是,我不确定我必须在ast.c文件中进行更

例如,如果我要创建一个名为“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)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!