在 python mongoengine 中通过计算属性查询

Question

我想知道是否可以使用 python 中的 mongoengine 通过计算属性来查询 MongoDB 中的文档。

目前，我的模型如下所示：

class SnapshotIndicatorKeyValue(db.Document):
    meta = {"collection": "snapshot_indicator_key_values"}

    snapshot_id = db.ObjectIdField(nullable=False)
    indicator_key_id = db.ObjectIdField(nullable=False)
    value = db.FloatField(nullable=False)
    created_at = db.DateTimeField()
    updated_at = db.DateTimeField()

    @property
    def snapshot(self):
        return Snapshot.objects(id=self.snapshot_id).first()

    def indicator_key(self):
        return IndicatorKey.objects(id=self.indicator_key_id).first()

当我执行 SnapshotIndicatorKeyValue .objects().first().snapshot 操作时，我可以访问 snapshot 属性。

但是当我尝试查询它时，它不起作用。例如：

SnapshotIndicatorKeyValue.objects(snapshot__date_time__lte=current_date_time)

我收到错误“mongoengine.errors.InvalidQueryError：无法解析字段“快照”``

有什么方法可以使其与查询一起使用吗？ 我需要根据snapshot 的属性查询SnapshotIndicatorKeyValue。

Answer 1

为了直接通过 mongoengine 查询 snapshot 属性，您可以在 SnapshotIndicatorKeyValue< 中引用相关的 snapshot 对象而不是 snapshot_id /code> 文档定义。

使用 参考字段 修改后的模型将如下所示：

from mongoengine import Document, ReferenceField

class Snapshot(Document)
property_abc = RelevantPropertyHere() # anything you need

class SnapshotIndicatorKeyValue(Document):
snapshot = ReferenceField(Snapshot)

您将连续保存 Snapshot 实例和 SnapshotIndicatorKeyValue 实例，如下所示：

sample_snapshot = Snapshot(property_abc=relevant_value_here) # anything you need
sample_snapshot.save()

sample_indicatorkeyvalue = SnapshotIndicatorKeyValue()
sample_indicatorkeyvalue.snapshot = sample_snapshot
sample_indicatorkeyvalue.save()

然后您可以引用 snapshot 的任何属性 通过：

SnapshotIndicatorKeyValue.objects.first().snapshot.property_abc