具有整数模拟的 Python 类

Question

给出以下示例：

class Foo(object):
    def __init__(self, value=0):
        self.value=value

    def __int__(self):
        return self.value

我想要一个类 Foo，它充当整数（或浮点数）。所以我想做以下事情：

f=Foo(3)
print int(f)+5 # is working
print f+5 # TypeError: unsupported operand type(s) for +: 'Foo' and 'int'

第一个语句 print int(f)+5 有效，因为有两个整数。第二个失败了，因为我必须实现 __add__ 来对我的类执行此操作。

因此，为了实现整数行为，我必须实现所有整数模拟方法。我该如何解决这个问题。我尝试继承int，但这次尝试没有成功。

更新

如果您想使用 __init__，则从 int 继承会失败：

class Foo(int):
    def __init__(self, some_argument=None, value=0):
        self.value=value
        # do some stuff

    def __int__(self):
        return int(self.value)

如果您随后调用：

f=Foo(some_argument=3)

您将得到：

TypeError: 'some_argument' is an invalid keyword argument for this function

使用 Python 2.5 和 2.6 进行测试

Answer 1

在 Python 2.4+ 中，继承 int 有效：

class MyInt(int):pass
f=MyInt(3)
assert f + 5 == 8

Answer 2

您需要重写 __new__，而不是 __init__：

class Foo(int):
    def __new__(cls, some_argument=None, value=0):
        i = int.__new__(cls, value)
        i._some_argument = some_argument
        return i

    def print_some_argument(self):
        print self._some_argument

现在您的类按预期工作：

>>> f = Foo(some_argument="I am a customized int", value=10)
>>> f
10
>>> f + 8
18
>>> f * 0.25
2.5
>>> f.print_some_argument()
I am a customized int

有关重写 new 的更多信息可以在 统一 Python 2.2 中的类型和类。

Answer 3

尝试使用最新版本的 python。您的代码适用于 2.6.1。