32.2. ast —抽象语法树

2.5 版中的新Function：仅包含节点类的低级_ast模块。

2.6 版的新Function：包含所有帮助程序的高级ast模块。

源代码： Lib/ast.py

ast模块可帮助 Python 应用程序处理 Python 抽象语法语法的树。每个 Python 版本都可能更改抽象语法。此模块有助于以编程方式找出当前语法的外观。

可以pass将ast.PyCF_ONLY_AST作为标志传递给compile()内置函数或使用此模块中提供的parse()帮助器来生成抽象语法树。结果将是一棵对象树，其所有类都继承自ast.AST。可以使用内置的compile()函数将抽象语法树编译为 Python 代码对象。

32.2.1. 节点类

在抽象语法中，为每个左侧符号定义了一个类(例如ast.stmt或ast.expr)。另外，在右侧为每个构造函数定义了一个类。这些类继承自左侧树的类。例如，ast.BinOp继承自ast.expr。对于具有替代选项(又称“和”)的生产规则，左侧类是抽象的：仅创建特定构造函数节点的实例。

具体类的每个实例对每个子节点都有一个属性，该属性的类型如语法中所定义。例如，ast.BinOp个实例具有类型ast.expr的属性left。

如果在语法中将这些属性标记为可选(使用问号)，则值可能为None。如果属性可以具有零个或多个值(标有星号)，则这些值将表示为 Python 列表。使用compile()编译 AST 时，所有可能的属性都必须存在且具有有效值。

类ast.T的构造函数按如下方式解析其参数：

例如，要创建并填充ast.UnaryOp节点，可以使用

node = ast.UnaryOp()
node.op = ast.USub()
node.operand = ast.Num()
node.operand.n = 5
node.operand.lineno = 0
node.operand.col_offset = 0
node.lineno = 0
node.col_offset = 0

或更紧凑

node = ast.UnaryOp(ast.USub(), ast.Num(5, lineno=0, col_offset=0),
                   lineno=0, col_offset=0)

2.6 版中的新增Function：添加了上述构造函数。在 Python 2.5 中，必须pass调用不带参数的类构造函数并随后设置属性来创建节点。

32.2.2. 抽象语法

模块定义一个字符串常量__version__，它是下面显示的文件的十进制 Subversion 修订版号。

当前，抽象语法的定义如下：

-- ASDL's five builtin types are identifier, int, string, object, bool

module Python version "$Revision$"
{
	mod = Module(stmt* body)
	    | Interactive(stmt* body)
	    | Expression(expr body)

	    -- not really an actual node but useful in Jython's typesystem.
	    | Suite(stmt* body)

	stmt = FunctionDef(identifier name, arguments args, 
                            stmt* body, expr* decorator_list)
	      | ClassDef(identifier name, expr* bases, stmt* body, expr* decorator_list)
	      | Return(expr? value)

	      | Delete(expr* targets)
	      | Assign(expr* targets, expr value)
	      | AugAssign(expr target, operator op, expr value)

	      -- not sure if bool is allowed, can always use int
 	      | Print(expr? dest, expr* values, bool nl)

	      -- use 'orelse' because else is a keyword in target languages
	      | For(expr target, expr iter, stmt* body, stmt* orelse)
	      | While(expr test, stmt* body, stmt* orelse)
	      | If(expr test, stmt* body, stmt* orelse)
	      | With(expr context_expr, expr? optional_vars, stmt* body)

	      -- 'type' is a bad name
	      | Raise(expr? type, expr? inst, expr? tback)
	      | TryExcept(stmt* body, excepthandler* handlers, stmt* orelse)
	      | TryFinally(stmt* body, stmt* finalbody)
	      | Assert(expr test, expr? msg)

	      | Import(alias* names)
	      | ImportFrom(identifier? module, alias* names, int? level)

	      -- Doesn't capture requirement that locals must be
	      -- defined if globals is
	      -- still supports use as a function!
	      | Exec(expr body, expr? globals, expr? locals)

	      | Global(identifier* names)
	      | Expr(expr value)
	      | Pass | Break | Continue

	      -- XXX Jython will be different
	      -- col_offset is the byte offset in the utf8 string the parser uses
	      attributes (int lineno, int col_offset)

