Add ability to filter the generic reference field by ObjectId and DBRef (#1425)

2016-12-09 12:56:06 -05:00
parent 25e0f12976
commit 1b9432824b
4 changed files with 51 additions and 4 deletions
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@@ -1249,7 +1249,7 @@ class GenericReferenceField(BaseField):
        if document is None:
            return None

-        if isinstance(document, (dict, SON)):
+        if isinstance(document, (dict, SON, ObjectId, DBRef)):
            return document

        id_field_name = document.__class__._meta['id_field']