pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make aggregate_sum/average compatible with pymongo 3.x

Browse Source
This commit is contained in:
Stefan Wojcik
2015-06-08 13:46:19 -07:00
parent dd095279c8
commit 1233780265
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mongoengine/queryset/base.py
View File

@@ -1261,6 +1261,11 @@ class BaseQuerySet(object):
{ '$match': self._query }, { '$match': self._query },
{ '$group': { '_id': 'sum', 'total': { '$sum': '$' + field } } } { '$group': { '_id': 'sum', 'total': { '$sum': '$' + field } } }
]) ])
if IS_PYMONGO_3:
result = list(result)
if result:
return result[0]['total']
else:
if result['result']: if result['result']:
return result['result'][0]['total'] return result['result'][0]['total']
return 0 return 0
@@ -1333,6 +1338,11 @@ class BaseQuerySet(object):
{ '$match': self._query }, { '$match': self._query },
{ '$group': { '_id': 'avg', 'total': { '$avg': '$' + field } } } { '$group': { '_id': 'avg', 'total': { '$avg': '$' + field } } }
]) ])
if IS_PYMONGO_3:
result = list(result)
if result:
return result[0]['total']
else:
if result['result']: if result['result']:
return result['result'][0]['total'] return result['result'][0]['total']
return 0 return 0