from typing import Any, Dict, List, Optional, Tuple, Type, Union

from psycopg2.extras import Json

from django.core.serializers.json import DjangoJSONEncoder
from django.db.models import Field, Transform
from django.db.models import lookups as builtin_lookups

from .mixins import CheckFieldDefaultMixin


class JsonAdapter(Json):
    encoder: Any = ...
    def __init__(
        self,
        adapted: Any,
        dumps: Optional[Any] = ...,
        encoder: Optional[Any] = ...,
    ) -> None: ...
    def dumps(self, obj: Any): ...

class JSONField(CheckFieldDefaultMixin, Field):
    empty_strings_allowed: bool = ...
    description: Any = ...
    default_error_messages: Any = ...
    encoder: Any = ...
    def __init__(
        self,
        verbose_name: None = ...,
        name: None = ...,
        encoder: Optional[Type[DjangoJSONEncoder]] = ...,
        **kwargs: Any
    ) -> None: ...
    def db_type(self, connection: Any): ...
    def deconstruct(
        self
    ) -> Tuple[
        None, str, List[Any], Dict[str, Union[Type[DjangoJSONEncoder], bool]]
    ]: ...
    def get_transform(self, name: Any): ...
    def get_prep_value(self, value: Any): ...
    def validate(self, value: Any, model_instance: Any) -> None: ...
    def value_to_string(self, obj: Any): ...
    def formfield(self, **kwargs: Any): ...

class KeyTransform(Transform):
    operator: str = ...
    nested_operator: str = ...
    key_name: Any = ...
    def __init__(self, key_name: Any, *args: Any, **kwargs: Any) -> None: ...
    def as_sql(self, compiler: Any, connection: Any): ...

class KeyTextTransform(KeyTransform):
    operator: str = ...
    nested_operator: str = ...
    output_field: Any = ...

class KeyTransformTextLookupMixin:
    def __init__(
        self, key_transform: Any, *args: Any, **kwargs: Any
    ) -> None: ...

class KeyTransformIExact(
    KeyTransformTextLookupMixin, builtin_lookups.IExact
): ...
class KeyTransformIContains(
    KeyTransformTextLookupMixin, builtin_lookups.IContains
): ...
class KeyTransformStartsWith(
    KeyTransformTextLookupMixin, builtin_lookups.StartsWith
): ...
class KeyTransformIStartsWith(
    KeyTransformTextLookupMixin, builtin_lookups.IStartsWith
): ...
class KeyTransformEndsWith(
    KeyTransformTextLookupMixin, builtin_lookups.EndsWith
): ...
class KeyTransformIEndsWith(
    KeyTransformTextLookupMixin, builtin_lookups.IEndsWith
): ...
class KeyTransformRegex(KeyTransformTextLookupMixin, builtin_lookups.Regex): ...
class KeyTransformIRegex(
    KeyTransformTextLookupMixin, builtin_lookups.IRegex
): ...

class KeyTransformFactory:
    key_name: Any = ...
    def __init__(self, key_name: Any) -> None: ...
    def __call__(self, *args: Any, **kwargs: Any): ...