From 3360b72531d4d702b09c205396a356bd78a414da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Wilson=20J=C3=BAnior?= Date: Fri, 4 May 2012 14:58:06 -0300 Subject: [PATCH] Small fixes for GenericReferenceField --- mongoengine/fields.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mongoengine/fields.py b/mongoengine/fields.py index a2250f4c..f7344488 100644 --- a/mongoengine/fields.py +++ b/mongoengine/fields.py @@ -734,6 +734,9 @@ class GenericReferenceField(BaseField): def to_mongo(self, document): if document is None: return None + + if isinstance(document, (dict, SON)): + return document id_field_name = document.__class__._meta['id_field'] id_field = document.__class__._fields[id_field_name]