from collections.abc import Iterable, Iterator, Mapping, MutableMapping, MutableSet
from typing import Any, Generic, NoReturn, Protocol, TypeAlias, overload, type_check_only

from _typeshed import Incomplete
from typing_extensions import Self, TypeVar, override

_K = TypeVar("_K")
_V = TypeVar("_V")
_Z = TypeVar("_Z")

# Unfortunately, there's often check `if isinstance(var, (list, tuple))` in django
# codebase. So we need sometimes to declare exactly list or tuple.
_ListOrTuple: TypeAlias = list[_K] | tuple[_K, ...] | tuple[()]  # noqa: PYI047

@type_check_only
class _PropertyDescriptor(Generic[_K, _V]):
    """
    This helper property descriptor allows defining asynmetric getter/setters
    which mypy currently doesn't support with either:

        class HttpResponse:
            @property
            def content(...): ...
            @property.setter
            def content(...): ...

    or:

        class HttpResponse:
            def _get_content(...): ...
            def _set_content(...): ...
            content = property(_get_content, _set_content)
    """

    def __get__(self, instance: Any, owner: Any | None) -> _V: ...
    def __set__(self, instance: Any, value: _K) -> None: ...

class OrderedSet(MutableSet[_K]):
    dict: dict[_K, None]
    def __init__(self, iterable: Iterable[_K] | None = None) -> None: ...
    @override
    def add(self, item: _K) -> None: ...
    @override
    def remove(self, item: _K) -> None: ...
    @override
    def discard(self, item: _K) -> None: ...
    @override
    def __iter__(self) -> Iterator[_K]: ...
    def __reversed__(self) -> Iterator[_K]: ...
    @override
    def __contains__(self, item: object) -> bool: ...
    def __bool__(self) -> bool: ...
    @override
    def __len__(self) -> int: ...

class MultiValueDictKeyError(KeyError): ...

class MultiValueDict(dict[_K, _V]):
    @overload
    def __init__(self, key_to_list_mapping: Mapping[_K, list[_V] | None]) -> None: ...
    @overload
    def __init__(self, key_to_list_mapping: Iterable[tuple[_K, list[_V]]] = ()) -> None: ...
    @override
    def __getitem__(self, key: _K) -> _V | list[object]: ...  # type: ignore[override]
    @override
    def __setitem__(self, key: _K, value: _V) -> None: ...
    def __copy__(self) -> Self: ...
    def __deepcopy__(self, memo: MutableMapping[int, Incomplete]) -> Self: ...
    @overload
    @override
    def get(self, key: _K, default: None = None) -> _V | None: ...
    @overload
    def get(self, key: _K, default: _V) -> _V: ...
    @overload
    def get(self, key: _K, default: _Z) -> _V | _Z: ...
    def getlist(self, key: _K, default: _Z | None = None) -> list[_V] | _Z: ...
    def setlist(self, key: _K, list_: list[_V]) -> None: ...
    @overload
    @override
    def setdefault(self: MultiValueDict[_K, _V | None], key: _K, default: None = None) -> _V | None: ...
    @overload
    def setdefault(self, key: _K, default: _V) -> _V: ...
    def setlistdefault(self, key: _K, default_list: list[_V] | None = None) -> list[_V]: ...
    def appendlist(self, key: _K, value: _V) -> None: ...
    @override
    def items(self) -> Iterator[tuple[_K, _V | list[object]]]: ...  # type: ignore[override]
    def lists(self) -> Iterable[tuple[_K, list[_V]]]: ...
    # Fake to make `values` work properly
    @override
    def values(self) -> Iterator[_V | list[object]]: ...  # type: ignore[override]
    @override
    def copy(self) -> Self: ...
    def dict(self) -> dict[_K, _V | list[object]]: ...  # ty: ignore[invalid-type-form]
    # These overrides are needed to convince mypy that this isn't an abstract class
    @override
    def __delitem__(self, item: _K, /) -> None: ...
    @override
    def __len__(self) -> int: ...
    @override
    def __iter__(self) -> Iterator[_K]: ...

class ImmutableList(tuple[_V, ...]):
    warning: str
    def __new__(cls, *args: object, warning: str = "ImmutableList object is immutable.", **kwargs: object) -> Self: ...
    def complain(self, *args: object, **kwargs: object) -> NoReturn: ...
    def __delitem__(self, *args: object, **kwargs: object) -> NoReturn: ...
    def __delslice__(self, *args: object, **kwargs: object) -> NoReturn: ...
    def __iadd__(self, *args: object, **kwargs: object) -> NoReturn: ...  # type: ignore[misc]
    def __imul__(self, *args: object, **kwargs: object) -> NoReturn: ...
    def __setitem__(self, *args: object, **kwargs: object) -> NoReturn: ...
    def __setslice__(self, *args: object, **kwargs: object) -> NoReturn: ...
    def append(self, *args: object, **kwargs: object) -> NoReturn: ...
    def extend(self, *args: object, **kwargs: object) -> NoReturn: ...
    def insert(self, *args: object, **kwargs: object) -> NoReturn: ...
    def pop(self, *args: object, **kwargs: object) -> NoReturn: ...
    def remove(self, *args: object, **kwargs: object) -> NoReturn: ...
    def sort(self, *args: object, **kwargs: object) -> NoReturn: ...
    def reverse(self, *args: object, **kwargs: object) -> NoReturn: ...

@type_check_only
class _ItemCallable(Protocol[_V]):
    """Don't mess with arguments when assigning in class body in stub"""

    def __call__(self, value: _V, /) -> _V: ...

class DictWrapper(dict[str, _V]):
    func: _ItemCallable[_V]
    prefix: str
    @overload
    def __init__(self, data: Mapping[str, _V], func: _ItemCallable[_V], prefix: str) -> None: ...
    @overload
    def __init__(self, data: Iterable[tuple[str, _V]], func: _ItemCallable[_V], prefix: str) -> None: ...
    @override
    def __getitem__(self, key: str) -> _V: ...

class CaseInsensitiveMapping(Mapping[str, _V]):
    _store: dict[str, tuple[str, _V]]
    def __init__(self, data: Mapping[str, _V] | Iterable[tuple[str, _V]]) -> None: ...
    @override
    def __getitem__(self, key: str) -> _V: ...
    @override
    def __len__(self) -> int: ...
    @override
    def __eq__(self, other: object) -> bool: ...
    @override
    def __iter__(self) -> Iterator[str]: ...
    def copy(self) -> Self: ...

class DeferredSubDict(Generic[_K, _V]):
    parent_dict: dict[Any, dict[_K, _V]]
    deferred_key: Any
    def __init__(self, parent_dict: dict[Any, dict[_K, _V]], deferred_key: Any) -> None: ...
    def __getitem__(self, key: _K) -> _V: ...
