python - 使用 __getattr__ 覆盖魔术方法

Question

我有一个类，它是成员的容器。所有成员都属于同一类型。

class A(int):
    def __init__(self, n):
        super().__init__()
        self.n = n

    def do(self):
        print('adding 10')
        return self.n + 10


class B:
    def __init__(self, a1, a2):
        self.a = [a1, a2]

    def __getattr__(self, item):
        return getattr(self.a[0], item)


a1 = A(10)
a2 = A(5)
b = B(a1, a2)

__getattr__ 覆盖 do 方法:

In[7]: b.do()
adding 10
Out[7]: 20

甚至在显式调用时覆盖__add__

In[8]: b.__add__(1)
Out[8]: 11

但是 __add__ 当作为 + 调用时失败

In[9]: b+1
TypeError: unsupported operand type(s) for +: 'B' and 'int'

如何重写魔术方法以按预期工作？

Answer 1

Why is __getattr__ capable of handling built-in operator overloads in python 2.x? 中介绍了 __getattr__ 不处理运算符的原因。

In new-style classes the special methods are always looked up in the class(implicit lookup) not instance.

我的解决方法(并不完美，但允许使其工作)是在 B 中显式定义所需的 dunder 方法，并在它们上显式调用 __getattr__ :

class B:
    def __init__(self, a1, a2):
        self.a = [a1, a2]

    def __add__(self,other):
        return self.__getattr__("__add__")(other)

    def __getattr__(self, item):
        return getattr(self.a[0], item)