动态添加内置方法以指向属性的内置方法[重复]

我有几个类和一个函数：

from functools import partial 

def fn(other, self, name):
    print(f"calling {name} with {other}")
    func =  getattr(self.a, name)
    return func(other)

class A:

    def __add__(self, other):
        return 9

    def __mul__(self, other):
        return 7

    def __sub__(self, other):
        return 8

class B:

    def __init__(self,a):
        self.a = a

        for name  in ['add', 'sub']:
            name = f"__{name}__"
            p = partial(fn, self=self,name=name)
            setattr(self, name, p)
            p.__name__ = name

我希望能够将魔术方法转发到现有属性。我不想继承该类，因为我不想要所有的内置函数。只是一对。例如，我可能想使用不同类的乘法。我试图避免这样的编码：

def __add__(self, other):
    self.a.__add__(other)

使用上面的代码我收到以下信息：

>>> b = B(A())
>>> b + 3
TypeError                                 Traceback (most recent call last)
<ipython-input-40-fa904b7bb783> in <module>
----> 1 b + 3
      2 

TypeError: unsupported operand type(s) for +: 'B' and 'int'
>>> b.__add__(3)
calling __add__ with 3
9

也许我错过了一些简单的东西，但我找不到动态添加内置函数的方法。

要解决的主要问题是像这样的魔术方法__add__在类上查找，而不是在对象本身上查找；否则你可以写self.__add__ = a.__add__ in the __init__方法。为了解决这个问题，我们需要在类上声明方法B，而不是它的个别实例。

功能delegate下面定义的工作原理是添加一个方法B班级。该方法需要self这将是一个B实例，因此它必须动态加载a属性，然后是它的__add__ method.

class A:
    def __add__(self, other):
        return 9
    def __mul__(self, other):
        return 7
    def __sub__(self, other):
        return 8

class B:
    def __init__(self, a):
        self.a = a

def delegate(cls, attr_name, method_name):
    def delegated(self, *vargs, **kwargs):
        a = getattr(self, attr_name)
        m = getattr(a, method_name)
        return m(*vargs, **kwargs)
    setattr(cls, method_name, delegated)

delegate(B, 'a', '__add__')
delegate(B, 'a', '__sub__')

Example:

>>> b = B(A())
>>> b + 3
9
>>> b - 4
8