Complete JSONObject (#2560)

Author: adamchainz

django-stubs/db/models/functions/comparison.pyi

@@ -1,9 +1,11 @@
 from typing import Any, ClassVar
 
+from django.db.backends.base.base import BaseDatabaseWrapper
 from django.db.models import Func
 from django.db.models.expressions import Combinable
 from django.db.models.fields import Field
 from django.db.models.fields.json import JSONField
+from django.db.models.sql.compiler import SQLCompiler, _AsSqlType
 
 class Cast(Func):
     def __init__(self, expression: Combinable | str, output_field: str | Field) -> None: ...
@@ -16,8 +18,16 @@ class Collate(Func):
 class Greatest(Func): ...
 
 class JSONObject(Func):
-    def __init__(self, **fields: Any) -> None: ...
     output_field: ClassVar[JSONField]
+    def __init__(self, **fields: Any) -> None: ...
+    def as_sql(self, compiler: SQLCompiler, connection: BaseDatabaseWrapper, **extra_context: Any) -> _AsSqlType: ...  # type: ignore [override]
+    def as_native(
+        self, compiler: SQLCompiler, connection: BaseDatabaseWrapper, *, returning: str, **extra_context: Any
+    ) -> _AsSqlType: ...
+    def as_postgresql(
+        self, compiler: SQLCompiler, connection: BaseDatabaseWrapper, **extra_context: Any
+    ) -> _AsSqlType: ...
+    def as_oracle(self, compiler: SQLCompiler, connection: BaseDatabaseWrapper, **extra_context: Any) -> _AsSqlType: ...
 
 class Least(Func): ...
 class NullIf(Func): ...

scripts/stubtest/allowlist_todo.txt

@@ -949,9 +949,6 @@ django.db.models.functions.Cot.as_oracle
 django.db.models.functions.Degrees.as_oracle
 django.db.models.functions.Extract.as_sql
 django.db.models.functions.Extract.lookup_name
-django.db.models.functions.JSONObject.as_oracle
-django.db.models.functions.JSONObject.as_postgresql
-django.db.models.functions.JSONObject.as_sql
 django.db.models.functions.Now.as_mysql
 django.db.models.functions.Now.as_postgresql
 django.db.models.functions.NullIf.as_oracle
@@ -967,9 +964,6 @@ django.db.models.functions.comparison.Coalesce.as_oracle
 django.db.models.functions.comparison.Coalesce.empty_result_set_value
 django.db.models.functions.comparison.Collate.as_sql
 django.db.models.functions.comparison.Collate.collation_re
-django.db.models.functions.comparison.JSONObject.as_oracle
-django.db.models.functions.comparison.JSONObject.as_postgresql
-django.db.models.functions.comparison.JSONObject.as_sql
 django.db.models.functions.comparison.NullIf.as_oracle
 django.db.models.functions.datetime.Extract.as_sql
 django.db.models.functions.datetime.Extract.lookup_name

scripts/stubtest/allowlist_todo_django51.txt

@@ -80,8 +80,6 @@ django.db.models.fields.related.RelatedField.cache_name
 django.db.models.fields.reverse_related.ForeignObjectRel.accessor_name
 django.db.models.fields.reverse_related.ForeignObjectRel.cache_name
 django.db.models.functions.ConcatPair.pipes_concat_sql
-django.db.models.functions.JSONObject.as_native
-django.db.models.functions.comparison.JSONObject.as_native
 django.db.models.functions.text.ConcatPair.pipes_concat_sql
 django.db.models.lookups.In.get_refs
 django.forms.ModelChoiceField.validate_no_null_characters