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

Add backwards compat for pickle

Browse Source
This commit is contained in:
Ross Lawley 2013-05-20 10:09:16 +00:00
parent 6299015039
commit 56cd73823e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mongoengine/base/document.py
View File

@ -148,9 +148,10 @@ class BaseDocument(object):
return data return data
def __setstate__(self, data): def __setstate__(self, data):
for k in ('_changed_fields', '_initialised', '_created'): if isinstance(data["_data"], SON):
data["_data"] = self.__class__._from_son(data["_data"])._data
for k in ('_changed_fields', '_initialised', '_created', '_data'):
setattr(self, k, data[k]) setattr(self, k, data[k])
self._data = self.__class__._from_son(data["_data"])._data
def __iter__(self): def __iter__(self):
if 'id' in self._fields and 'id' not in self._fields_ordered: if 'id' in self._fields and 'id' not in self._fields_ordered: