Merge pull request #326 from wpjunior/fixes-325

Fixes for #325 issue - Thanks @wpjunior
2011-10-27 00:33:36 -07:00
parent 634e1f661f 8e87648d53
commit c42aef74de
2 changed files with 20 additions and 0 deletions
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@@ -666,6 +666,9 @@ class ReferenceField(BaseField):
        return pymongo.dbref.DBRef(collection, id_)

    def prepare_query_value(self, op, value):
+        if value is None:
+            return None
+
        return self.to_mongo(value)

    def validate(self, value):
@@ -743,6 +746,9 @@ class GenericReferenceField(BaseField):
        return {'_cls': document._class_name, '_ref': ref}

    def prepare_query_value(self, op, value):
+        if value is None:
+            return None
+
        return self.to_mongo(value)