让类支持比较操作的同时,不必编写大量的特殊方法

    我们想让类支持各种比较操作符,但是又不想编写那么多的特殊方法。

    毕竟只定义一个方法不算什么,但是如果要实现每种可能的比较操作,那么实现这么多特殊方法就会很快变得繁琐。
    我们可以使用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显然让这一过程更加简单。

你可能感兴趣的:(让类支持比较操作的同时,不必编写大量的特殊方法)