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

Fixes tree based circular references

Browse Source 
Thanks to jpfarias for the fix.
Also normalised the other circular checks.
This commit is contained in:
Ross Lawley
2011-10-10 09:16:32 -07:00
parent 8797565606
commit 7b1860d17b
4 changed files with 71 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mongoengine/document.py
View File

@@ -185,18 +185,18 @@ class Document(BaseDocument):
id_field = self._meta['id_field']
self[id_field] = self._fields[id_field].to_python(object_id)
def reset_changed_fields(doc, inspected_docs=None):
def reset_changed_fields(doc, inspected=None):
"""Loop through and reset changed fields lists"""
inspected_docs = inspected_docs or []
inspected_docs.append(doc)
inspected = inspected or []
inspected.append(doc)
if hasattr(doc, '_changed_fields'):
doc._changed_fields = []
for field_name in doc._fields:
field = getattr(doc, field_name)
if field not in inspected_docs and hasattr(field, '_changed_fields'):
reset_changed_fields(field, inspected_docs)
if field not in inspected and hasattr(field, '_changed_fields'):
reset_changed_fields(field, inspected)
reset_changed_fields(self)
self._changed_fields = []