python - 根据对象的属性(即使它们是 NoneType)对 Python 中的对象列表进行排序

Question

我有包含发票对象的发票列表。 我想根据它们的日期并使用下面的方法对这些对象进行排序。

from operator import attrgetter
invoices_list.sort(key=attrgetter('date'))

这就是我遇到的错误。

TypeError: can't compare FakeDatetime to NoneType

我想根据日期对对象进行升序排列，并且没有日期应该是第一个。然后其他人应该按升序排列。

$ invoices_list[0].date
$ FakeDatetime(2015, 7, 3, 0, 0)

Answer 1

一个简单的 key 包装器就可以完成这项工作:

class DateKey(object):
    def __init__(self, invoice):
        self.value = invoice.date
    def __lt__(self, other):
        if not isinstance(other, (datetime.date, type(None))):
            return NotImplemented
        elif self.value is None:
            return True
        elif other.value is None:
            return False
        else:
            return self.value < other.value

然后使用它:

invoices_list.sort(key=lambda i: DateKey(i))