fixes for #325 issue

2011-10-19 06:30:41 -02:00
parent 76d771d20f
commit b09c52fc7e
3 changed files with 13 additions and 1 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)