from django.db.backends.sqlite3.base import DatabaseWrapper
from django.db.backends.utils import CursorWrapper
from django.db.models.expressions import (
    Col,
    CombinedExpression,
    Expression,
    OrderBy,
)
from django.db.models.functions.datetime import Trunc
from django.db.models.options import Options
from django.db.models.sql.query import (
    Query,
    RawQuery,
)
from itertools import chain
from typing import (
    Any,
    Dict,
    Iterator,
    List,
    Optional,
    Set,
    Tuple,
    Union,
)


class SQLAggregateCompiler:
    def as_sql(self) -> Any: ...


class SQLCompiler:
    def __init__(
        self,
        query: Union[Query, RawQuery],
        connection: DatabaseWrapper,
        using: Optional[str]
    ) -> None: ...
    def _setup_joins(self, pieces: List[str], opts: Options, alias: Optional[str]) -> Any: ...
    def apply_converters(self, rows: chain, converters: Dict[int, Any]) -> Iterator[Any]: ...
    def as_sql(self, with_limits: bool = ..., with_col_aliases: bool = ...): ...
    def as_subquery_condition(
        self,
        alias: str,
        columns: List[str],
        compiler: SQLCompiler
    ) -> Union[Tuple[str, Tuple[str]], Tuple[str, Tuple]]: ...
    def collapse_group_by(
        self,
        expressions: Union[List[Col], List[Union[Col, Trunc]], List[Expression], List[Union[Col, CombinedExpression]]],
        having: Union[List[Col], Tuple, List[CombinedExpression]]
    ) -> Union[List[Col], List[Union[Col, Trunc]], List[Expression], List[Union[Col, CombinedExpression]]]: ...
    def compile(self, node: Any, select_format: object = ...) -> Any: ...
    def deferred_to_columns(self) -> Any: ...
    def execute_sql(
        self,
        result_type: str = ...,
        chunked_fetch: bool = ...,
        chunk_size: int = ...
    ) -> Optional[CursorWrapper]: ...
    def explain_query(self) -> Iterator[str]: ...
    def find_ordering_name(
        self,
        name: str,
        opts: Options,
        alias: Optional[str] = ...,
        default_order: str = ...,
        already_seen: Optional[Union[Set[Tuple[None, Tuple[Tuple[str, str]]]], Set[Tuple[None, Tuple[Tuple[str, str]], Tuple[Tuple[str, str]], Tuple[Tuple[str, str]]]], Set[Union[Tuple[None, Tuple[Tuple[str, str]]], Tuple[Tuple[Tuple[str, str]], Tuple[Tuple[str, str]]]]]]] = ...
    ) -> List[Tuple[OrderBy, bool]]: ...
    def get_combinator_sql(
        self,
        combinator: str,
        all: bool
    ) -> Union[Tuple[List[str], List[Any]], Tuple[List[str], List[int]], Tuple[List[str], List[str]]]: ...
    def get_converters(self, expressions: Any) -> Dict[int, Any]: ...
    def get_default_columns(
        self,
        start_alias: Optional[str] = ...,
        opts: Optional[Options] = ...,
        from_parent: Any = ...
    ) -> List[Col]: ...
    def get_distinct(self) -> Tuple[List[Any], List[Any]]: ...
    def get_extra_select(
        self,
        order_by: Union[List[Tuple[OrderBy, Tuple[str, List[Any], bool]]], List[Union[Tuple[OrderBy, Tuple[str, List[int], bool]], Tuple[OrderBy, Tuple[str, List[Any], bool]]]], List[Tuple[OrderBy, Tuple[str, Tuple, bool]]]],
        select: Any
    ) -> List[Tuple[OrderBy, Tuple[str, List[Any]], None]]: ...
    def get_from_clause(
        self
    ) -> Union[Tuple[List[str], List[Union[int, str]]], Tuple[List[str], List[Any]], Tuple[List[str], List[int]], Tuple[List[str], List[str]]]: ...
    def get_group_by(
        self,
        select: Any,
        order_by: Any
    ) -> Union[List[Union[Tuple[str, List[Any]], Tuple[str, List[str]]]], List[Tuple[str, List[Any]]], List[Union[Tuple[str, List[Any]], Tuple[str, List[int]]]]]: ...
    def get_order_by(self) -> Any: ...
    def get_related_selections(
        self,
        select: Any,
        opts: Optional[Options] = ...,
        root_alias: Optional[str] = ...,
        cur_depth: int = ...,
        requested: Any = ...,
        restricted: Optional[bool] = ...
    ) -> Any: ...
    def get_select(self) -> Any: ...
    def has_results(self) -> bool: ...
    def pre_sql_setup(self) -> Any: ...
    def quote_name_unless_alias(self, name: str) -> str: ...
    def results_iter(
        self,
        results: Any = ...,
        tuple_expected: bool = ...,
        chunked_fetch: bool = ...,
        chunk_size: int = ...
    ) -> chain: ...