编程语言
首页 > 编程语言> > Python AST:如何获取节点的子节点

Python AST:如何获取节点的子节点

作者:互联网

我正在研究Python 2.6.5.

给定一个抽象语法树,我想获得它的孩子.

我搜索了堆栈溢出但无济于事.大多数帖子都在ast.NodeVisitor上,并在其中定义的方法访问,generic_visit().
但是AFAIU,visit()和generic_visit()不给我们孩子,而是直接在他们身上递归应用函数

有人可以写一个简短的代码来证明吗?
python库中是否存在预定义的函数?

谢谢!

解决方法:

包含节点子节点的attaributes取决于节点所代表的语法类型.每个节点类还有一个特殊的_fields属性,该属性列出了该类具有的子节点的属性名称.例如,

>>> ast.parse('5+a')
<_ast.Module object at 0x02C1F730>
>>> ast.parse('5+a').body
[<_ast.Expr object at 0x02C1FF50>]
>>> ast.parse('5+a').body[0]
<_ast.Expr object at 0x02C1FBF0>
>>> ast.parse('5+a').body[0]._fields
('value',)
>>> ast.parse('5+a').body[0].value
<_ast.BinOp object at 0x02C1FF90>
>>> ast.parse('5+a').body[0].value._fields
('left', 'op', 'right')
>>> ast.parse('5+a').body[0].value.left
<_ast.Num object at 0x02C1FB70>

等等.

编辑,以澄清发生了什么

在继续前进之前,请先看一下CPython Abstract Grammar

考虑:

>>> type(ast.parse('5+a'))
<class '_ast.Module'>

实际上,如果你看一下语法,第一个生产规则就是Module.它似乎采用一系列陈述,作为一个称为身体的论证.

>>> ast.parse('5+a')._fields
('body',)
>>> ast.parse('5+a').body
[<_ast.Expr object at 0x02E965B0>]

AST的_fields属性只是“body”,body属性是AST节点的序列.回到语法,查看stmt的生产规则,我们看到Expr采用单个expr,命名值

>>> ast.parse('5+a').body[0].value
<_ast.BinOp object at 0x02E96330>

如果我们查找BinOp的定义,我们会看到它需要3个不同的参数,left,op和right.我希望你能从那里继续前进.

标签:python,abstract-syntax-tree
来源: https://codeday.me/bug/20190827/1735771.html