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

guard against potentially destructive updates with no update parameters

Browse Source
This commit is contained in:
Leo Honkanen
2011-07-12 16:10:47 +03:00
parent cace665858
commit 7f0d3638ba
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mongoengine/queryset.py
View File

@@ -1268,6 +1268,9 @@ class QuerySet(object):
.. versionadded:: 0.2 .. versionadded:: 0.2
""" """
if not update:
raise OperationError("No update parameters, would remove data")
if pymongo.version < '1.1.1': if pymongo.version < '1.1.1':
raise OperationError('update() method requires PyMongo 1.1.1+') raise OperationError('update() method requires PyMongo 1.1.1+')
@@ -1298,6 +1301,9 @@ class QuerySet(object):
.. versionadded:: 0.2 .. versionadded:: 0.2
""" """
if not update:
raise OperationError("No update parameters, would remove data")
if not write_options: if not write_options:
write_options = {} write_options = {}
update = QuerySet._transform_update(self._document, **update) update = QuerySet._transform_update(self._document, **update)