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

Document._get_update_doc helper method

Browse Source
This commit is contained in:
Stefan Wojcik 2017-05-07 19:26:10 -04:00
parent e2a0b42d03
commit 2cf23e33e3
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mongoengine/document.py
View File

@ -300,7 +300,7 @@ class Document(BaseDocument):
created. created.
:param force_insert: only try to create a new document, don't allow :param force_insert: only try to create a new document, don't allow
updates of existing documents updates of existing documents.
:param validate: validates the document; set to ``False`` to skip. :param validate: validates the document; set to ``False`` to skip.
:param clean: call the document clean method, requires `validate` to be :param clean: call the document clean method, requires `validate` to be
True. True.
@ -441,6 +441,21 @@ class Document(BaseDocument):
return object_id return object_id
def _get_update_doc(self):
"""Return a dict containing all the $set and $unset operations
that should be sent to MongoDB based on the changes made to this
Document.
"""
updates, removals = self._delta()
update_doc = {}
if updates:
update_doc['$set'] = updates
if removals:
update_doc['$unset'] = removals
return update_doc
def _save_update(self, doc, save_condition, write_concern): def _save_update(self, doc, save_condition, write_concern):
"""Update an existing document. """Update an existing document.
@ -466,15 +481,10 @@ class Document(BaseDocument):
val = val[ak] val = val[ak]
select_dict['.'.join(actual_key)] = val select_dict['.'.join(actual_key)] = val
updates, removals = self._delta() update_doc = self._get_update_doc()
update_query = {} if update_doc:
if updates:
update_query['$set'] = updates
if removals:
update_query['$unset'] = removals
if updates or removals:
upsert = save_condition is None upsert = save_condition is None
last_error = collection.update(select_dict, update_query, last_error = collection.update(select_dict, update_doc,
upsert=upsert, **write_concern) upsert=upsert, **write_concern)
if not upsert and last_error['n'] == 0: if not upsert and last_error['n'] == 0:
raise SaveConditionError('Race condition preventing' raise SaveConditionError('Race condition preventing'