class Doc(Document):
emb = EmbDocField('test_fields.EmbDoc')
ref = RefField('test_fields.RefDoc')
lst_emb = ListField(EmbDocField('test_fields.EmbDoc'))
lst_ref = ListField(RefField('test_fields.RefDoc'))
lst_int = ListField(IntField())
wrong_emb = EmbDocField('xxx')
wrong_ref = RefField('xxx')
wrong_lst_emb = ListField(EmbDocField('xxx'))
wrong_emb_doc = EmbDocField('test_fields.RefDoc')
wrong_ref_doc = RefField('test_fields.EmbDoc')
class Comment(ValidatingDocument):
_id = ObjectIdField(required=True, default=lambda: ObjectId())
user = RefField(User, required=True)
post = RefField(User, required=True)
in_response_to = ObjectIdField()
content = StrField(allow_blank=False, required=True)
created = DateTimeField(required=True)
updated = DateTimeField(required=True)
class Meta:
collection = 'comments'
async def pre_save(self):
timestamp = datetime.utcnow()
if not self.created:
self.created = timestamp
self.updated = timestamp
class User(Document):
_id = StrField(regex=r'[a-zA-Z0-9_]{3, 20}')
is_active = BoolField(default=True)
posts = ListField(RefField('models.Post'), default=lambda: list())
quote = StrField(required=False)
# create a synonym field
name = SynonymField(_id)
class Meta:
collection = 'users'
class User(Document):
"""User model."""
_id = StrField(required=True, allow_blank=False, max_length=10)
active = BoolField(required=True, default=True)
posts = ListField(RefField('models.Post'),
required=True,
default=lambda: list())
data = AnyField(required=False)
# synonyms
name = SynonymField(_id)
class Post(Document):
# _id field will be added automatically as
# _id = ObjectIdField(defalut=lambda: ObjectId())
title = StrField(allow_blank=False, max_length=50)
body = StrField()
created = DateTimeField(default=lambda: datetime.utcnow())
views = IntField(default=0)
rate = FloatField(default=0.0)
author = RefField(User, mongo_name='user')
comments = ListField(EmbDocField('models.Comment'), default=lambda: list())
class Meta:
collection = 'posts'
indexes = [IndexModel([('created', DESCENDING)])]
default_sort = [('created', DESCENDING)]
class Post(ValidatingDocument):
_id = ObjectIdField(required=True, default=lambda: ObjectId())
user = RefField(User, required=True)
title = StrField(allow_blank=False, required=True)
body = StrField(allow_blank=False, required=True)
created = DateTimeField(required=True)
updated = DateTimeField(required=True)
class Meta:
collection = 'posts'
async def pre_save(self):
timestamp = datetime.utcnow()
if not self.created:
self.created = timestamp
self.updated = timestamp
class Permission(ValidatingDocument):
_id = ObjectIdField(required=True, default=lambda: ObjectId())
user = RefField(User, required=True)
roles = ListField(StrField(allow_blank=False, required=True))
created = DateTimeField(required=True)
updated = DateTimeField(required=True)
class Meta:
collection = 'permissions'
indexes = [
IndexModel([('user', ASCENDING)], unique=True)
]
async def pre_save(self):
timestamp = datetime.utcnow()
if not self.created:
self.created = timestamp
self.updated = timestamp
class Post(Document):
"""Post model."""
title = StrField(required=True, allow_blank=False)
views = IntField(required=True, default=0)
created = DateTimeField(required=True, default=lambda: datetime.utcnow())
rate = FloatField(required=True, default=0.0)
author = RefField(User, required=True)
comments = ListField(EmbDocField('models.Comment'),
required=True,
default=lambda: list())
class Meta:
"""Post meta."""
collection = 'posts'
indexes = [
IndexModel([('title', ASCENDING)], unique=True,
name='title_index'),
IndexModel([('author', ASCENDING), ('created', DESCENDING)],
name='author_created_index')
]
class Doc(Document):
str_field = StrField(regex=r'[abc]+')
int_field = IntField(gt=0, lt=10)
emb_field = EmbDocField(EmbDoc)
lst_field = ListField(RefField(RefDoc), min_length=1, max_length=1)
class Comment(EmbeddedDocument):
"""Comment model."""
_id = ObjectIdField(required=True, default=lambda: ObjectId())
body = StrField(required=True, allow_blank=False)
author = RefField(User, required=True, mongo_name='user')
class Comment(EmbeddedDocument):
_id = ObjectIdField(default=lambda: ObjectId())
author = RefField(User)
body = StrField()
class WrongRefDoc(Document):
wrong_ref = RefField(EmbDoc)
(ObjectIdField, ObjectId('58ce6d537e592254b67a503d')),
(EmailField, '*****@*****.**'),
(DecimalField, Decimal('0.005')),
]
@pytest.mark.parametrize('field, expected', [
(StrField(required=False), None),
(IntField(required=False), None),
(FloatField(required=False), None),
(BoolField(required=False), None),
(DateTimeField(required=False), None),
(ObjectIdField(required=False), None),
(EmbDocField(EmbDoc, required=False), None),
(ListField(EmbDocField(EmbDoc), required=False), None),
(RefField(RefDoc, required=False), None),
(EmailField(required=False), None),
])
def test_field_not_exist_get_value(field, expected):
class Doc(Document):
value = field
assert Doc().value is expected
@pytest.mark.parametrize('field, default', FIELD_DEFAULT)
def test_field_attributes(field, default):
class Doc(Document):
value = field(required=False)
assert isinstance(Doc.value, field)