Merge pull request #979 from DavidBord/fix-453

fix-#453: Queryset update doesn't go through field validation
This commit is contained in:
David Bordeynik
2015-05-02 20:26:56 +03:00
5 changed files with 38 additions and 13 deletions

View File

@@ -17,6 +17,11 @@ __all__ = ("BaseField", "ComplexBaseField",
"ObjectIdField", "GeoJsonBaseField")
UPDATE_OPERATORS = set(['set', 'unset', 'inc', 'dec', 'pop', 'push',
'push_all', 'pull', 'pull_all', 'add_to_set',
'set_on_insert', 'min', 'max'])
class BaseField(object):
"""A base class for fields in a MongoDB document. Instances of this class
@@ -151,6 +156,8 @@ class BaseField(object):
def prepare_query_value(self, op, value):
"""Prepare a value that is being used in a query for PyMongo.
"""
if op in UPDATE_OPERATORS:
self.validate(value)
return value
def validate(self, value, clean=True):