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

Current class fields when unpickling. Fixes #888

Browse Source 
Optimize dereferencing map by using sets.
This commit is contained in:
Michael Chase
2015-04-08 19:40:43 -07:00
parent a092910fdd
commit a6a7cba121
6 changed files with 50 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
mongoengine/base/document.py
View File

@@ -206,7 +206,12 @@ class BaseDocument(object):
if k in data:
setattr(self, k, data[k])
if '_fields_ordered' in data:
setattr(type(self), '_fields_ordered', data['_fields_ordered'])
if self._dynamic:
setattr(self, '_fields_ordered', data['_fields_ordered'])
else:
_super_fields_ordered = type(self)._fields_ordered
setattr(self, '_fields_ordered', _super_fields_ordered)
dynamic_fields = data.get('_dynamic_fields') or SON()
for k in dynamic_fields.keys():
setattr(self, k, data["_data"].get(k))