More clean ups.

Author: WaVEV

django_mongodb_backend/fields/json.py

@@ -54,10 +54,7 @@ def data_contains(self, compiler, connection, as_path=False):  # noqa: ARG001
 def _has_key_predicate(path, root_column=None, negated=False, as_path=False):
     """Return MQL to check for the existence of `path`."""
     if as_path:
-        # if not negated:
         return {path: {"$exists": not negated}}
-        # return {"$and": [{path: {"$exists": True}}, {path: {"$ne": None}}]}
-        # return {"$or": [{path: {"$exists": False}}, {path: None}]}
     result = {
         "$and": [
             # The path must exist (i.e. not be "missing").

django_mongodb_backend/functions.py

@@ -67,7 +67,7 @@
 
 def cast(self, compiler, connection):
     output_type = connection.data_types[self.output_field.get_internal_type()]
-    lhs_mql = process_lhs(self, compiler, connection, as_path=False)[0]
+    lhs_mql = process_lhs(self, compiler, connection)[0]
     if max_length := self.output_field.max_length:
         lhs_mql = {"$substrCP": [lhs_mql, 0, max_length]}
     # Skip the conversion for "object" as it doesn't need to be transformed for
@@ -81,7 +81,7 @@ def cast(self, compiler, connection):
 
 
 def concat(self, compiler, connection):
-    return self.get_source_expressions()[0].as_mql(compiler, connection, as_path=False)
+    return self.get_source_expressions()[0].as_mql(compiler, connection)
 
 
 def concat_pair(self, compiler, connection):
@@ -91,12 +91,12 @@ def concat_pair(self, compiler, connection):
 
 
 def cot(self, compiler, connection):
-    lhs_mql = process_lhs(self, compiler, connection, as_path=False)
+    lhs_mql = process_lhs(self, compiler, connection)
     return {"$divide": [1, {"$tan": lhs_mql}]}
 
 
-def extract(self, compiler, connection, as_path=False):
-    lhs_mql = process_lhs(self, compiler, connection, as_path=as_path)
+def extract(self, compiler, connection):
+    lhs_mql = process_lhs(self, compiler, connection)
     operator = EXTRACT_OPERATORS.get(self.lookup_name)
     if operator is None:
         raise NotSupportedError(f"{self.__class__.__name__} is not supported.")
@@ -105,22 +105,8 @@ def extract(self, compiler, connection, as_path=False):
     return {f"${operator}": lhs_mql}
 
 
-def func(self, compiler, connection, as_path=False):
-    lhs_mql = process_lhs(self, compiler, connection, as_path=False)
-    if self.function is None:
-        raise NotSupportedError(f"{self} may need an as_mql() method.")
-    operator = MONGO_OPERATORS.get(self.__class__, self.function.lower())
-    if as_path:
-        return {"$expr": {f"${operator}": lhs_mql}}
-    return {f"${operator}": lhs_mql}
-
-
-def func_path(self, compiler, connection):  # noqa: ARG001
-    raise NotSupportedError(f"{self} may need an as_mql_path() method.")
-
-
-def func_expr(self, compiler, connection):
-    lhs_mql = process_lhs(self, compiler, connection, as_path=False)
+def func(self, compiler, connection):
+    lhs_mql = process_lhs(self, compiler, connection)
     if self.function is None:
         raise NotSupportedError(f"{self} may need an as_mql() method.")
     operator = MONGO_OPERATORS.get(self.__class__, self.function.lower())
@@ -291,8 +277,8 @@ def register_functions():
     ConcatPair.as_mql_expr = concat_pair
     Cot.as_mql_expr = cot
     Extract.as_mql_expr = extract
-    Func.as_mql_path = func_path
-    Func.as_mql_expr = func_expr
+    Func.as_mql_expr = func
+    Func.can_use_path = False
     JSONArray.as_mql_expr = process_lhs
     Left.as_mql_expr = left
     Length.as_mql_expr = length
@@ -312,4 +298,3 @@ def register_functions():
     TruncDate.as_mql_expr = trunc_date
     TruncTime.as_mql_expr = trunc_time
     Upper.as_mql_expr = preserve_null("toUpper")
-    Func.can_use_path = False