python 中重载增强算术赋值

Question

我是 Python 新手，所以如果这是一个愚蠢的问题，请提前道歉。

对于赋值，我需要为类 myInt 重载增强算术赋值（+=、-=、/=、*=、**=、%=）。我检查了Python文档，这就是我得出的结论：

def __iadd__(self, other):

    if isinstance(other, myInt):
        self.a += other.a
    elif type(other) == int:
        self.a += other
    else:
        raise Exception("invalid argument")

self.a和other.a指的是存储在每个类实例中的int。我尝试按如下方式对此进行测试，但每次我得到“无”而不是预期值 5：

c = myInt(2)
b = myInt(3)
c += b
print c

任何人都可以告诉我为什么会发生这种情况吗？提前致谢。

Answer 1

您需要将 return self 添加到您的方法中。说明：

当 type(a) 有一个特殊方法 __iadd__ 时，a += b 的语义定义为：

  a = a.__iadd__(b)

所以 if < code>__iadd__ 返回与 self 不同的内容，这就是操作后将绑定到名称 a 的内容。由于缺少 return 语句，您发布的方法相当于 return None 的方法。

Answer 2

Python 中的增强运算符必须返回要分配给它们被调用的名称的最终值，通常（在您的情况下）self。与所有 Python 方法一样，缺少 return 语句意味着返回 None。

另外，

• 永远不要引发Exception，这是不可能被理智地捕获的。执行此操作的代码必须为 except Exception，它将捕获所有 异常。在这种情况下，您需要 ValueError 或 TypeError。
• 不要使用 type(foo) == SomeType 进行类型检查。在这种（以及几乎所有）情况下，isinstance 效果更好，或者至少是相同的。
• 每当您创建自己的类型（例如 myInt）时，都应该使用大写字母命名它，以便人们可以将其识别为类名。

Answer 3

是的，你需要“返回自我”，它看起来像这样：

def __iadd__(self, other):
    if isinstance(other, myInt):
        self.a += other.a
        return self
    elif type(other) == int:
        self.a += other
        return self
    else:
        raise Exception("invalid argument")