如何在 Python 中获取完整的 AST？

Question

我喜欢 _ast 模块提供的选项，它真的很强大。 有没有办法从中获取完整的 AST？

例如，如果我获得以下代码的 AST：

import os
os.listdir(".")

通过使用：

ast = compile(source_string,"<string>","exec",_ast.PyCF_ONLY_AST)

ast 对象的主体将有两个元素，一个 import 对象和一个 expr 对象。 但是，我想更进一步，获取 import 和 listdir 的 AST，换句话说，我想制作 _ast 下降到可能的最低水平。

我认为这种事情应该是可能的，这是合乎逻辑的。 问题是如何？

编辑：通过尽可能最低的级别，我并不是指访问“可见”的内容。 我还想获取 listdir 实现的 AST：例如 stat 以及可能为其执行的其他函数调用。

Answer 1

您确实可以通过这种方式获得整个树 - 一直到底部 - 但是，它是作为一棵树保存的，确切地说......所以在每个级别，为了获得子级，您必须显式访问所需的属性。 例如（我将 compile 结果命名为 cf 而不是 ast 因为这会隐藏标准库 ast 模块 - 我假设你只有 2.5 而不是 2.6，这就是为什么您使用较低级别的 _ast 模块？）...：

>>> cf.body[0].names[0].name
'os'

这就是告诉您 import 语句正在导入 name os< /code> （这只是因为 1 是 .body[0] 的 .names 字段的长度，即 import） 。

在 Python 2.6 的模块 ast 中，您还可以获得帮助程序，让您在树上更轻松地导航（例如通过 Visitor 设计模式）——但整个树都在那里在 2.5（使用 _ast）或 2.5（使用 ast）中，并且在任何一种情况下都以完全相同的方式表示。

要方便地访问树中的所有节点，在 2.6 中，请使用模块 ast（无前导下划线）并根据需要子类 ast.NodeVisitor（或等效地递归地使用 ast.iter_child_nodes和 ast.iter_fields 根据需要）。 当然，如果您由于某种原因陷入 2.5 版本，这些助手可以在 _ast 之上用纯 Python 实现。

Answer 2

py> ast._fields
('body',)
py> ast.body
[<_ast.Import object at 0xb7978e8c>, <_ast.Expr object at 0xb7978f0c>]
py> ast.body[1]
<_ast.Expr object at 0xb7978f0c>
py> ast.body[1]._fields
('value',)
py> ast.body[1].value
<_ast.Call object at 0xb7978f2c>
py> ast.body[1].value._fields
('func', 'args', 'keywords', 'starargs', 'kwargs')
py> ast.body[1].value.args
[<_ast.Str object at 0xb7978fac>]
py> ast.body[1].value.args[0]
<_ast.Str object at 0xb7978fac>
py> ast.body[1].value.args[0]._fields
('s',)
py> ast.body[1].value.args[0].s
'.'

华泰