在EmbeddedDocumentList中匹配EmbeddedDocument

时间:2015-09-06 05:48:06

标签: python django mongodb mongoengine

所以我有这个测试数据,例如mongo,Cart model:

{
"_id" : ObjectId("55eb513c516ddc8fa6e68886"),
    "user" : ObjectId("55e3f236516ddc78296968be"),
    "items" : [
        {
            "item" : ObjectId("55eb10b8516ddc7508dba7c2"),
            "quantity" : 1,
            "added_date" : ISODate("2015-09-05T20:32:16.527Z"),
            "coupons" : ObjectId("55eb10cd516ddc751d3d5e25"),
            "order_type" : [
                "in_store",
                "curbside"
            ]
        },
        {
            "item" : ObjectId("55eb10b8516ddc7508dba7cc"),
            "quantity" : 1,
            "added_date" : ISODate("2015-09-05T20:32:16.527Z"),
            "coupons" : ObjectId("55eb10cd516ddc751d3d5e25")
        },
        {
            "item" : ObjectId("55eb10b8516ddc7508dba7c8"),
            "quantity" : 1,
            "added_date" : ISODate("2015-09-05T20:32:16.527Z"),
            "coupons" : ObjectId("55eb10cd516ddc751d3d5e25")
        }
    ]
}

以下是我的模特:

class CartItem(mongoengine.EmbeddedDocument):
    item = mongoengine.ReferenceField('Item')
    quantity = mongoengine.IntField()
    added_date = mongoengine.DateTimeField( default=timezone.now() )
    coupons = mongoengine.ReferenceField('Coupon')
    order_type = mongoengine.ListField(mongoengine.StringField(choices=ORDER_TYPE_CHOICES))

class Cart(mongoengine.Document):
    user = mongoengine.ReferenceField('User')
    items = mongoengine.EmbeddedDocumentListField(CartItem)
    savings = mongoengine.DecimalField(precision=2, default=0)

现在我希望能够获得 CartItems ,例如只有order_type为" in_store" 。所以我这样的事情:

cart = Cart.objects.filter(id="id")
items = cart.items
curbside_items = items.filter(order_type="curbside")

这不会返回任何内容。所以我试试这个:

curbside_items = items.filter(order_type__math="curbside")

但这并没有给我任何回报。所以我的问题是如何在mongoengine中做到这一点。谢谢你!

1 个答案:

答案 0 :(得分:1)

MongoEngine使用"double underscore"属性转换为"dot notation"形式的MongoDB查询。因此,您正在寻找的属性的“父”是“项目”,这意味着您使用完整路径引用,如下所示:

Cart.objects.filter(items__order_type="in_store")

文档中包含“in_store”作为order_type数组中的items条目。请注意,这将返回“整个文档”而不仅仅是匹配的条目。还有其他方法可以处理这种反应。

一种方法是使用底层的pymongo驱动程序获得“原始”结果:

Cart._get_collection().find(
    { "items.order_type": "in_store" },
    { "items.$": 1 }
)

利用positional $运算符只返回匹配的数组元素。

或者对于“多”匹配,您需要.aggregate()

Cart._get_collection().aggregate([
    { "$match": {
        "items.order_type": "in_store"
    }},
    { "$redact": {
        "$cond": {
            "if": { "$eq": [ { "$ifNull": [ "$order_type", "in_store" ] }, "in_store" ] },
            "then": "$$DESCEND",
            "else": "$$PRUNE"
        }
    }}
])

或者通过$filter在以后的版本中更有意义:

Cart._get_collection().aggregate([
    { "$match": {
        "items.order_type": "in_store"
    }},
    { "$project": {
        "items": {
            "$filter": {
                "input": "$items",
                "as": "el",
                "cond": { "$eq": [ "$$el.order_type", "in_store" ] }
            }
        }
    }}
])

“djangoesque”行中的标准mongoengine查询不支持此类操作,只返回未修改的整个文档。例外情况是$slice,其中有一个选项可以完成MongoDB原生对象的工作。