Python Functools – total_ordering()

在Python中使用functools模块中的total_ordering()函数可以使自定义类中比较运算更简便。这个函数只需要实现其中的__eq__()和一个其它的比较运算方法(lt,le,gt,ge)就可以自动地生成所有这些比较运算方法了。下面我们就来看一下这个函数的使用方法以及示例。

使用方法

total_ordering()函数只需要作用在一个类上，这个类只需要实现其中的__eq__()和一个其它比较运算方法，函数就可以自动为这个类生成所有的比较运算方法了。

语法

functools.total_ordering(cls)

参数

• cls：必需，实现了__eq__()和一个其它比较运算方法的类。

返回值

这个函数不会返回任何值，但是会为cls类生成所有的比较运算方法。

示例

下面我们来看一个例子，利用total_ordering()函数来实现一个自定义的日期类：

import functools

@functools.total_ordering
class Date:
    def __init__(self, year, month, day):
        self.year = year
        self.month = month
        self.day = day

    def __eq__(self, other):
        if isinstance(other, Date):
            return ((self.year, self.month, self.day) ==
                    (other.year, other.month, other.day))
        return NotImplemented

    def __lt__(self, other):
        if isinstance(other, Date):
            return ((self.year, self.month, self.day) <
                    (other.year, other.month, other.day))
        return NotImplemented

这个类实现了__eq__()和__lt__()方法。我们可以通过total_ordering()函数自动为这个类生成其它比较运算方法。现在我们来使用这个类：

d1 = Date(2019, 1, 1)
d2 = Date(2018, 12, 31)

print(d1 > d2)
print(d1 >= d2)
print(d1 < d2)
print(d1 <= d2)
print(d1 == d2)
print(d1 != d2)

输出：

True
True
False
False
False
True

可以看到我们并没有实现__gt__()和__ge__()方法，但是通过total_ordering()函数，我们可以使用这些运算符了。

小结

使用total_ordering()函数可以使自定义类中比较运算更简便。通过实现__eq__()和一个其它比较运算方法，我们就可以自动地生成所有比较运算方法。这个函数可以提高程序开发效率，减少代码冗余。