python - 如何修补函数的 __code__？

Question

我正在探索 __code__ 属性和 ast 模块，发现奇怪的行为，函数的第一次调用不会输出任何内容:

In [3]: def foo():
   ...:     print('foo')
   ...:
In [4]: p = ast.parse("".join(inspect.getsourcelines(foo)[0]))
In [5]: p.body[0].body = ast.parse("print('bar')").body
In [6]: foo.__code__ = compile(p, foo.__code__.co_filename, 'exec')
In [7]: foo()  # Nothing happens??

In [8]: foo()  # Okay, let's try again
bar

这里发生了什么？

Answer 1

您可能认为您已将 foo.__code__ 设置为打印 bar 的代码对象，但这不是您所做的。您已将 foo.__code__ 设置为一个代码对象，该对象定义了一个用于打印 bar 的新 foo 函数。有点像

def foo():
    global foo
    def foo():
        print('bar')

第一次运行后，foo是新的条形打印函数，运行新函数会打印bar。