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_ord.OrdTests",
             "db_functions.text.test_pad.PadTests",

django_mongodb_backend/functions.py

@@ -25,7 +25,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 (
     MD5,
     SHA256,
@@ -239,6 +249,23 @@ 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,  # Return null for null input.
+            "else": {
+                "$cond": {
+                    "if": {"$eq": [lhs_mql, 0]},
+                    "then": 0,  # Return zero for zero input.
+                    "else": {"$cmp": [lhs_mql, 0]},  # Otherwise, +1 or -1.
+                },
+            },
+        }
+    }
+
+
 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.
@@ -366,6 +393,7 @@ def register_functions():
     Round.as_mql_expr = round_
     RTrim.as_mql_expr = trim("rtrim")
     SHA256.as_mql_expr = hash_func("sha256")
+    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

@@ -16,6 +16,7 @@ New features
   - :class:`~django.db.models.functions.MD5` (requires MongoDB 8.3+)
   - :class:`~django.db.models.functions.Right`
   - :class:`~django.db.models.functions.SHA256` (requires MongoDB 8.3+)
+  - :class:`~django.db.models.functions.Sign`
 
 - Added support for fixed-offset ``tzinfo`` in the
   :class:`~django.db.models.functions.Extract` database functions.

docs/topics/known-issues.rst

@@ -85,7 +85,6 @@ Database functions
     :class:`~django.db.models.functions.SHA224`,
     :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