如何在 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，这就是为什么你使用较低的-level _ast 模块代替？）...：

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

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

在 Python 2.6 的模块 ast 中，您还可以获得帮助程序，让您更轻松地在树上导航（例如，通过 Visitor 设计模式）——但整个树都存在于 2.5 中（使用 @987654332 @) 或 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
'.'

HTH