Apply defaults to fields with None value at 'set' time.

If a field has a default, and you explicitly set it to None, the
behaviour before this patch was very confusing:

    class Person(Document):
        created = DateTimeField(default=datetime.datetime.utcnow)

    >>> p = Person(created=None)
    >>> p.created
    datetime.datetime(2013, 5, 30, 0, 18, 20, 242628)
    >>> p.created
    datetime.datetime(2013, 5, 30, 0, 18, 20, 995248)
    >>> p.created
    datetime.datetime(2013, 5, 30, 0, 18, 21, 370578)

It would be stored as None, and then at 'get' time, the default would be
applied. As you can see, if the default is a generator, this leads to some
crazy behaviour.

There's an argument that if I asked it to be set to None, why not respect that?
But I don't think that's how the rest of mongoengine seems to work (for
example, setting a field to None seems to mean it doesn't even get set in mongo
- as opposed to being set but with a 'null' value). Besides, as the code shows
above, you'd expect p.created to return None. So clearly, mongoengine is
already expecting None to mean 'default' where a default is available.

This bug also interacts nastily with required=True - if you're forcibly setting
the field to None, then at validation time, the None will fail validation
despite a perfectly valid default being available.

With this patch, when the field is set, the default is immediately applied.
This means any generation happens once, the getter always returns the same
value, and 'required' validation always respects the default.

Note: this breakage seems to be new since mongoengine 0.8.
This commit is contained in:
Nigel McNie 2013-05-30 16:37:40 +12:00
parent c5ce96c391
commit 4c9e90732e
2 changed files with 19 additions and 0 deletions

View File

@ -82,6 +82,12 @@ class BaseField(object):
def __set__(self, instance, value): def __set__(self, instance, value):
"""Descriptor for assigning a value to a field in a document. """Descriptor for assigning a value to a field in a document.
""" """
if value is None:
value = self.default
# Allow callable default values
if callable(value):
value = value()
if instance._initialised: if instance._initialised:
try: try:
if (self.name not in instance._data or if (self.name not in instance._data or

View File

@ -44,6 +44,19 @@ class FieldTest(unittest.TestCase):
self.assertEqual(person._fields['age'].help_text, "Your real age") self.assertEqual(person._fields['age'].help_text, "Your real age")
self.assertEqual(person._fields['userid'].verbose_name, "User Identity") self.assertEqual(person._fields['userid'].verbose_name, "User Identity")
class Person2(Document):
created = DateTimeField(default=datetime.datetime.utcnow)
person = Person2()
date1 = person.created
date2 = person.created
self.assertEqual(date1, date2)
person = Person2(created=None)
date1 = person.created
date2 = person.created
self.assertEqual(date1, date2)
def test_required_values(self): def test_required_values(self):
"""Ensure that required field constraints are enforced. """Ensure that required field constraints are enforced.
""" """