如何将装饰器应用于超类的方法？

Question

假设我有一个 python 类

class Foo(object):
    def method1(self, x):
        print("method1 called with %s" %(x,))

    def method2(self, x):
        print("method2 called with %s" %(x,))

和一个装饰器

def myDecorator(func):
    newFunc = someModificationOf(func)
    return newFunc

我希望能够将Foo 子类化，并且在子类定义中以某种方式表明我希望将myDecorator 应用于从Foo 继承的某些方法子集。例如，我的子类可能喜欢

class Baz(Foo):
    methodsToDecorate = ['method2']

这可能吗？

我尝试了什么：

我认为我可以使用元类来做到这一点：

class Baz(Foo):
    __metaclass__ = Meta
    methodsToDecorate = ['method2']

class Meta(type):
    def __new__(cls, name, bases, d):
        for m in d['methodsToDecorate']:
            d[m] = myDecorator(d[m])
        return type(name, bases, d)

但是，它不起作用，因为'method2' 不在传递给元类的__new__ 方法的字典中，所以我们得到了KeyError。我喜欢使用元类的想法，因为它允许装饰器操作函数，而不是方法，这似乎是一个很好的简化。

Answer 1

我假设您真正想要做的不是装饰基类的这些方法的副本，而是装饰子类中这些方法的新覆盖副本。

如果是这样，那很容易。这些方法在传递给元类的字典中不存在，因为它们没有在类定义中定义，它们是从基类继承的。元类也可以访问。所以：

def __new__(cls, name, bases, d):
    def find_method(m):
        for base in bases:
            try:
                return getattr(base, m)
            except AttributeError:
                pass
        raise AttributeError("No bases have method '{}'".format(m))
    for m in d['methodsToDecorate']:
        d[m] = myDecorator(find_method(m))
    return type(name, bases, d)

如果你真的想进入基类并替换它的方法，你当然可以使用相同的技巧来做到这一点。只需将 base 与找到的方法一起返回，这样您就可以setattr 它。

---

您要求重写基类中的方法。但是，如果该方法实际上来自更远的祖先呢？那么这仍然可以工作......但它可能无法按照您想要的方式工作。如果您有多个共享共同祖先的基础，则在每个直接基础上调用 getattr 将在每个基础上搜索该共同祖先一次，但在新类上调用 getattr 只会在最后搜索一次。

正如 user2357112 在评论中指出的那样，如果您想要覆盖的是“如果我没有覆盖任何东西会被调用的任何东西”，您可以通过构造新类 first 来做到这一点，然后通过在结果类本身上调用getattr 来修改它。在 2.x 中，这确实意味着您必须处理未绑定的方法而不是函数（在 3.x 中，这不是问题，因为未绑定的方法只是是函数），但好处可能超过了成本。