From bfae93e57ef725ec1ac9db1dc483a90f7c780201 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Wilson=20J=C3=BAnior?= Date: Fri, 13 Apr 2012 04:56:20 -0300 Subject: [PATCH] small fixes for ReferenceField --- mongoengine/fields.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mongoengine/fields.py b/mongoengine/fields.py index 13b7ed8e..831186d0 100644 --- 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]