Fixed cascade save edge case

refs MongoEngine/mongoengine#40
2012-07-19 11:52:26 +01:00 · 2012-07-19 11:52:26 +01:00 · ae39ed94c9
commit ae39ed94c9
parent 1e51180d42
3 changed files with 30 additions and 0 deletions
--- a/mongoengine/document.py
+++ b/mongoengine/document.py
@ -248,11 +248,16 @@ class Document(BaseDocument):
        _refs = kwargs.get('_refs', []) or []

        for name, cls in self._fields.items():
+
            if not isinstance(cls, (ReferenceField, GenericReferenceField)):
                continue
+
            ref = getattr(self, name)
            if not ref:
                continue
+            if isinstance(ref, DBRef):
+                continue
+
            ref_id = "%s,%s" % (ref.__class__.__name__, str(ref._data))
            if ref and ref_id not in _refs:
                _refs.append(ref_id)