7

我正在考虑修改一些第 3 方 Python 代码的行为。有许多类派生自基类,为了轻松实现我的目标,最简单的方法是重写用于派生所有其他类的基类。有没有一种简单的方法可以做到这一点而无需接触任何第 3 方代码?如果我未能清楚地解释这一点:

class Base(object):
    '...'

class One(Base)
    '...'

class Two(Base)
    '...'

...我想在Base 实际修改上述代码的情况下进行修改。也许是这样的:

# ...code to modify Base somehow...

import third_party_code

# ...my own code

Python 可能有一些可爱的内置解决方案来解决这个问题,但我还没有意识到这一点。

4

1 回答 1

8

也许您可以将方法修补到Base?

#---- This is the third-party module ----#

class Base(object):
  def foo(self):
    print 'original foo'

class One(Base):
  def bar(self):
    self.foo()

class Two(Base):
  def bar(self):
    self.foo()

#---- This is your module ----#

# Test the original
One().bar()
Two().bar()

# Monkey-patch and test
def Base_foo(self):
  print 'monkey-patched foo'

Base.foo = Base_foo

One().bar()
Two().bar()

这打印出来

original foo
original foo
monkey-patched foo
monkey-patched foo
于 2012-11-25T11:05:42.603 回答