Merge branch 'dev' of github.com:hmarr/mongoengine into dev

2012-04-18 10:28:06 +01:00
parent d9d2291837 6d503119a1
commit 360624eb6e
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]