在mongoDB中通过键获取列表中的字典(mongoengine)

时间:2013-06-28 08:28:24

标签: python mongodb mongoengine

我在我的Flask应用程序(mongoengine)中使用mongoDB(mongoHQ)。 我有Document看起来像这样:

{items: [{id: 1}, {id: 2}, {id: 3}]}

有没有办法在单个查询中与dict id: 1进行联系?

目前我正在使用items语句循环遍历next()列表,我希望有更明智的解决方案。感谢

2 个答案:

答案 0 :(得分:5)

我不熟悉MongoEngine,但$ projection运算符可以过滤数组以仅显示匹配的元素。在Mongo shell中:

> db.foo.insert({"items": [{"id": 1}, {"id": 2}, {"id": 3}]})
> db.foo.find({'items.id': 1}, {'items.$': true})
{ "_id" : ObjectId("51ce29b68b178484ff2a01ed"), "items" : [  {  "id" : 1 } ] }

更多信息:http://docs.mongodb.org/manual/reference/projection/positional/

在MongoEngine中你似乎做到了:

Foo.objects(__raw__={'items.id': 1}).only('items.$')

更多:http://mongoengine-odm.readthedocs.org/en/latest/guide/querying.html#raw-queries

答案 1 :(得分:0)

在DB查询或python语句中?

使用python:

In [21]: d = {"items": [{"id": 1}, {"id": 2}, {"id": 3}]}

In [22]: [i for i in d["items"] if i["id"] == 1]
Out[22]: [{'id': 1}]

(将[0]添加到列表comp的末尾以获取字典)

还可以使用生成器:

In [23]: (i for i in d["items"] if i["id"] == 1)
Out[23]: <generator object <genexpr> at 0x052A1D50>

In [24]: _.next()
Out[24]: {'id': 1}

我个人是这样做的,因为我们正在处理来自数据库的返回记录。

如果要直接提取字段(假设我们只希望第一个返回的文档中包含{"id":1}):

In [35]: class Doc(Document):
    ...:     items = ListField(DictField())
    ...:     

In [36]: newdoc = Doc(items=[{"id":1},{"id":2},{"id":3}])

In [37]: newdoc.save()
Out[37]: <Doc: Doc object>

In [38]: for document in [d for d in Doc.objects(items__id=1).first().items if d["id"]==1]:
    ...:     print document
    ...:     
{u'id': 1}

仅提取匹配的字典更难,因为字段 实际上是字典列表。你不是通过修改查询来保存任何速度,只返回匹配的字段部分,所以返回完整字段,然后从python中获取你想要的位。速度相同,更易于维护。