python - 在 Python 中修补一个类的所有实例

Question

我正在使用 Python 3.5.1。我写了一个库，在某些情况下(当文件直接运行时，即 __name__ == '__main__')我想在其中一个类中装饰某些方法。它应该装饰所有可以创建的实例。我想以一种非侵入性的方式进行，即理想情况下，我的库中的类不需要任何特殊代码。

一段时间后，我设法实现了这样的东西，满足了我的要求:

def patch(clazz, name, replacement):

    def wrap_original(orig):
        # when called with the original function, a new function will be returned
        # this new function, the wrapper, replaces the original function in the class
        # and when called it will call the provided replacement function with the
        # original function as first argument and the remaining arguments filled in by Python

        def wrapper(*args, **kwargs):
            return replacement(orig, *args, **kwargs)

        return wrapper

    orig = getattr(clazz, name)
    setattr(clazz, name, wrap_original(orig))

def replacement_function(orig, self, ... other argumnents ...):
    # orig here is the original function, so can be called like this:
    # orig(self, ... args ...)
    pass

patch(mylib.MyClass, 'method_name', replacemment_function)

令人惊讶的是，这段代码有效，虽然我还没有用类方法测试它，但我现在不需要它。它还会修补在修补之前创建的实例，尽管我还不确定它是否好；d

上面的代码可以说是很难的，我在写完之后需要一段时间来了解它的工作方式，以便写出解释注释。我想要更简单的东西。

问题:Python 库中是否有任何东西可以使这样的代码变得不必要，它已经实现了我正在做的事情，但更好？

Answer 1

方法是在查找实例时动态创建的；实例没有所有方法的副本，而是 descriptor protocol从类中获取函数并根据需要将它们绑定(bind)到实例。这就是为什么 monkeypatching 类在这里起作用的原因； instance.method_name 将在执行属性查找时找到 mylib.MyClass.method_name 。

默认库中没有执行您在此处执行的操作的任何内容，不，因为不同的代码可能需要不同的处理委托(delegate)返回旧方法的模式。

您的方法看起来非常接近 how the Mercurial project支持函数包装，因为原始函数被传递到包装器中。