python中的Monkey patching绑定方法

Question

>>> class A:
...     def foo(self):
...             print(self)
...
>>>
>>> a = A()
>>> a.foo()
<__main__.A instance at 0x7f4399136cb0>
>>> def foo(self):
...     print(self)
...
>>> a.foo = foo
>>> a.foo()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: foo() takes exactly 1 argument (0 given)

我正在尝试理解Python中的猴子补丁。请说明错误的原因以及如何修复它。

Answer 1

正如this SO answer中所述，在执行此操作时，您需要使用types.MethodType或类似的工具，例如：

a.foo = types.MethodType(foo, a)

原因是a.foo = foo只是将函数foo设置为a的一个属性-并没有执行“绑定魔术”。要让Python“神奇地”在调用a.foo时将实例作为第一个参数传递，您需要告诉Python进行这样的绑定，例如使用types.MethodType。

有关更多详细信息，请参阅上面的链接答案。

Answer 2

所以这里的棘手之处在于，您得到的结果取决于方法所在的位置：

class A(object):
    def foo(self):
        print("Hello world")


def patch(self):
    print("patched!")


print(type(A.foo))
a = A()
print(type(a.foo))

如果运行此命令，您将在python2.x和3.x上得到不同的结果：

$ python ~/sandbox/test.py  # python2.x
<type 'instancemethod'>
<type 'instancemethod'>
$ python3 ~/sandbox/test.py  # python3.x
<class 'function' at 0x100228020>
<class 'method' at 0x10021d0c0>

但在这两种情况下，很明显，a.foo是某种方法。

如果我们试图用猴子修补它，会发生什么？

a.foo = patch
print(type(a.foo))  # <type 'function'> (2.x) / <class 'function'> (3.x)

好了，现在我们看到a.foo的类型是function (不是一个方法)。因此，问题是我们如何从“补丁”中创建一个方法？答案是我们在将它添加为属性时使用它的描述符协议：

a.foo = patch.__get__(a, A)

对于类上的方法，当你执行a.some_method时，python实际上做的是：a.some_method.__get__(a, type(a))，所以我们在这里(显式地)重现调用序列。