zkurtz
zkurtz

Reputation: 3270

write an empty dict field using mongoengine

I have observed mongo documents for which document[field] returns {}, but I can't seem to create such a document. Example:

import mongoengine

mongoengine.connect('FOO', host='localhost', port=27017)

class Foo(mongoengine.Document):
    some_dict = mongoengine.DictField()
    message = mongoengine.StringField()

ID = '59b97ec7c5d65e0c4740b886'
foo = Foo()
foo.some_dict = {}
foo.id = ID
foo.save()

But when I query the record, some_dict is not among the fields, so the last line throws an error:

import pymongo
CON = pymongo.MongoClient('localhost',27017)
x = CON.FOO.foo.find_one()
assert str(x['_id']) == ID
assert 'some_dict' in x.keys()

Upvotes: 3

Views: 2047

Answers (1)

zkurtz
zkurtz

Reputation: 3270

One obvious workaround is to update the document after it's created, but it feels like too much of a hack. I would prefer if mongoengine made it easy to do directly:

import mongoengine
import pymongo
CON = pymongo.MongoClient('localhost',27017)
CON.Foo.foo.remove()

mongoengine.connect('FOO', host='localhost', port=27017)

class Foo(mongoengine.Document):
    some_dict = mongoengine.DictField()
    some_list = mongoengine.ListField()

ID = '59b97ec7c5d65e0c4740b886'
foo = Foo()
foo.id = ID
foo.save()

from bson.objectid import ObjectId
CON.FOO.foo.update_one({'_id': ObjectId(ID)},
                       {'$set': {'some_dict': {}}}, upsert=False)

x = CON.FOO.foo.find_one()
assert str(x['_id']) == ID
assert 'some_dict' in x.keys()

Upvotes: 1

Related Questions