将函数包装为python中类的子类中的方法

Question

这是（简化的）案例：

我有一个包含子类的类：

>>> inst = Foo()
>>> inst.sub.bar()

它从子类的 bar() 方法返回了一些很棒的东西。
这是这样写的：

class Sub() :
    def __init__(inst) :
        self.inst = inst

    def bar(self) :
        '''does something magnificent  with the Foo instance (self.inst)'''
        return smtg

class Foo() :
    def __init__(self, *args, **kwargs):
        self.sub = Sub(self)

当在第三类中使用 Foo 来处理特定的东西时：

class superFoo(Foo) :
    
    def specific(self,*a,**kw) :
        ...

我还想使用@decorator 将特定方法包装在子类中，就像我为 Foo 所做的那样：

from functools import wraps

def add_method(cls):
    def decorator(func):
        @wraps(func) 
        def wrapper(self, *args, **kwargs): 
            return func(self, *args, **kwargs)
        setattr(cls, func.__name__, wrapper)
    return decorator
    
@add_method(superFoo)
def spam(self, *args, **kwargs) :
    return self.specific() + ...
    
>>> inst = superFoo()
>>> inst.spam() # yep

但我想要的是这个：

>>> inst.sub.spam()

所以我尝试了：

@add_method(superFoo)
def spam(self, *args, **kwargs) :
    return self.inst.specific() + ...

在 add_method 函数中：

setattr(cls.sub, func.__name__, wrapper) # ouch, sub not instantiated

失败。

我可以继承 Sub（实际上我现在已经这样做了）来满足需求：

class superSub(Sub) :
    def spam(self, *args, **kwargs) :
        return self.inst.specific() + ...

class superFoo(Foo) :
    def __init__(self, *args, **kwargs):
        super().__init__(*args,**kwargs) # there is actualy other inheritances 
        self.sub = superSub(self) # replaces Sub() by its subclass

但我想知道我是否可以使用装饰器，如果可能的话，这样做：
add_method 将在需要时导入到模块中，并且模块的具体内容会一见钟情查看代码时：

from abyssal import add_method

class superFoo(Foo) : # folded in my editor

@add_method(superFoo) :
def spam(self) : # folded

非常感谢

Answer 1

您的 add_method 装饰器可以是您的 superFoo 类的方法：

class superFoo(Foo):

    def add_method(self):
        def decorator(func):
            setattr(self, func.__name__, func)
            return func
        return decorator

（注意我简化了一点）

然后你将它用作：

sub = Sub()
inst = superFoo(sub)

@inst.add_method(superFoo)
def spam(self, *args, **kwargs) :
    return self.inst.specific() + ...

但又一次。我建议使用继承。仅仅为Sub 定义子类会更简单、更容易理解。