Merge pull request #482 from wpjunior/patch-7

Small fixes for ReferenceField
2012-04-16 05:25:42 -07:00 · 2012-04-16 05:25:42 -07:00 · 6d503119a1
commit 6d503119a1
parent d0e42a4798 bfae93e57e
1 changed files with 3 additions and 0 deletions
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@ -657,6 +657,9 @@ class ReferenceField(BaseField):
        return super(ReferenceField, self).__get__(instance, owner)

    def to_mongo(self, document):
+        if isinstance(document, DBRef):
+            return document
+        
        id_field_name = self.document_type._meta['id_field']
        id_field = self.document_type._fields[id_field_name]