How to add additional fields to embedded documents without changing the original model

Question

I want to add extra properties to a document before embedding that into other document, but I don't know how to do that.

Here's my code and what I have tried so far:

from mongoengine import *
from datetime import datetime


class User(Document):
    name = StringField(max_length=80, required=True)
    created_at = DateTimeField(default=datetime.now(), required=True)
    updated_at = DateTimeField(default=datetime.now(), required=True)

    meta = {
        'collection': 'users'
    }

    def save(self, *args, **kwargs):
        self.updated_at = datetime.now()
        return super(User, self).save(*args, **kwargs)



class Stream(Document):
    users = EmbeddedDocumentListField(document_type='User')
    created_at = DateTimeField(default=datetime.now(), required=True)
    updated_at = DateTimeField(default=datetime.now(), required=True)

    meta = {
        'collection': 'streams'
    }

    def save(self, *args, **kwargs):
        self.updated_at = datetime.now()
        return super(Stream, self).save(*args, **kwargs)

When I embed the user document to the stream's users EmbeddedDocumentListField, It will be added and look like this

{
    "_id" : ObjectId("57e6123fe8c39b18b1a13431"),
    "users" : [
        {
            "_id" : ObjectId("57e6123fe8c39b18b1a13432"),
            "name": "Rohit Khatri",
            "created_at" : ISODate("2016-09-24T11:12:23.182Z"),
            "updated_at" : ISODate("2016-09-24T11:12:23.301Z")
        }
    ],
    "created_at" : ISODate("2016-09-24T11:12:23.189Z"),
    "updated_at" : ISODate("2016-09-24T11:12:23.323Z")
}

Now I want to embed the User document with additional properties, like roles. Here's what I have tried:-

user = User.objects.create(name='Rohit Khatri')
user.roles = ['admin','writer']

stream = Stream.objects.create()
stream.users.append(user)
stream.save()

But It doesn't add the roles field, It would be thankful if somebody can help me with this.

Thanks

score 1 · Accepted Answer · answered Sep 27 '16 at 06:47

1

use update method to set the new attribute

user.update(set__roles = ['admin','writer'])

answered Sep 27 '16 at 06:47

armak

560
5
13

It will change the user document, and I don't want to make any changes in the original document, but into the embedded user document. – Rohit Khatri Sep 27 '16 at 06:51
@RohitKhatri no, just find the user who you want to change and then run the update. – armak Sep 27 '16 at 06:53
It throws error `mongoengine.errors.InvalidQueryError: Cannot resolve field "roles"` – Rohit Khatri Sep 27 '16 at 06:54

How to add additional fields to embedded documents without changing the original model

1 Answers1