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

Merge pull request #1636 from erdenezul/fix_shard_key_modify

Browse Source 
Fix Document.modify fail on sharded collection #1569
This commit is contained in:
erdenezul
2018-02-06 07:25:42 +08:00
committed by GitHub
parent 9b04391f82 1eae97731f
commit ffdfe99d37
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mongoengine/document.py
View File

@@ -280,6 +280,9 @@ class Document(BaseDocument):
elif query[id_field] != self.pk: elif query[id_field] != self.pk:
raise InvalidQueryError('Invalid document modify query: it must modify only this document.') raise InvalidQueryError('Invalid document modify query: it must modify only this document.')
# Need to add shard key to query, or you get an error
query.update(self._object_key)
updated = self._qs(**query).modify(new=True, **update) updated = self._qs(**query).modify(new=True, **update)
if updated is None: if updated is None:
return False return False