子类化datetime.datetime

2 投票
1 回答
3587 浏览
提问于 2025-04-18 14:50

我正在研究创建一个Python的日期时间子类,这个子类在创建时可以提供一个默认的时区。

为了让这个问题简单明了,我们假设我总是想把我的日期时间硬编码为UTC时区。

我搞不明白为什么下面这个代码能正常工作:

import datetime, dateutil.tz

def foo(*args, **kwargs):
    kwargs['tzinfo'] = dateutil.tz.tzutc()
    return datetime.datetime(*args, **kwargs)

但下面这个代码却不行:

class Foo(datetime.datetime):
    def __init__(self, *args, **kwargs):
        kwargs['tzinfo'] = dateutil.tz.tzutc()
        super(Foo, self).__init__(*args, **kwargs)

运行第一个方法时,我得到了我预期的日期时间对象:

>>> foo(2012, 11, 10)
datetime.datetime(2012, 11, 10, 0, 0, tzinfo=tzutc())
>>> foo(2012, 11, 10).tzinfo is None
False

但是创建Foo类的实例似乎没有设置tzinfo对象。

>>> Foo(2012, 11, 10)
Foo(2012, 11, 10, 0, 0)
>>> Foo(2012, 11, 10).tzinfo is None
True

有没有人知道这是怎么回事?

谢谢!

1 个回答

3

为了完整性,这里提供了解决方案,详细内容可以参考为什么我不能继承datetime.date?,感谢dano提供的链接。

因为datetime对象是不可变的,所以额外的功能需要在new方法中实现:

class Foo(datetime.datetime):
    def __new__(cls, *args, **kwargs):
        kwargs['tzinfo'] = dateutil.tz.tzutc()
        return datetime.datetime.__new__(cls, *args, **kwargs)

撰写回答