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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mongoengine/queryset/base.py
View File

@ -1261,8 +1261,13 @@ class BaseQuerySet(object):
{ '$match': self._query }, { '$match': self._query },
{ '$group': { '_id': 'sum', 'total': { '$sum': '$' + field } } } { '$group': { '_id': 'sum', 'total': { '$sum': '$' + field } } }
]) ])
if result['result']: if IS_PYMONGO_3:
return result['result'][0]['total'] result = list(result)
if result:
return result[0]['total']
else:
if result['result']:
return result['result'][0]['total']
return 0 return 0
def average(self, field): def average(self, field):
@ -1333,8 +1338,13 @@ class BaseQuerySet(object):
{ '$match': self._query }, { '$match': self._query },
{ '$group': { '_id': 'avg', 'total': { '$avg': '$' + field } } } { '$group': { '_id': 'avg', 'total': { '$avg': '$' + field } } }
]) ])
if result['result']: if IS_PYMONGO_3:
return result['result'][0]['total'] result = list(result)
if result:
return result[0]['total']
else:
if result['result']:
return result['result'][0]['total']
return 0 return 0
def item_frequencies(self, field, normalize=False, map_reduce=True): def item_frequencies(self, field, normalize=False, map_reduce=True):