如何在Python中模拟“+”运算符（特别是datetime.date + datetime.timedelta）

我已经解决了 Django 中的一些日期模拟问题，并且最后的障碍（我希望）是以下情况。我有一个 FakeDate 类，它派生于datetime.date，它嘲笑了。

FakeDate 类按预期工作，但是在添加datetime.timedelta到 FakeDate，因为它返回一个真实的datetime.date，而不是模拟。这很重要，因为在第三方库的其他地方有一个isinstance(value, datetime.date)检查，使用 timedelta 时总是会失败。

>>> import mock
>>> import datetime
>>>
>>> class FakeDate(datetime.date):
...     @classmethod
...     def today(cls):
...         return cls(1999, 12, 31)
...
>>> FakeDate.today()
FakeDate(1999, 12, 31)
>>> FakeDate(2000, 1, 1)
FakeDate(2000, 1, 1)
>>> FakeDate(1999, 12, 31) + datetime.timedelta(days=1)
datetime.date(2000, 1, 1)

我希望 FakeDate + timedelta 添加返回一个 FakeDate 对象而不是 datetime.date 对象 - 我想这涉及以某种方式修补 timedelta - 但我如何/在哪里可以做到这一点？

Add a __add__方法给你的FakeDate() class:

class FakeDate(datetime.date):
     @classmethod
     def today(cls):
         return cls(1999, 12, 31)
     def __add__(self, other):
         res = super(FakeDate, self).__add__(other)
         return type(self)(res.year, res.month, res.day)

Demo:

>>> class FakeDate(datetime.date):
...      @classmethod
...      def today(cls):
...          return cls(1999, 12, 31)
...      def __add__(self, other):
...          res = super(FakeDate, self).__add__(other)
...          return type(self)(res.year, res.month, res.day)
... 
>>> FakeDate.today() + datetime.timedelta(days=1)
FakeDate(2000, 1, 1)

请注意，您可以简单地将实际添加委托给datetime.date在这里上课；我们需要做的就是将结果转换回FakeDate()实例。