Python AST 节点无法在函数内执行

Question

有人可以给我关于这种行为的建议吗？下面的代码运行良好:

import ast


node = ast.parse('def nsd(a, b):\n    if b == 0: return a    \n    return nsd(b, a%b)\n\nprint nsd(18,15)')
obj = compile(node, filename="<ast>", mode="exec")
exec obj

但是当我在函数内执行相同操作时:

import ast


def foo():
    node = ast.parse('def nsd(a, b):\n    if b == 0: return a    \n    return nsd(b, a%b)\n\nprint nsd(18,15)')
    obj = compile(node, filename="<ast>", mode="exec")
    exec obj

foo()

它会引发错误:

Traceback (most recent call last):
  File "C:/Users/Vectoun/PycharmProjects/untitled3/test.py", line 9, in <module>
    foo()
  File "C:/Users/Vectoun/PycharmProjects/untitled3/test.py", line 7, in foo
    exec obj
  File "<ast>", line 5, in <module>
  File "<ast>", line 3, in nsd
NameError: global name 'nsd' is not defined

我希望能够在函数内运行它。有谁知道如何解决这个问题吗？

Answer 1

如果我没记错的话，这里的问题是，当使用 exec 时，你仍然绑定(bind)到一个范围。

因此在方法外部运行 AST 代码会将 nsd 定义为全局方法，而在 foo() 内运行 AST 代码将使 nsd 是 foo 直接作用域的成员，因此调用在查找方法时将失败。

您可以通过首先将 nsd 定义为全局来修复它:

import ast


def foo():
    code = """global nsd
def nsd(a, b):
    if b == 0:
        return a
    return nsd(b, a%b)
print nsd(18,15)
    """
    node = ast.parse(code)
    obj = compile(node, filename="<ast>", mode="exec")
    exec obj

foo()