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

Fixed sum and average queryset function

Browse Source 
* Fixed sum and average map reduce functions for sum and average so that
        it works with mongo dot notation.

* Added unittest cases / updated them for the new changes
This commit is contained in:
kelvinhammond
2013-06-21 09:39:11 -04:00
parent e0d2fab3c3
commit caff44c663
2 changed files with 60 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
mongoengine/queryset/queryset.py
View File

@@ -1068,7 +1068,22 @@ class QuerySet(object):
"""
map_func = Code("""
function() {
emit(1, eval("this." + field) || 0);
function deepFind(obj, path) {
var paths = path.split('.')
, current = obj
, i;
for (i = 0; i < paths.length; ++i) {
if (current[paths[i]] == undefined) {
return undefined;
} else {
current = current[paths[i]];
}
}
return current;
}
emit(1, deepFind(this, field) || 0);
}
""", scope={'field': field})
@@ -1098,8 +1113,24 @@ class QuerySet(object):
"""
map_func = Code("""
function() {
if (this.hasOwnProperty(field))
emit(1, {t: eval("this." + field) || 0, c: 1});
function deepFind(obj, path) {
var paths = path.split('.')
, current = obj
, i;
for (i = 0; i < paths.length; ++i) {
if (current[paths[i]] == undefined) {
return undefined;
} else {
current = current[paths[i]];
}
}
return current;
}
val = deepFind(this, field)
if (val !== undefined)
emit(1, {t: val || 0, c: 1});
}
""", scope={'field': field})