mongoengine - использовать QuerySet в качестве ReferenceField

Question

На основе этого урока:

from mongoengine import *

connect('tumblelog')

class User(Document):
    email = StringField(required=True, unique=True)
    first_name = StringField(max_length=50)
    last_name = StringField(max_length=50)

class Comment(EmbeddedDocument):
    content = StringField()
    name = StringField(max_length=120)

class Post(Document):
    title = StringField(max_length=120, required=True)
    author = ReferenceField(User, reverse_delete_rule=CASCADE)
    tags = ListField(StringField(max_length=30))
    comments = ListField(EmbeddedDocumentField(Comment))
    meta = {'allow_inheritance': True}

class TextPost(Post):
    content = StringField()

class ImagePost(Post):
    image_path = StringField()

class LinkPost(Post):
    link_url = StringField()


def main():
    john = User(email='[e-mail]', first_name='John', last_name='Doe')
    john.save()
    john_from_db = User.objects(email='[e-mail]')
    post1 = TextPost(title='Fun with MongoEngine', author = john_from_db)
    post1.content = 'Took a look at MongoEngine today, looks pretty cool.'
    post1.tags = ['mongodb','mongoengine']
    post1.save()


if __name__ == '__main__':
    main()

Я получаю эту ошибку на post1.save():

mongoengine.base.ValidationError: ValidationError(A ReferenceField only accepts DBRef or documents: ['author'])

Я попытался добавить опцию dbref=True/False в объявлении ReferenceField, но это ничего не исправило. Объект john имеет тип User, а john_from_db - это QuerySet. Конечно, я что-то здесь упускаю, как вы получаете объект из БД и используете его в другом?

Answer 1

Вы должны вызвать first() для QuerySet, чтобы получить первую запись, которая соответствует запросу, например

john_from_db = User.objects(email='[e-mail]').first()