python >= 类 上的运算符

python >= operator on classes

我有一个关于 python '>=' 行为的问题。

我有一个旧的 TimeStamp class,它包含(小时、分钟)元组并提供一些方法,例如 __eq____gt____lt__

我正在重构它以考虑天和秒,并将数据存储为总秒数。这里我也实现了__eq____gt____lt__

但是,在代码中,我为此 class 使用 >= 运算符,虽然旧的 class 版本工作正常,但我正在使用新版本

TypeError: unorderable types: TimeStamp() >= TimeStamp() error.

代码如下:

class TimeStamp(tuple): # OLD, WORKING VERSION
    """TimeStamp, hh:mm tuple supporting comparison and addition"""
    __slots__ = ()
    def __new__(cls, *args):
        if len(args) == 1:  # tuple entrance
            hour, minute = args[0]
        elif len(args) == 2: # hour, minute entrance
            hour, minute = args[0], args[1]
        else:
            raise TypeError('wrong input to TimeStamp')
        div, minute = divmod(minute, 60)
        hour += div
        _, hour = divmod(hour, 24)
        return tuple.__new__(cls, (hour, minute))

    @property
    def abs_min(self):
        return self.hour * 60 + self.minute

    def __gt__(self, rhs):
        return self.abs_min > rhs.abs_min

    def __lt__(self, rhs):
        return self.abs_min < rhs.abs_min

    def __eq__(self, rhs):
        return self.abs_min == rhs.abs_min

新版本:

class TimeStamp:
    def __init__(self, *args):
        for argument in args:
            if not isinstance(argument, int):
                raise TypeError("Can only build TimeStamp from ints, not: " + str(argument))

        if len(args) == 1:  # init by abs
            self.abs = args[0] # put the ELEMENT, not the tuple itself
        elif len(args) == 2:    # init by hour:minute
            hour, minute = args
            self.abs = hour * 60 * 60 + minute * 60
        elif len(args) == 4:    #init by day:hour:minute:second
            day, hour, minute, second = args
            self.abs = day * 24 * 60 * 60 + hour * 60 * 60 + minute * 60 + second
        else:
            raise TypeError("wrong data for TimeStamp: " + str(args))

    def __eq__(self, other):
        if isinstance(other, TimeStamp):
            return self.abs == other.abs
        else:
            raise TypeError("wrong argument for comparison: " + str(other))

    def __gt__(self, other):
        if isinstance(other, TimeStamp):
            return self.abs > other.abs
        else:
            raise TypeError("wrong argument for comparison: " + str(other))

    def __lt__(self, other):
        if isinstance(other, TimeStamp):
            return self.abs < other.abs
        else:
            raise TypeError("wrong argument for comparison: " + str(other))

现在进行比较部分:

if args[1] >= self.start:
>>TypeError: unorderable types: TimeStamp() >= TimeStamp()

我找到了两个修复方法:首先,将我的比较行替换为

if args[1] > self.start or args[1] == self.start:

或替代方法,添加

def __ge__(self, other):
    if isinstance(other, TimeStamp):
        return self.abs >= other.abs
    else:
        raise TypeError("wrong argument for comparison: " + str(other))

我的新 class。但是,旧版本确实无法使用这些修复程序。在我看来,好像 Python 停止推导 ((a>b) 或 (a==b)) 暗示 (a>=b)。但是为什么以前有效呢?跟我subclassing tuple有关系吗?

PS。不要被我的 __init__ 代码吓到,我把它包含进来是为了完整性。它应该是类似超载的,但我可能会以非 pythonic 方式进行(仍在学习)

旧的之所以有效,是因为它继承自 tuple,并且 tuple 提供了 __ge__。您的新版本没有继承自 tuple,因此它没有 __ge__ 方法。

即使在您的旧版本中,您的 __gt____lt__ 方法在使用 >= 时也从未被调用(因为您可以通过将 print 放入这些方法中来验证).底层 tuple.__ge__ 被调用。但是,对于您的情况,效果是一样的,所以您没有注意到。也就是说,假设 "minutes" 数字总是小于 60,以通常的方式比较 (hours, minutes) 元组等同于比较 60*hours + minutes。所以我认为如果你从元组继承,你真的不需要定义比较方法。