Add search text lookup.

Author: WaVEV

django_mongodb_backend/compiler.py

@@ -128,6 +128,9 @@ def _prepare_search_query_for_aggregation_pipeline(self, order_by):
         self._prepare_search_expressions_for_pipeline(
             self.having, annotation_group_idx, replacements
         )
+        self._prepare_search_expressions_for_pipeline(
+            self.get_where(), annotation_group_idx, replacements
+        )
         return replacements
 
     def _prepare_annotations_for_aggregation_pipeline(self, order_by):
@@ -291,6 +294,8 @@ def pre_sql_setup(self, with_col_aliases=False):
             for target, expr in self.query.annotation_select.items()
         }
         self.order_by_objs = [expr.replace_expressions(all_replacements) for expr, _ in order_by]
+        where_ = self.get_where().replace_expressions(all_replacements)
+        self.set_where(where_)
         return extra_select, order_by, group_by
 
     def execute_sql(
@@ -692,6 +697,9 @@ def _get_ordering(self):
     def get_where(self):
         return getattr(self, "where", self.query.where)
 
+    def set_where(self, value):
+        self.where = value
+
     def explain_query(self):
         # Validate format (none supported) and options.
         options = self.connection.ops.explain_query_prefix(
@@ -778,6 +786,9 @@ def check_query(self):
     def get_where(self):
         return self.query.where
 
+    def set_where(self, value):
+        self.query.where = value
+
     @cached_property
     def collection_name(self):
         return self.query.base_table
@@ -849,6 +860,9 @@ def check_query(self):
     def get_where(self):
         return self.query.where
 
+    def set_where(self, value):
+        self.query.where = value
+
     @cached_property
     def collection_name(self):
         return self.query.base_table

django_mongodb_backend/expressions/search.py

@@ -1,6 +1,9 @@
 from django.db import NotSupportedError
-from django.db.models import Expression, FloatField
+from django.db.models import CharField, Expression, FloatField, TextField
 from django.db.models.expressions import F, Value
+from django.db.models.lookups import Lookup
+
+from ..query_utils import process_lhs, process_rhs
 
 
 def cast_as_value(value):
@@ -1009,3 +1012,27 @@ def __init__(self, definitions=None):
 
     def as_mql(self, compiler, connection):
         return self.definitions
+
+
+class SearchTextLookup(Lookup):
+    lookup_name = "search"
+
+    def __init__(self, lhs, rhs):
+        super().__init__(lhs, rhs)
+        self.lhs = SearchText(self.lhs, self.rhs)
+        self.rhs = Value(0)
+
+    def __str__(self):
+        return f"SearchText({self.lhs}, {self.rhs})"
+
+    def __repr__(self):
+        return f"SearchText({self.lhs}, {self.rhs})"
+
+    def as_mql(self, compiler, connection):
+        lhs_mql = process_lhs(self, compiler, connection)
+        value = process_rhs(self, compiler, connection)
+        return {"$gte": [lhs_mql, value]}
+
+
+CharField.register_lookup(SearchTextLookup)
+TextField.register_lookup(SearchTextLookup)

tests/queries_/test_search.py

@@ -298,6 +298,10 @@ def test_search_text(self):
         qs = Article.objects.annotate(score=SearchText(path="body", query="lazy"))
         self.wait_for_assertion(lambda: self.assertCountEqual([self.article], qs))
 
+    def test_search_lookup(self):
+        qs = Article.objects.filter(body__search="lazy")
+        self.wait_for_assertion(lambda: self.assertCountEqual([self.article], qs))
+
     def test_search_text_with_fuzzy_and_criteria(self):
         qs = Article.objects.annotate(
             score=SearchText(