自动生成重载运算符？

Question

考虑以下在 Python 中具有所有向量操作的向量对象的实现：

import operator

class Vector:

    def __init__(self, value):
        self._vals = value.copy()


    @classmethod
    def _op(cls, this, that, oper, rev=False):

        assert isinstance(this, cls)

        if rev:
            op = lambda a, b : oper(b, a)
        else:
            op = oper

        if isinstance(that, list):
            result = [op(x, y) for (x, y) in zip(this._vals, that)]
        elif isinstance(that, cls):
            result = [op(x, y) for (x, y) in zip(this._vals, that._vals)]
        else:
            # assume other is scalar
            result = [op(x, that) for x in this._vals]

        return cls(result)

    def __add__(self, other):
        return Vector._op(self, other, operator.add, False)


    def __radd__(self, other):
        return Vector._op(self, other, operator.add, True)

    def __sub__(self, other):
        return Vector._op(self, other, operator.sub, False)

    def __rsub__(self, other):
        return Vector._op(self, other, operator.sub, True)

    def __mul__(self, other):
        return Vector._op(self, other, operator.mul, False)

    def __rmul__(self, other):
        return Vector._op(self, other, operator.mul, True)

    def __truediv__(self, other):
        return Vector._op(self, other, operator.truediv, False)

    def __rtruediv__(self, other):
        return Vector._op(self, other, operator.truediv, True)

    def __str__(self):
        return str(self._vals)

很明显，所有重载的运算符（__add__、__radd__、...）都有完全相同的代码，除了传递给私有类方法 @ 的 oper 和 rev 参数987654327@。

有没有办法避免所有的复制粘贴和自动（或半自动）这些操作？

Answer 1

第一件事：类方法也可以在实例上使用，所以你应该把你的方法写成

def __someop__(self, other):
    return self._op(self, other, ...)

另外，获取对象的类型 (cls = type(obj)) 也很容易，因此除非您有其他理由将 _op 设为类方法，否则您可以将其设为实例方法。

现在回答您的问题：从技术上讲，您可以使用自定义元类或类装饰器来创建方法，即：

class VectorType(type):
    def __new__(meta, name, bases, attrs):
        todo = [
            # name, operator, reverse
            ("__add__", operator.add, False), 
            ("__sub__", operator.sub, False),
            # etc
            ]
         for name, oper, rev in todo:
             # avoids overriding already defined methods
             if name not in attrs:
                 def f(self, other):  
                     return cls._op(self, other, oper=oper, rev=rev)
                 attrs[name] = f

          return type.__new__(meta, name,   bases, attr) 

class Vector(metaclass=VectorType):
    # etc

但这确实不会改进你现有的代码，除非你有很多（不相关的）类使用这个方案，即使这样，一个普通的 mixin 类也可以工作。

如果您只有这个类的问题，那么您当前的代码是清晰、简单、易读和明确的，IOW 完美的 Pythonic。