INTPYTHON-972 Add support for the Sign database function

Author: timgraham

django_mongodb_backend/features.py

@@ -258,7 +258,6 @@ def django_test_expected_failures(self):
             "db_functions.tests.FunctionTests.test_func_transform_bilateral_multivalue",
         },
         "MongoDB does not support this database function.": {
-            "db_functions.math.test_sign.SignTests",
             "db_functions.text.test_chr.ChrTests",
             "db_functions.text.test_md5.MD5Tests",
             "db_functions.text.test_ord.OrdTests",

django_mongodb_backend/functions.py

@@ -24,7 +24,17 @@
     TruncDate,
     TruncTime,
 )
-from django.db.models.functions.math import Ceil, Cot, Degrees, Log, Power, Radians, Random, Round
+from django.db.models.functions.math import (
+    Ceil,
+    Cot,
+    Degrees,
+    Log,
+    Power,
+    Radians,
+    Random,
+    Round,
+    Sign,
+)
 from django.db.models.functions.text import (
     Concat,
     ConcatPair,
@@ -175,6 +185,19 @@ def round_(self, compiler, connection):
     }
 
 
+def sign(self, compiler, connection):
+    lhs_mql = process_lhs(self, compiler, connection, as_expr=True)
+    return {
+        "$cond": {
+            "if": {"$eq": [lhs_mql, None]},
+            "then": None,
+            "else": {
+                "$cond": {"if": {"$eq": [lhs_mql, 0]}, "then": 0, "else": {"$cmp": [lhs_mql, 0]}}
+            },
+        }
+    }
+
+
 def str_index(self, compiler, connection):
     lhs = process_lhs(self, compiler, connection, as_expr=True)
     # StrIndex should be 0-indexed (not found) but it's -1-indexed on MongoDB.
@@ -298,6 +321,7 @@ def register_functions():
     Replace.as_mql_expr = replace
     Round.as_mql_expr = round_
     RTrim.as_mql_expr = trim("rtrim")
+    Sign.as_mql_expr = sign
     StrIndex.as_mql_expr = str_index
     Substr.as_mql_expr = substr
     Trim.as_mql_expr = trim("trim")

docs/releases/6.0.x.rst

@@ -10,6 +10,9 @@ Django MongoDB Backend 6.0.x
 New features
 ------------
 
+- Added support for the :class:`~django.db.models.functions.Sign` database
+  function.
+
 - Added support for Django's warning when performing import-time queries.
 
 Bug fixes

docs/topics/known-issues.rst

@@ -89,7 +89,6 @@ Database functions
     :class:`~django.db.models.functions.SHA256`,
     :class:`~django.db.models.functions.SHA384`,
     :class:`~django.db.models.functions.SHA512`
-  - :class:`~django.db.models.functions.Sign`
 
 - The ``tzinfo`` parameter of the
   :class:`~django.db.models.functions.TruncDate` and