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

Introduced new create_default field argument. If set to true, mongoengine will automagically create an instance of the desired document class (useful if using EmbeddedDocumentField for example):

Browse Source 
class SubDoc(EmbeddedDocument):
        url = URLField()

    class MyDoc(Document):
        subdoc = EmbeddedDocumentField(SubDoc, create_default=True)

With create_default MyDoc().subdoc is automatically instantiated. Hint: default=SubDoc() WON'T work (that's why I've introduced create_default)
This commit is contained in:
Florian Schlachter
2010-04-17 01:23:14 +02:00
parent ef172712da
commit da57572409
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mongoengine/base.py
View File

@@ -24,7 +24,8 @@ class BaseField(object):
_index_with_types = True _index_with_types = True
def __init__(self, db_field=None, name=None, required=False, default=None, def __init__(self, db_field=None, name=None, required=False, default=None,
unique=False, unique_with=None, primary_key=False, validation=None): unique=False, unique_with=None, primary_key=False, validation=None,
create_default=False):
self.db_field = (db_field or name) if not primary_key else '_id' self.db_field = (db_field or name) if not primary_key else '_id'
if name: if name:
import warnings import warnings
@@ -37,6 +38,7 @@ class BaseField(object):
self.unique_with = unique_with self.unique_with = unique_with
self.primary_key = primary_key self.primary_key = primary_key
self.validation = validation self.validation = validation
self.create_default = create_default
def __get__(self, instance, owner): def __get__(self, instance, owner):
"""Descriptor for retrieving a value from a field in a document. Do """Descriptor for retrieving a value from a field in a document. Do
@@ -53,6 +55,10 @@ class BaseField(object):
# Allow callable default values # Allow callable default values
if callable(value): if callable(value):
value = value() value = value()
# Check whether we should auto-create a default document
if self.create_default and hasattr(self, 'document'):
value = self.document()
return value return value
def __set__(self, instance, value): def __set__(self, instance, value):