Python：如何在保留装饰器并调用原始方法的同时重写类的方法？

Question

在这种情况下，如何覆盖“类”方法？ 但是，我无法编辑solid.py和run_me.py。 同时维护装饰并能够调用原始装饰。

# - solid.py - (no control)
import http

class Solid(object):
    _cp_path = '/pos'

    @http.jsonrequest
    def break_apart(self):
        return "to pieces!"

;

# - me.py -
import solid

def break_apart(self):
    return "to sand! and " + super(solid.Solid, self).break_apart()

solid.Solid.break_apart = break_apart

;

# - run_me.py - (no control)
import me    # yes, me first
import solid

pebble = solid.Solid()
pebble.break_apart() # "to sand! and to pieces!"

编辑

多谢您的协助，不完整之处敬请见谅。 我忘记添加了solid.py具有装饰器。 但是， monkeypatch就像一个魅力一样：

• 我失去了原来的装饰

• 我不能超级称呼原著

  AttributeError：“超级”对象没有属性“ break_apart”

Answer 1

由于Python的功能是一流的，因此，请执行以下步骤：

import solid

# replacement method
def break_apart(self):
    return "to sand!"

# override Solid.break_apart with the new method
solid.Solid.break_apart = break_apart

另外，由于您调用了pebble.break_apart（），因此这意味着solid.py应该为：

class Solid(object):
    def break_apart(self):
        return "to pieces!"

请注意，将self参数添加到break_apart中。 当您调用pebble.break_apart（）（即self = pebble实例）时，这是隐含的

Answer 2

首先，您实际上拥有哪种类方法？

它看起来像一个staticmethod但是您没有使用staticmethod装饰器。

实例方法如下所示：

class C:
    def instance_method(self, *args, **kwargs):
        return self.foo

类方法如下所示：

    @classmethod
    def class_method(cls, *args, **kwargs):
        return cls.foo

静态方法如下所示：

    @staticmethod
    def static_method(*args, **kwargs):
        return foo

如果您无法更改run_me.py则不能使用继承来更改此代码。

相反，您可以使用monkeypatch的兼容实现简单地对Solid类进行break_apart如下所示：

import solid

def break_apart(self):
    return "whatever you want"

solid.Solid.break_apart = break_apart

当然，由于代码是首先导入的，因此实际上您可以替换整个类。

因此，只需在您已经在做的工作中定义Concrete ，然后猴子就将整个模块修补在solid模块中：

import solid

class Concrete:
    ...

solid.Solid = Concrete

Answer 3

我认为您可以采用两种不同的方法来覆盖无法编辑源代码的类中的方法。

首先是创建一个覆盖该方法的子类。 这很简单，但是它只会影响您使用子类的构造函数创建的对象，而不是原始类：

me.py：

import http

import solid

class MySolid(solid.Solid):
    @http.jsonjequest
    def break_apart(self):
        return "to sand! and " + super(solid.Solid, self).break_apart()

我实际上不知道http.jsonrequest装饰器的作用，因此，如果它以某种方式修改了呼叫签名，则可能需要将其更改为原始值才能使用它。

另一种方法是猴子修补现有的类。 这意味着您用自己的替代版本替换了类中的方法实现。 如果原始类的实例是由您也不控制的其他代码创建的，则这很有用。 请注意，如果仍然需要访问原始方法的实现，则需要自己保存对其的引用（ super不会处理这种情况）。

me.py：

import http

import solid

_orig_break_apart = solid.Solid.break_apart

@http.jsonrequest
def break_apart(self):
    return "to sand! and " + _orig_break_apart(self)

solid.Solid.break_apart = break_apart

在这里，如果装饰者更改了签名，则可能需要更改调用原始方法的方式。

该方法的修饰版本可能与未修饰版本有很大差异（例如，它会从根本上更改呼叫签名或返回值）。 在这种情况下，“逆转”此过程以获取原始方法实现可能并不容易，以便您可以在新版本中调用它。 在这种情况下，您可能需要将代码从现有实现复制到覆盖版本中（使用上述两种覆盖方法之一）。 在示例代码中，您只需返回"to sand! and to pieces!" 无需尝试调用原始方法。 在实际代码中，这可能会更复杂，但适用相同的想法。

Answer 4

在stackoverflow上站在巨人的肩膀上，这对我有用。 我知道这有点复杂，但是我还没有偶然发现更简单的东西。 该代码适用于Python3。

import functools
def log( fn ):
    @functools.wraps( fn )
    def wrapper( *args, **kwargs ):
        print( "log", fn.__name__, args[ 1: ] )
        return fn( *args, **kwargs )
    return wrapper

def prepare_class( clazz ):
    @classmethod
    def on_class_patcher( cls, func_name, context ):
        def patch_by_name( new_func) :
            original_func = getattr( cls, func_name )
            def patched_func( self, *args, **kwargs ):
                return new_func( self, original_func, context, *args, **kwargs )
            setattr( cls, func_name, patched_func )
        return patch_by_name

    setattr( clazz, "patch", on_class_patcher )

# --- Use like this ---
class Demo:
    @log
    def m1( self, x, y ):
        print( "Demo.m1" )
        print( x, y )

    def m2( self ):
        print( "Demo.m2" )

class Foo:
    def m3( self ):
        print( "Foo.m3" )

prepare_class( Demo )
foo = Foo()

@Demo.patch( "m1", context = { "foo": foo } )
def patched_m1( self, original_func, context, *args, **kwargs ):
    print( "Demo.m1_patched" )
    self.m2()
    context[ "foo" ].m3()
    x = args[ 0 ] * 2
    y = args[ 1 ] * 2
    return original_func( self, x, y )

demo = Demo()
demo.m1( 1, 2 )

结果：

Demo.m1_patched
Demo.m2
Foo.m3
log m1 (2, 4)
Demo.m1
2 4