如何擴展datetime.timedelta？

Question

我正在嘗試擴展 Python datetime.timedelta以用於越野比賽結果。 我想從格式為u"mm:ss.s"的字符串構造一個對象。 我能夠使用工廠設計模式和@classmethod注釋來完成此操作。 我將如何通過覆蓋__init__和/或__new__來完成同樣的__new__ ？

使用下面的代碼，構造一個對象會引發一個 TypeError。 請注意，未調用__init__ ，因為未打印'in my __init__' 。

import datetime
import re

class RaceTimedelta(datetime.timedelta):
    def __init__(self, timestr = ''):
        print 'in my __init__'
        m = re.match(r'(\d+):(\d+\.\d+)', timestr)
        if m:
            mins = int(m.group(1))
            secs = float(m.group(2))
            super(RaceTimedelta, self).__init__(minutes = mins, seconds = secs)
        else:
            raise ValueError('timestr not in format u"mm:ss.d"')

這是錯誤：

>>> from mytimedelta import RaceTimedelta
>>> RaceTimedelta(u'24:45.7')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported type for timedelta days component: unicode
>>> 

如果我將代碼從__init__移動到__new__ ，我會得到以下信息。 請注意，這一次，輸出顯示調用了我的__new__函數。

>>> RaceTimedelta(u'24:45.7')
in my __new__
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "mytimedelta.py", line 16, in __new__
    super(RaceTimedelta, self).__new__(minutes = mins, seconds = secs)
TypeError: datetime.timedelta.__new__(): not enough arguments
>>> 

Answer 1

顯然timedelta對象是不可變的，這意味着它們的值實際上是在類的__new__()方法中設置的——因此您需要覆蓋該方法而不是它的__init__() ：

import datetime
import re

class RaceTimedelta(datetime.timedelta):
    def __new__(cls, timestr=''):
        m = re.match(r'(\d+):(\d+\.\d+)', timestr)
        if m:
            mins, secs = int(m.group(1)), float(m.group(2))
            return super(RaceTimedelta, cls).__new__(cls, minutes=mins, seconds=secs)
        else:
            raise ValueError('timestr argument not in format "mm:ss.d"')

print RaceTimedelta(u'24:45.7')

輸出：

0:24:45.700000

順便說一句，我覺得奇怪的是，您為timestr關鍵字參數提供了一個默認值，該值將被視為非法並引發ValueError 。