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

removed usage of 'pushAll' operator

Browse Source
This commit is contained in:
Calgary Michael 2018-02-02 21:47:04 -06:00
parent 9b04391f82
commit 9e80da705a
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mongoengine/queryset/transform.py
View File

@ -335,7 +335,7 @@ def update(_doc_cls=None, **update):
value = {key: value} value = {key: value}
elif op == 'addToSet' and isinstance(value, list): elif op == 'addToSet' and isinstance(value, list):
value = {key: {'$each': value}} value = {key: {'$each': value}}
elif op == 'push': elif op in ('push', 'pushAll'):
if parts[-1].isdigit(): if parts[-1].isdigit():
key = parts[0] key = parts[0]
position = int(parts[-1]) position = int(parts[-1])
@ -345,7 +345,13 @@ def update(_doc_cls=None, **update):
value = [value] value = [value]
value = {key: {'$each': value, '$position': position}} value = {key: {'$each': value, '$position': position}}
else: else:
value = {key: value} if op == 'pushAll':
op = 'push' # convert to non-deprecated keyword
if not isinstance(value, (set, tuple, list)):
value = [value]
value = {key: {'$each': value}}
else:
value = {key: value}
else: else:
value = {key: value} value = {key: value}
key = '$' + op key = '$' + op