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

Added support for addToSet and each

Browse Source 
fixes MongoEngine/mongoengine#33
This commit is contained in:
Ross Lawley
2012-07-11 14:22:50 +01:00
parent 7a1b110f62
commit b58bf3e0ce
4 changed files with 31 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mongoengine/queryset.py
View File

@@ -1387,11 +1387,16 @@ class QuerySet(object):
# Convert value to proper value
field = cleaned_fields[-1]
if op in (None, 'set', 'push', 'pull', 'addToSet'):
if op in (None, 'set', 'push', 'pull'):
if field.required or value is not None:
value = field.prepare_query_value(op, value)
elif op in ('pushAll', 'pullAll'):
value = [field.prepare_query_value(op, v) for v in value]
elif op == 'addToSet':
if isinstance(value, (list, tuple, set)):
value = [field.prepare_query_value(op, v) for v in value]
elif field.required or value is not None:
value = field.prepare_query_value(op, value)
key = '.'.join(parts)
@@ -1407,6 +1412,8 @@ class QuerySet(object):
parts.reverse()
for key in parts:
value = {key: value}
elif op == 'addToSet' and isinstance(value, list):
value = {key: {"$each": value}}
else:
value = {key: value}
key = '$' + op