From 56cd73823e7ed4b216bb740c612c21eea59fd1a7 Mon Sep 17 00:00:00 2001 From: Ross Lawley Date: Mon, 20 May 2013 10:09:16 +0000 Subject: [PATCH] Add backwards compat for pickle --- mongoengine/base/document.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mongoengine/base/document.py b/mongoengine/base/document.py index e3202b9d..018adbf3 100644 --- a/mongoengine/base/document.py +++ b/mongoengine/base/document.py @@ -148,9 +148,10 @@ class BaseDocument(object): return 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]) - self._data = self.__class__._from_son(data["_data"])._data def __iter__(self): if 'id' in self._fields and 'id' not in self._fields_ordered: