猴子补丁方法和参考的问题

Question

我想知道是否有人可以解释并提供此问题的解决方案：

$ cat object-override-methods.py 
class A:
    def foo(self):
        return 1

class B:
    def foo(self):
        return 1

for klass in A, B:
    orig_foo = klass.foo
    def foo(self):
        return orig_foo(self) * 2
    klass.foo = foo

A().foo()
B().foo()
$ python object-override-methods.py
Traceback (most recent call last):
  File "object-override-methods.py", line 15, in <module>
    A().foo()
  File "object-override-methods.py", line 12, in foo
    return orig_foo(self) * 2
TypeError: unbound method foo() must be called with B instance as first argument (got A instance instead)

提前致谢。

Answer 1

orig_foo 是一个全局变量，每次通过循环都会改变值。循环完成后，orig_foo 引用B.foo。

内部函数 foo（一个或每个通过循环）在调用时都使用 orig_foo 的全局值。所以他们都调用B.foo(self)。

当调用像orig_foo这样的“未绑定方法”时，Python2会检查第一个参数是否是适当类的实例。 A().foo() 未通过此检查。 （有趣的是，这个检查在 Python3 中被删除，因此不会引发 TypeError，并且这个 bug 可能会变得更难发现。）

要解决这个问题，您必须将 orig_foo 的值绑定到适当的 <代码>类。
您可以通过将 orig_foo 设置为 foo 的局部变量来实现这一点。一种方法是使用默认值使 orig_foo 成为 foo 的参数。 Python 在定义函数时绑定默认值。因此，orig_foo=orig_foo 将局部变量 orig_foo 绑定到 klass.foo 的当前值：

for klass in A, B:
    orig_foo = klass.foo
    def foo(self, orig_foo=orig_foo):
        return orig_foo(self) * 2
    klass.foo = foo

Answer 2

因为 orig_foo 是在全局范围内定义的，所以每次循环都会践踏它的值。然后，每个新的 foo 方法都会共享该被践踏的值。

一个简单的解决方法是将代码移至函数中，如下所示：

def rebind_foo(klass):
    orig_foo = klass.foo
    def foo(self):
        return orig_foo(self) * 2
    klass.foo = foo

for klass in A, B:
   rebind_foo(klass)

这确保每个新的 foo 方法都获得其自己的 orig_foo 值。