Python ast 库 - 在 function 调用中检索节点的值

Question

此处的后续问题： Python ast Libary - 如何检索特定节点的值

我使用以下代码来检索{'console_scripts': ['main=smamesdemo.run.main:main']}

导入 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'))

这很好用，但这个例子的复杂性降低了。

我的真实示例是 Python Package 我想在其中检索此值：

尽可能最小，它看起来像这样：

import ast

code = '''extras_require={"dev": dev_reqs}
if __name__ == "__main__":
    setup(
        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)

围绕这个包装一个 function 调用将使ast.Assign节点消失？，据我了解，我只会再获得一个节点。 但这种情况并非如此。

为什么会发生这种情况以及如何在这样的结构中检索所需的值？

Answer 1

在这种情况下，所需字典的值来自关键字参数，因此您应该寻找ast.keyword节点而不是ast.Assign节点：

for node in ast.walk(ast.parse(code)):
    if isinstance(node, ast.keyword) and node.arg == 'entry_points':
        expr = ast.Expression(body=node.value)
        ast.fix_missing_locations(expr)
        entry_points = eval(compile(expr, filename='', mode='eval'))

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