small fixes for ReferenceField

2012-04-13 04:56:20 -03:00
parent d0e42a4798
commit 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]