python - 覆盖类的方法定义

Question

假设我有一个类 FooClass，它定义了一个方法 foo_method。这个类在第三方库中，我想明确地覆盖它。我有许多继承自 FooClass 的类，所以我不想为每个子类重写 foo_method。相反，我想在不深入研究第三方库代码的情况下覆盖类的方法定义。当我尝试显而易见的方法时，

from thirdparty import FooClass

class FooClass(object):

  def foo_method():
    newstuff

我得到了奇怪的行为——NotImplementedErrors 等等。我错过了什么吗？

Answer 1

您正在本地重新定义类；这不会改变原始类或影响在其他地方使用它的任何人。

您可以猴子修补类上的方法:

from thirdparty import FooClass

orig_foo_method = FooClass.foo_method

def new_foo_method(self):
    # do new stuff
    # perhaps even call orig_foo_method(self)

FooClass.foo_method = new_foo_method

现在，任何试图在 Python 解释器中的任何位置对 FooClass 实例调用 .foo_method() 的代码都会找到并调用您的 new_foo_method() 方法代替。