	      -- BoolOp() can use left & right?
	expr = BoolOp(boolop op, expr* values)
	     | BinOp(expr left, operator op, expr right)
	     | UnaryOp(unaryop op, expr operand)
	     | Lambda(arguments args, expr body)
	     | IfExp(expr test, expr body, expr orelse)
	     | Dict(expr* keys, expr* values)
	     | Set(expr* elts)
	     | ListComp(expr elt, comprehension* generators)
	     | SetComp(expr elt, comprehension* generators)
	     | DictComp(expr key, expr value, comprehension* generators)
	     | GeneratorExp(expr elt, comprehension* generators)
	     -- the grammar constrains where yield expressions can occur
	     | Yield(expr? value)
	     -- need sequences for compare to distinguish between
	     -- x < 4 < 3 and (x < 4) < 3
	     | Compare(expr left, cmpop* ops, expr* comparators)
	     | Call(expr func, expr* args, keyword* keywords,
			 expr? starargs, expr? kwargs)
	     | Repr(expr value)
	     | Num(object n) -- a number as a PyObject.
	     | Str(string s) -- need to specify raw, unicode, etc?
	     -- other literals? bools?

	     -- the following expression can appear in assignment context
	     | Attribute(expr value, identifier attr, expr_context ctx)
	     | Subscript(expr value, slice slice, expr_context ctx)
	     | Name(identifier id, expr_context ctx)
	     | List(expr* elts, expr_context ctx) 
	     | Tuple(expr* elts, expr_context ctx)

	      -- col_offset is the byte offset in the utf8 string the parser uses
	      attributes (int lineno, int col_offset)

	expr_context = Load | Store | Del | AugLoad | AugStore | Param

	slice = Ellipsis | Slice(expr? lower, expr? upper, expr? step) 
	      | ExtSlice(slice* dims) 
	      | Index(expr value) 

	boolop = And | Or 

	operator = Add | Sub | Mult | Div | Mod | Pow | LShift 
                 | RShift | BitOr | BitXor | BitAnd | FloorDiv

	unaryop = Invert | Not | UAdd | USub

	cmpop = Eq | NotEq | Lt | LtE | Gt | GtE | Is | IsNot | In | NotIn

	comprehension = (expr target, expr iter, expr* ifs)

	-- not sure what to call the first argument for raise and except
	excepthandler = ExceptHandler(expr? type, expr? name, stmt* body)
	                attributes (int lineno, int col_offset)

	arguments = (expr* args, identifier? vararg, 
		     identifier? kwarg, expr* defaults)

        -- keyword arguments supplied to call
        keyword = (identifier arg, expr value)

        -- import name with optional 'as' alias.
        alias = (identifier name, identifier? asname)
}

32.2.3. ast 助手

2.6 版的新Function。

除了节点类之外，ast模块还定义了以下 Util 函数和类，用于遍历抽象语法树：

Warning

由于 Python AST 编译器中的堆栈深度限制，使用足够大/复杂的字符串可能会使 Python 解释器崩溃。

这可用于安全地评估包含来自不受信任来源的 Python 值的字符串，而无需自己解析值。它不能评估任意复杂的表达式，例如涉及运算符或索引的表达式。

Warning

由于 Python AST 编译器中的堆栈深度限制，使用足够大/复杂的字符串可能会使 Python 解释器崩溃。

该类打算被子类化，并且该子类添加了 visitor 方法。

请注意，除非访问者调用generic_visit()或自己访问它们，否则将不会访问具有自定义访问者方法的节点的子节点。

如果要在遍历期间将更改应用于节点，请不要使用NodeVisitor。为此，存在允许修改的特殊访问者(NodeTransformer)。

NodeTransformer将遍历 AST，并使用 visitor 方法的返回值替换或删除旧节点。如果 visitor 方法的返回值为None，则将从其位置中删除该节点，否则将其替换为返回值。返回值可能是原始节点，在这种情况下不会发生替换。

这是一个示例转换器，它将所有出现的名称查找(foo)重写为data['foo']：

class RewriteName(NodeTransformer):

    def visit_Name(self, node):
        return copy_location(Subscript(
            value=Name(id='data', ctx=Load()),
            slice=Index(value=Str(s=node.id)),
            ctx=node.ctx
        ), node)

请记住，如果要操作的节点具有子节点，则必须自己转换子节点或首先为该节点调用generic_visit()方法。

对于属于语句集合(适用于所有语句节点)的节点，访问者还可能返回节点列表，而不仅仅是一个节点。

通常，您使用变压器是这样的：

node = YourTransformer().visit(node)
首页