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

Fixed item_frequency methods to handle null values

Browse Source 
[fixes #216]
This commit is contained in:
Ross Lawley
2011-07-11 16:50:31 +01:00
parent 031c507fde
commit 1452d3fac5
2 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
mongoengine/queryset.py
View File

@@ -1435,7 +1435,7 @@ class QuerySet(object):
path = '{{~%(field)s}}'.split('.');
field = this;
for (p in path) { field = field[path[p]]; }
if (field.constructor == Array) {
if (field && field.constructor == Array) {
field.forEach(function(item) {
emit(item, 1);
});
@@ -1481,7 +1481,7 @@ class QuerySet(object):
db[collection].find(query).forEach(function(doc) {
field = doc;
for (p in path) { field = field[path[p]]; }
if (field.constructor == Array) {
if (field && field.constructor == Array) {
total += field.length;
} else {
total++;
@@ -1497,7 +1497,7 @@ class QuerySet(object):
db[collection].find(query).forEach(function(doc) {
field = doc;
for (p in path) { field = field[path[p]]; }
if (field.constructor == Array) {
if (field && field.constructor == Array) {
field.forEach(function(item) {
frequencies[item] = inc + (isNaN(frequencies[item]) ? 0: frequencies[item]);
});
@@ -1509,8 +1509,11 @@ class QuerySet(object):
return frequencies;
}
"""
return self.exec_js(freq_func, field, normalize=normalize)
data = self.exec_js(freq_func, field, normalize=normalize)
if 'undefined' in data:
data[None] = data['undefined']
del(data['undefined'])
return data
def __repr__(self):
limit = REPR_OUTPUT_SIZE + 1