子类化datetime.datetime
我正在研究创建一个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)