如何获取与Python AST节点相对应的源代码？

Question

Python AST节点具有lineno和col_offset属性，它们指示相应代码范围的开始。是否有一种简单的方法也可以终止代码范围？第三方图书馆？

Answer 1

我们有类似的需求，为此我创建了asttokens库。它以文本和标记化形式维护源，并用标记信息标记AST节点，从中也可以轻松获得文本。

它适用于Python 2和3（经过2.7和3.5测试）。例如：

import ast, asttokens

st='''

def greet(a):

  say("hello") if a else say("bye")

'''

atok = asttokens.ASTTokens(st, parse=True)

for node in ast.walk(atok.tree):

  if hasattr(node, 'lineno'):

    print atok.get_text_range(node), node.__class__.__name__, atok.get_text(node)

印刷

(1, 50) FunctionDef def greet(a):

  say("hello") if a else say("bye")

(17, 50) Expr say("hello") if a else say("bye")

(11, 12) Name a

(17, 50) IfExp say("hello") if a else say("bye")

(33, 34) Name a

(17, 29) Call say("hello")

(40, 50) Call say("bye")

(17, 20) Name say

(21, 28) Str "hello"

(40, 43) Name say

(44, 49) Str "bye"

Answer 2

ast.get_source_segment 是在python 3.8中添加的：

import ast

code = """

if 1 == 1 and 2 == 2 and 3 == 3:

     test = 1

"""

node = ast.parse(code)

ast.get_source_segment(code, node.body[0])

产生： if 1 == 1 and 2 == 2 and 3 == 3:\n     test = 1