转换MongoDB查询做PyMongo格式

时间:2018-05-23 14:15:28

标签: python mongodb pymongo mongoengine

民间,

我有一个MongoDB查询,我想在我的Python代码中执行:

db.authentication.aggregate(
    {$group:{_id:{'cpf':"$person.cpf", 'data':"$created_at"}, 'chamadas':{$push:"$device_data.metaChamadas.duration"}}},
    {$project:{
        'cpf':"$_id.cpf",
        'tamanho':{$size:{$arrayElemAt:["$chamadas",0]}}
    }}
)

我使用的是Mongoengine,因为它支持原始PyMongo查询,但到目前为止我无法弄清楚转换:

documents = list(Authentication.objects(__raw__={

PYMONGO QUERY HERE

}).limit(10))

有什么想法吗?

1 个答案:

答案 0 :(得分:2)

所有mongoengine.queryset.QuerySet都有一个聚合方法,可以使用聚合命令作为参数调用。

翻译您的MongoDb查询,您应该有这样的查询:

pipeline = (
        {
        '$group': {
            '_id': {
                'cpf': '$person.cpf',
                'data': '$created_at'
            }, 
            'chamadas': {
                '$push': '$device_data.metaChamadas.duration'
            }
        }
    },
    {
        '$project': {
            'cpf': '$_id.cpf',
            'tamanho':  {
                '$size': {
                    '$arrayElemAt': ['$chamadas', 0]
                }
            }
        }
    }    
)

data = Authentication.objects.aggregate(*pipeline)
print(list(data))