Merge branch 'patch-8' of https://github.com/wpjunior/mongoengine

2012-05-09 11:21:54 +01:00 · 2012-05-09 11:21:54 +01:00 · 43a989978a
commit 43a989978a
parent 27734a7c26 3360b72531
1 changed files with 3 additions and 0 deletions
--- 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]