我们想让类支持各种比较操作符,但是又不想编写那么多的特殊方法。
毕竟只定义一个方法不算什么,但是如果要实现每种可能的比较操作,那么实现这么多特殊方法就会很快变得繁琐。
我们可以使用functools.total_ordering装饰器来简化这个过程。使用它,可以用来装饰一个类,然后只要定义eq()方法即支持==操作符的方法以及另一个比较方法(lt()小于操作符,le()小于等于操作符,gt()大于操作符,ge()大于等于操作符)。那么装饰器就可以自动为我们实现其他比较方法。
下面是使用了该装饰器的类代码:
from functools import total_ordering
@total_ordering
class person:
def __init__(self, name, height):
self.name = name
self.height = height
def __eq__(self, other):
return self.height == other.height
def __lt__(self, other):
return self.height < other.height
def __str__(self):
return '{} is {} cm'.format(self.name, self.height)
演示代码,支持各类的比较操作符,并且支持max()和min()函数
>>> amos = person('Amos', 175)
>>> ming = person('ming', 177)
>>> lilei = person('lilei', 172)
>>> persons = [amos, ming, lilei]
>>> print('Is ming as high as lilei?', ming == lilei)
Is ming as high as lilei? False
>>> print('Is Amos taller or equal to ming?', amos >= ming)
Is Amos taller or equal to ming? False
>>> print('Who is the tallest one?', max(persons))
Who is the tallest one? ming is 177 cm
装饰器total_ordering从字面上定义了从每个比较方法到其他所有需要该方法的映射关系。自行编写全部的特殊方法并不难,但是使用了@total_ordering显然让这一过程更加简单。