Merge pull request #6 from cloudblue/fix/LITE-25721

LITE-25721 Fixed null comparison filtering (Django __isnull operator)

Author: maxipavlovic

dj_mongoengine_rql/filter_cls.py

@@ -2,6 +2,7 @@
 #  Copyright © 2022 CloudBlue LLC. All rights reserved.
 #
 
+from dj_rql.constants import DjangoLookups
 from dj_rql.filter_cls import RQLFilterClass
 from django_mongoengine.fields.djangoflavor import DjangoField
 from py_rql.constants import FilterLookups
@@ -39,6 +40,10 @@ def _get_decimal_field_precision(cls, field):
         return field.precision
 
     def _build_django_q(self, filter_item, django_lookup, filter_lookup, typed_value):
+        if django_lookup == DjangoLookups.NULL:
+            q = self.Q_CLS(**{filter_item['orm_route']: None})
+            return ~q if filter_lookup == FilterLookups.NE else q
+
         if filter_lookup != FilterLookups.NE:
             kwargs = {'{0}__{1}'.format(filter_item['orm_route'], django_lookup): typed_value}
         else:

tests/test_filter_cls.py

@@ -61,6 +61,18 @@ def test_not():
     assert qs.query.q == {'$nor': [{'other_int_f': 120}], 'other_int_f': 1}
 
 
+def test_null():
+    _, qs = DocFilterClass(Doc.objects).apply_filters('flt=null()')
+
+    assert qs.query.q == {'flt': None}
+
+
+def test_not_null():
+    _, qs = DocFilterClass(Doc.objects).apply_filters('flt=ne=null()')
+
+    assert qs.query.q == {'$nor': [{'flt': None}]}
+
+
 def test_db_operation(is_real_mongo):
     if is_real_mongo:
         doc = Doc.objects.create(str_f='a')