Added update() convenience method to a document

Thanks to @dcrosta for the initial code
[closes #229]
This commit is contained in:
Ross Lawley
2011-07-13 14:15:46 +01:00
parent 7a3412dc13
commit a4c197a83c
3 changed files with 36 additions and 1 deletions

View File

@@ -193,6 +193,18 @@ class Document(BaseDocument):
reset_changed_fields(self)
signals.post_save.send(self.__class__, document=self, created=created)
def update(self, **kwargs):
"""Performs an update on the :class:`~mongoengine.Document`
A convenience wrapper to :meth:`~mongoengine.QuerySet.update`.
Raises :class:`OperationError` if called on an object that has not yet
been saved.
"""
if not self.pk:
raise OperationError('attempt to update a document not yet saved')
return self.__class__.objects(pk=self.pk).update_one(**kwargs)
def delete(self, safe=False):
"""Delete the :class:`~mongoengine.Document` from the database. This
will only take effect if the document has been previously saved.