Python ast 库 - 如何检索特定节点的值

Question

我有以下脚本：

extras_require={"dev": dev_reqs}
x = 3
entry_points={
    "console_scripts": ["main=smamesdemo.run.main:main"]
}

我想检索以下部分作为 python 数据， dict和list not Nodes。

entry_points={
    "console_scripts": ["main=smamesdemo.run.main:main"]
}

我尝试了以下方法，但实际上我无法通过以下方式获取值：

import ast


class CustomNodeTransformer(ast.NodeTransformer):
    def visit_Assign(self, node):
        print(node.value.__dict__)
        return node


with open("./setup.py") as f:
    code = f.read()

    node = ast.parse(code)
    CustomNodeTransformer().visit(node)

{'keys': [<ast.Constant object at 0x0000029E992962C0>], 'values': [<ast.Name object at 0x0000029E99296260>], 'lineno': 1, 'col_offset': 15, 'end_lineno': 1, 'end_col_offset': 32}

预期： entry_points={"console_scripts": ["main=smamesdemo.run.main:main"]}

谁能帮我归档这个？

Answer 1

您应该使用ast.NodeVisitor而不是ast.NodeTransformer ，因为您没有对节点进行修改。 但是为了获取单个节点，使用ast.walk更简单，而是以扁平方式遍历 AST，找到目标 id 为'entry_points'的ast.Assign节点，然后将其值包装在ast.Expression中ast.Expression节点用于编译并使用eval compile和评估为实际的 Python 字典。 当您将它植入另一棵树时，请记住使用ast.fix_missing_locations重新对齐节点的行号和偏移量：

import ast

code = '''extras_require={"dev": dev_reqs}
x = 3
entry_points={
    "console_scripts": ["main=smamesdemo.run.main:main"]
}'''

for node in ast.walk(ast.parse(code)):
    if isinstance(node, ast.Assign) and node.targets[0].id == 'entry_points':
        expr = ast.Expression(body=node.value)
        ast.fix_missing_locations(expr)
        entry_points = eval(compile(expr, filename='', mode='eval'))

print(entry_points)

这输出：

{'console_scripts': ['main=smamesdemo.run.main:main']}

演示： https://replit.com/@blhsing/OrangeredCurlyApplication

Answer 2

从 Python 3.9 您可以使用ast.unparse function 从Node获取等效代码作为字符串

import ast


class CustomNodeTransformer(ast.NodeTransformer):
    def visit_Assign(self, node):
        print(ast.unparse(node))
        return node


code = """
extras_require={"dev": dev_reqs}
x = 3
entry_points={
    "console_scripts": ["main=smamesdemo.run.main:main"]
}
"""
node = ast.parse(code)
CustomNodeTransformer().visit(node)

# Output:
# extras_require = {'dev': dev_reqs}
# x = 3
# entry_points = {'console_scripts': ['main=smamesdemo.run.main:main']}