如何将 CodecOptions 应用于 mongoengine 集合以获取 tzinfo？

Question

使用 mongoengine Document ，当我从 mongo 加载DateTimeField时，它缺少tzinfo 。 我们的策略应用程序希望所有日期时间都有tzinfo 。

当我保存到 mongo 时，我知道 pymongo 会在时区上做正确且可预测的事情； 如果它是天真的，则存储为 UTC 时间，如果它有时区，则将其转换为 UTC，然后存储为 UTC 时间。 到目前为止一切都很好。

但是，当我加载时， DateTimeField总是给我一个天真的datetime 。 我知道这个datetime是 UTC，所以我可以根据需要添加tzinfo ，但我必须在我的应用程序的几十个地方这样做，这是未来时区错误的保证，除非我使用自定义字段类型（见下面的例子）。

参考： https://api.mongodb.com/python/current/examples/datetimes.html我知道 pymongo 支持将tzinfo从数据库到达时放到对象上。 我也知道我自己可以做到这一点，如下面的最小示例所示。 DateTimeTZField在to_python添加了tzinfo 。

from datetime import datetime

from mongoengine import connect, Document, fields

from pytz import timezone

def utcnowTZ():
    return datetime.utcnow().replace(tzinfo=timezone('UTF'))

class DateTimeTZField(fields.DateTimeField):
    """
    This seems like a hack. I would like to use CodecOptions instead
    """
    def to_python(self, value):
        converted = super(DateTimeTZField, self).to_python(value)
        return converted.replace(tzinfo=timezone('UTC'))

class Thing(Document):
    dtTZ = DateTimeTZField(default=utcnowTZ)
    dtXX = fields.DateTimeField(default=utcnowTZ)

connect(host="mongodb://localhost/datetimewithtz")
Thing.objects().delete()

t1 = Thing()
print '%r.dtXX (default): %s' % (t1, t1.dtXX)
print '%r.dtTZ (default): %s' % (t1, t1.dtTZ)
t1.save()
print 'saved %r' % t1.id; print
t1 = Thing.objects(id=t1.id).first()
print 'reloaded %r' % t1.id
print '%r.dtXX (loaded ): %s' % (t1, t1.dtXX)
print '%r.dtTZ (loaded ): %s' % (t1, t1.dtTZ)

在不替换整个应用程序中的 DateTimeField 的情况下，有没有办法使用 bson CodecOptions使其全局应用于所有 DateTimeFields：从数据库加载时，它们应该附加 tzinfo ？

Answer 1

怎么样

connect(...,  tz_aware=True,...)

参数将被转发到MongoClient(..., tz_aware=True, ...)

Answer 2

我的理解是 mongoengine 不支持每个模型的codec_options 。 如果是这样，它们应该是这里opts字典的一部分。

但是，当 Collection 没有自己的任何codec_options时，PyMongo 默认为数据库的codec_options ，并且您可以在向数据库发出任何请求之前通过在应用程序中手动调用register_connection来设置这些。

在上面的示例中，您应该像这样传递codec_options ：

with_timezone = CodecOptions()
connect(host="mongodb://localhost/datetimewithtz", codec_options=with_timezone)