Как подсчитать курсор агрегации pymongo без итерации

Question

Я хочу получить общее количество записей в агрегированном курсоре в версии pymongo версии 3.0+. Есть ли способ получить общий счет без повторения курсора?

cursor = db.collection.aggregate([{"$match": options},{"$group": {"_id": groupby,"count": {"$sum":1}}} ]) 
cursorlist = [c for c in cursor] 
print len(cursorlist) 

Есть ли способ пропустить вышеупомянутую итерацию?

Answer 1

Вы можете добавить еще один групповой конвейер, в котором вы указываете значение _idNone, чтобы вычислить накопленные значения для всех входных документов в целом, здесь вы можете получить общее количество, а также исходные сгруппированные счета, хотя в скопированном массиве:

>>> pipeline = [ 
...  {"$match": options}, 
...  {"$group": {"_id": groupby, "count": {"$sum":1}}}, 
...  {"$group": {"_id": None, "total": {"$sum": 1}, "details":{"$push":{"groupby": "$_id", "count": "$count"}}}} 
... ] 
>>> list(db.collection.aggregate(pipeline))