pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small fixes for GenericReferenceField

Browse Source
This commit is contained in:
Wilson Júnior 2012-05-04 14:58:06 -03:00
parent 233b13d670
commit 3360b72531
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mongoengine/fields.py
View File

@ -734,6 +734,9 @@ class GenericReferenceField(BaseField):
def to_mongo(self, document): def to_mongo(self, document):
if document is None: if document is None:
return None return None
if isinstance(document, (dict, SON)):
return document
id_field_name = document.__class__._meta['id_field'] id_field_name = document.__class__._meta['id_field']
id_field = document.__class__._fields[id_field_name] id_field = document.__class__._fields[id_field_name]