Improve database backend typing

Author: dekkers

django-stubs/db/backends/base/base.pyi

@@ -1,4 +1,5 @@
-from collections.abc import Callable
+from collections import deque
+from collections.abc import Callable, Iterable
 from contextlib import AbstractContextManager
 from datetime import tzinfo
 from logging import Logger
@@ -13,6 +14,7 @@ from django.db.backends.base.schema import BaseDatabaseSchemaEditor
 from django.db.backends.base.validation import BaseDatabaseValidation
 from django.db.backends.utils import CursorDebugWrapper, CursorWrapper
 from django.db.transaction import Atomic
+from django.db.utils import DatabaseErrorWrapper
 from django.utils.functional import cached_property
 from typing_extensions import Self
 
@@ -42,7 +44,7 @@ class BaseDatabaseWrapper:
     connection: Any
     settings_dict: dict[str, Any]
     alias: str
-    queries_log: Any
+    queries_log: deque
     force_debug_cursor: bool
     autocommit: bool
     in_atomic_block: bool
@@ -78,7 +80,7 @@ class BaseDatabaseWrapper:
     def get_connection_params(self) -> dict[str, Any]: ...
     def get_new_connection(self, conn_params: Any) -> Any: ...
     def init_connection_state(self) -> None: ...
-    def create_cursor(self, name: Any | None = None) -> Any: ...
+    def create_cursor(self, name: str | None = None) -> Any: ...
     def connect(self) -> None: ...
     def check_settings(self) -> None: ...
     def ensure_connection(self) -> None: ...
@@ -101,7 +103,7 @@ class BaseDatabaseWrapper:
     def constraint_checks_disabled(self) -> AbstractContextManager[None]: ...
     def disable_constraint_checking(self) -> bool: ...
     def enable_constraint_checking(self) -> None: ...
-    def check_constraints(self, table_names: Any | None = None) -> None: ...
+    def check_constraints(self, table_names: Iterable[str] | None = None) -> None: ...
     def is_usable(self) -> bool: ...
     def close_if_health_check_failed(self) -> None: ...
     def close_if_unusable_or_obsolete(self) -> None: ...
@@ -112,7 +114,7 @@ class BaseDatabaseWrapper:
     def validate_thread_sharing(self) -> None: ...
     def prepare_database(self) -> None: ...
     @cached_property
-    def wrap_database_errors(self) -> Any: ...
+    def wrap_database_errors(self) -> DatabaseErrorWrapper: ...
     def chunked_cursor(self) -> CursorWrapper: ...
     def make_debug_cursor(self, cursor: CursorWrapper) -> CursorDebugWrapper: ...
     def make_cursor(self, cursor: CursorWrapper) -> CursorWrapper: ...

django-stubs/db/backends/base/introspection.pyi

@@ -28,14 +28,14 @@ class BaseDatabaseIntrospection:
     def get_field_type(self, data_type: str, description: FieldInfo) -> str: ...
     def identifier_converter(self, name: str) -> str: ...
     def table_names(self, cursor: CursorWrapper | None = None, include_views: bool = False) -> list[str]: ...
-    def get_table_list(self, cursor: CursorWrapper | None) -> Any: ...
-    def get_table_description(self, cursor: CursorWrapper | None, table_name: str) -> Any: ...
+    def get_table_list(self, cursor: CursorWrapper) -> list: ...
+    def get_table_description(self, cursor: CursorWrapper, table_name: str) -> list: ...
     def get_migratable_models(self) -> Iterable[type[Model]]: ...
     def django_table_names(self, only_existing: bool = False, include_views: bool = True) -> list[str]: ...
     def installed_models(self, tables: list[str]) -> set[type[Model]]: ...
     def sequence_list(self) -> list[dict[str, str]]: ...
-    def get_sequences(self, cursor: CursorWrapper | None, table_name: str, table_fields: Any = ()) -> Any: ...
-    def get_relations(self, cursor: CursorWrapper | None, table_name: str) -> dict[str, tuple[str, str]]: ...
-    def get_primary_key_column(self, cursor: CursorWrapper | None, table_name: str) -> str | None: ...
-    def get_primary_key_columns(self, cursor: CursorWrapper | None, table_name: str) -> list[str] | None: ...
-    def get_constraints(self, cursor: CursorWrapper | None, table_name: str) -> Any: ...
+    def get_sequences(self, cursor: CursorWrapper, table_name: str, table_fields: Iterable = ()) -> list[dict]: ...
+    def get_relations(self, cursor: CursorWrapper, table_name: str) -> dict[str, tuple[str, str]]: ...
+    def get_primary_key_column(self, cursor: CursorWrapper, table_name: str) -> str | None: ...
+    def get_primary_key_columns(self, cursor: CursorWrapper, table_name: str) -> list[str] | None: ...
+    def get_constraints(self, cursor: CursorWrapper, table_name: str) -> dict[str, dict]: ...

django-stubs/db/backends/base/operations.pyi

@@ -71,10 +71,10 @@ class BaseDatabaseOperations:
     def savepoint_rollback_sql(self, sid: str) -> str: ...
     def set_time_zone_sql(self) -> str: ...
     def sql_flush(
-        self, style: Any, tables: Sequence[str], *, reset_sequences: bool = False, allow_cascade: bool = False
+        self, style: Style, tables: Sequence[str], *, reset_sequences: bool = False, allow_cascade: bool = False
     ) -> list[str]: ...
     def execute_sql_flush(self, sql_list: Iterable[str]) -> None: ...
-    def sequence_reset_by_name_sql(self, style: Style | None, sequences: list[Any]) -> list[Any]: ...
+    def sequence_reset_by_name_sql(self, style: Style, sequences: list[Any]) -> list[Any]: ...
     def sequence_reset_sql(self, style: Style, model_list: Sequence[type[Model]]) -> list[Any]: ...
     def start_transaction_sql(self) -> str: ...
     def end_transaction_sql(self, success: bool = True) -> str: ...

django-stubs/db/backends/mysql/operations.pyi

@@ -1,5 +1,6 @@
 from typing import Any
 
+from django.core.management.color import Style
 from django.db.backends.base.operations import BaseDatabaseOperations
 from django.db.backends.mysql.base import DatabaseWrapper
 from django.db.models.constants import OnConflict
@@ -37,7 +38,7 @@ class DatabaseOperations(BaseDatabaseOperations):
     @override
     def quote_name(self, name: str) -> Any: ...
     @override
-    def sequence_reset_by_name_sql(self, style: Any, sequences: Any) -> Any: ...
+    def sequence_reset_by_name_sql(self, style: Style, sequences: Any) -> Any: ...
     @override
     def validate_autopk_value(self, value: Any) -> Any: ...
     @override

django-stubs/db/backends/oracle/operations.pyi

@@ -1,5 +1,6 @@
 from typing import Any
 
+from django.core.management.color import Style
 from django.db.backends.base.operations import BaseDatabaseOperations
 from django.db.backends.oracle.base import DatabaseWrapper
 from typing_extensions import override
@@ -66,9 +67,9 @@ class DatabaseOperations(BaseDatabaseOperations):
     @override
     def regex_lookup(self, lookup_type: str) -> str: ...
     @override
-    def sequence_reset_by_name_sql(self, style: Any, sequences: Any) -> list[str]: ...
+    def sequence_reset_by_name_sql(self, style: Style, sequences: Any) -> list[str]: ...
     @override
-    def sequence_reset_sql(self, style: Any, model_list: Any) -> list[str]: ...
+    def sequence_reset_sql(self, style: Style, model_list: Any) -> list[str]: ...
     @override
     def start_transaction_sql(self) -> str: ...
     @override