homeassistant-stubs 2024.7.4__py3-none-any.whl → 2024.8.0b0__py3-none-any.whl

homeassistant-stubs/components/recorder/migration.pyi

@@ -1,13 +1,13 @@
 import abc
 from . import Recorder as Recorder
 from .auto_repairs.statistics.duplicates import delete_statistics_duplicates as delete_statistics_duplicates, delete_statistics_meta_duplicates as delete_statistics_meta_duplicates
-from .const import CONTEXT_ID_AS_BINARY_SCHEMA_VERSION as CONTEXT_ID_AS_BINARY_SCHEMA_VERSION, EVENT_TYPE_IDS_SCHEMA_VERSION as EVENT_TYPE_IDS_SCHEMA_VERSION, STATES_META_SCHEMA_VERSION as STATES_META_SCHEMA_VERSION, SupportedDialect as SupportedDialect
-from .db_schema import Base as Base, CONTEXT_ID_BIN_MAX_LENGTH as CONTEXT_ID_BIN_MAX_LENGTH, DOUBLE_PRECISION_TYPE_SQL as DOUBLE_PRECISION_TYPE_SQL, EventTypes as EventTypes, Events as Events, LEGACY_STATES_ENTITY_ID_LAST_UPDATED_INDEX as LEGACY_STATES_ENTITY_ID_LAST_UPDATED_INDEX, LEGACY_STATES_EVENT_ID_INDEX as LEGACY_STATES_EVENT_ID_INDEX, MYSQL_COLLATE as MYSQL_COLLATE, MYSQL_DEFAULT_CHARSET as MYSQL_DEFAULT_CHARSET, MigrationChanges as MigrationChanges, SCHEMA_VERSION as SCHEMA_VERSION, STATISTICS_TABLES as STATISTICS_TABLES, SchemaChanges as SchemaChanges, States as States, StatesMeta as StatesMeta, Statistics as Statistics, StatisticsMeta as StatisticsMeta, StatisticsRuns as StatisticsRuns, StatisticsShortTerm as StatisticsShortTerm, TABLE_STATES as TABLE_STATES
+from .const import CONTEXT_ID_AS_BINARY_SCHEMA_VERSION as CONTEXT_ID_AS_BINARY_SCHEMA_VERSION, EVENT_TYPE_IDS_SCHEMA_VERSION as EVENT_TYPE_IDS_SCHEMA_VERSION, LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION as LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION, STATES_META_SCHEMA_VERSION as STATES_META_SCHEMA_VERSION, SupportedDialect as SupportedDialect
+from .db_schema import BIG_INTEGER_SQL as BIG_INTEGER_SQL, Base as Base, CONTEXT_ID_BIN_MAX_LENGTH as CONTEXT_ID_BIN_MAX_LENGTH, DOUBLE_PRECISION_TYPE_SQL as DOUBLE_PRECISION_TYPE_SQL, EventTypes as EventTypes, Events as Events, LEGACY_STATES_ENTITY_ID_LAST_UPDATED_INDEX as LEGACY_STATES_ENTITY_ID_LAST_UPDATED_INDEX, LEGACY_STATES_EVENT_ID_INDEX as LEGACY_STATES_EVENT_ID_INDEX, LegacyBase as LegacyBase, MYSQL_COLLATE as MYSQL_COLLATE, MYSQL_DEFAULT_CHARSET as MYSQL_DEFAULT_CHARSET, MigrationChanges as MigrationChanges, SCHEMA_VERSION as SCHEMA_VERSION, STATISTICS_TABLES as STATISTICS_TABLES, SchemaChanges as SchemaChanges, States as States, StatesMeta as StatesMeta, Statistics as Statistics, StatisticsMeta as StatisticsMeta, StatisticsRuns as StatisticsRuns, StatisticsShortTerm as StatisticsShortTerm, TABLE_STATES as TABLE_STATES
 from .models import process_timestamp as process_timestamp
 from .models.time import datetime_to_timestamp_or_none as datetime_to_timestamp_or_none
 from .queries import batch_cleanup_entity_ids as batch_cleanup_entity_ids, delete_duplicate_short_term_statistics_row as delete_duplicate_short_term_statistics_row, delete_duplicate_statistics_row as delete_duplicate_statistics_row, find_entity_ids_to_migrate as find_entity_ids_to_migrate, find_event_type_to_migrate as find_event_type_to_migrate, find_events_context_ids_to_migrate as find_events_context_ids_to_migrate, find_states_context_ids_to_migrate as find_states_context_ids_to_migrate, find_unmigrated_short_term_statistics_rows as find_unmigrated_short_term_statistics_rows, find_unmigrated_statistics_rows as find_unmigrated_statistics_rows, has_entity_ids_to_migrate as has_entity_ids_to_migrate, has_event_type_to_migrate as has_event_type_to_migrate, has_events_context_ids_to_migrate as has_events_context_ids_to_migrate, has_states_context_ids_to_migrate as has_states_context_ids_to_migrate, has_used_states_event_ids as has_used_states_event_ids, migrate_single_short_term_statistics_row_to_timestamp as migrate_single_short_term_statistics_row_to_timestamp, migrate_single_statistics_row_to_timestamp as migrate_single_statistics_row_to_timestamp
 from .statistics import get_start_time as get_start_time
-from .tasks import CommitTask as CommitTask, EntityIDMigrationTask as EntityIDMigrationTask, EventTypeIDMigrationTask as EventTypeIDMigrationTask, EventsContextIDMigrationTask as EventsContextIDMigrationTask, PostSchemaMigrationTask as PostSchemaMigrationTask, RecorderTask as RecorderTask, StatesContextIDMigrationTask as StatesContextIDMigrationTask, StatisticsTimestampMigrationCleanupTask as StatisticsTimestampMigrationCleanupTask
+from .tasks import CommitTask as CommitTask, EntityIDPostMigrationTask as EntityIDPostMigrationTask, PostSchemaMigrationTask as PostSchemaMigrationTask, RecorderTask as RecorderTask, StatisticsTimestampMigrationCleanupTask as StatisticsTimestampMigrationCleanupTask
 from .util import database_job_retry_wrapper as database_job_retry_wrapper, execute_stmt_lambda_element as execute_stmt_lambda_element, get_index_by_name as get_index_by_name, retryable_database_job as retryable_database_job, session_scope as session_scope
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
@@ -17,10 +17,13 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.util.enum import try_parse_enum as try_parse_enum
 from homeassistant.util.ulid import ulid_at_time as ulid_at_time, ulid_to_bytes as ulid_to_bytes
 from sqlalchemy.engine import CursorResult as CursorResult, Engine as Engine
+from sqlalchemy.engine.interfaces import ReflectedForeignKeyConstraint as ReflectedForeignKeyConstraint
 from sqlalchemy.orm.session import Session as Session
 from sqlalchemy.sql.lambdas import StatementLambdaElement as StatementLambdaElement
+from typing import Any
 
 LIVE_MIGRATION_MIN_SCHEMA_VERSION: int
+MIGRATION_NOTE_OFFLINE: str
 _EMPTY_ENTITY_ID: str
 _EMPTY_EVENT_TYPE: str
 _LOGGER: Incomplete
@@ -41,26 +44,180 @@ def raise_if_exception_missing_str(ex: Exception, match_substrs: Iterable[str])
 def _get_schema_version(session: Session) -> int | None: ...
 def get_schema_version(session_maker: Callable[[], Session]) -> int | None: ...
 
-@dataclass
+@dataclass(frozen=True)
 class SchemaValidationStatus:
     current_version: int
+    migration_needed: bool
     schema_errors: set[str]
-    valid: bool
-    def __init__(self, current_version, schema_errors, valid) -> None: ...
+    start_version: int
+    def __init__(self, current_version, migration_needed, schema_errors, start_version) -> None: ...
 
 def _schema_is_current(current_version: int) -> bool: ...
 def validate_db_schema(hass: HomeAssistant, instance: Recorder, session_maker: Callable[[], Session]) -> SchemaValidationStatus | None: ...
 def _find_schema_errors(hass: HomeAssistant, instance: Recorder, session_maker: Callable[[], Session]) -> set[str]: ...
 def live_migration(schema_status: SchemaValidationStatus) -> bool: ...
-def migrate_schema(instance: Recorder, hass: HomeAssistant, engine: Engine, session_maker: Callable[[], Session], schema_status: SchemaValidationStatus) -> None: ...
+def pre_migrate_schema(engine: Engine) -> None: ...
+def _migrate_schema(instance: Recorder, hass: HomeAssistant, engine: Engine, session_maker: Callable[[], Session], schema_status: SchemaValidationStatus, end_version: int) -> SchemaValidationStatus: ...
+def migrate_schema_non_live(instance: Recorder, hass: HomeAssistant, engine: Engine, session_maker: Callable[[], Session], schema_status: SchemaValidationStatus) -> SchemaValidationStatus: ...
+def migrate_schema_live(instance: Recorder, hass: HomeAssistant, engine: Engine, session_maker: Callable[[], Session], schema_status: SchemaValidationStatus) -> SchemaValidationStatus: ...
 def _create_index(session_maker: Callable[[], Session], table_name: str, index_name: str) -> None: ...
 def _execute_or_collect_error(session_maker: Callable[[], Session], query: str, errors: list[str]) -> bool: ...
 def _drop_index(session_maker: Callable[[], Session], table_name: str, index_name: str, quiet: bool | None = None) -> None: ...
 def _add_columns(session_maker: Callable[[], Session], table_name: str, columns_def: list[str]) -> None: ...
 def _modify_columns(session_maker: Callable[[], Session], engine: Engine, table_name: str, columns_def: list[str]) -> None: ...
 def _update_states_table_with_foreign_key_options(session_maker: Callable[[], Session], engine: Engine) -> None: ...
-def _drop_foreign_key_constraints(session_maker: Callable[[], Session], engine: Engine, table: str, columns: list[str]) -> None: ...
+def _drop_foreign_key_constraints(session_maker: Callable[[], Session], engine: Engine, table: str, column: str) -> list[tuple[str, str, ReflectedForeignKeyConstraint]]: ...
+def _restore_foreign_key_constraints(session_maker: Callable[[], Session], engine: Engine, dropped_constraints: list[tuple[str, str, ReflectedForeignKeyConstraint]]) -> None: ...
 def _apply_update(instance: Recorder, hass: HomeAssistant, engine: Engine, session_maker: Callable[[], Session], new_version: int, old_version: int) -> None: ...
+
+class _SchemaVersionMigrator(ABC, metaclass=abc.ABCMeta):
+    __migrators: dict[int, type[_SchemaVersionMigrator]]
+    def __init_subclass__(cls, target_version: int, **kwargs: Any) -> None: ...
+    instance: Incomplete
+    hass: Incomplete
+    engine: Incomplete
+    session_maker: Incomplete
+    old_version: Incomplete
+    column_types: Incomplete
+    def __init__(self, instance: Recorder, hass: HomeAssistant, engine: Engine, session_maker: Callable[[], Session], old_version: int) -> None: ...
+    @classmethod
+    def get_migrator(cls, target_version: int) -> type[_SchemaVersionMigrator]: ...
+    def apply_update(self) -> None: ...
+    @abstractmethod
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion1Migrator(_SchemaVersionMigrator, target_version=1):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion2Migrator(_SchemaVersionMigrator, target_version=2):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion3Migrator(_SchemaVersionMigrator, target_version=3):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion4Migrator(_SchemaVersionMigrator, target_version=4):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion5Migrator(_SchemaVersionMigrator, target_version=5):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion6Migrator(_SchemaVersionMigrator, target_version=6):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion7Migrator(_SchemaVersionMigrator, target_version=7):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion8Migrator(_SchemaVersionMigrator, target_version=8):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion9Migrator(_SchemaVersionMigrator, target_version=9):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion10Migrator(_SchemaVersionMigrator, target_version=10):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion11Migrator(_SchemaVersionMigrator, target_version=11):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion12Migrator(_SchemaVersionMigrator, target_version=12):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion13Migrator(_SchemaVersionMigrator, target_version=13):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion14Migrator(_SchemaVersionMigrator, target_version=14):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion15Migrator(_SchemaVersionMigrator, target_version=15):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion16Migrator(_SchemaVersionMigrator, target_version=16):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion17Migrator(_SchemaVersionMigrator, target_version=17):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion18Migrator(_SchemaVersionMigrator, target_version=18):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion19Migrator(_SchemaVersionMigrator, target_version=19):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion20Migrator(_SchemaVersionMigrator, target_version=20):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion21Migrator(_SchemaVersionMigrator, target_version=21):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion22Migrator(_SchemaVersionMigrator, target_version=22):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion23Migrator(_SchemaVersionMigrator, target_version=23):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion24Migrator(_SchemaVersionMigrator, target_version=24):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion25Migrator(_SchemaVersionMigrator, target_version=25):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion26Migrator(_SchemaVersionMigrator, target_version=26):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion27Migrator(_SchemaVersionMigrator, target_version=27):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion28Migrator(_SchemaVersionMigrator, target_version=28):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion29Migrator(_SchemaVersionMigrator, target_version=29):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion30Migrator(_SchemaVersionMigrator, target_version=30):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion31Migrator(_SchemaVersionMigrator, target_version=31):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion32Migrator(_SchemaVersionMigrator, target_version=32):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion33Migrator(_SchemaVersionMigrator, target_version=33):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion34Migrator(_SchemaVersionMigrator, target_version=34):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion35Migrator(_SchemaVersionMigrator, target_version=35):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion36Migrator(_SchemaVersionMigrator, target_version=36):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion37Migrator(_SchemaVersionMigrator, target_version=37):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion38Migrator(_SchemaVersionMigrator, target_version=38):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion39Migrator(_SchemaVersionMigrator, target_version=39):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion40Migrator(_SchemaVersionMigrator, target_version=40):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion41Migrator(_SchemaVersionMigrator, target_version=41):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion42Migrator(_SchemaVersionMigrator, target_version=42):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion43Migrator(_SchemaVersionMigrator, target_version=43):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion44Migrator(_SchemaVersionMigrator, target_version=44):
+    def _apply_update(self) -> None: ...
+
 def _migrate_statistics_columns_to_timestamp_removing_duplicates(hass: HomeAssistant, instance: Recorder, session_maker: Callable[[], Session], engine: Engine) -> None: ...
 def _correct_table_character_set_and_collation(table: str, session_maker: Callable[[], Session]) -> None: ...
 def post_schema_migration(instance: Recorder, old_version: int, new_version: int) -> None: ...
@@ -71,51 +228,91 @@ def _migrate_statistics_columns_to_timestamp_one_by_one(instance: Recorder, sess
 def _migrate_statistics_columns_to_timestamp(instance: Recorder, session_maker: Callable[[], Session], engine: Engine) -> None: ...
 def _context_id_to_bytes(context_id: str | None) -> bytes | None: ...
 def _generate_ulid_bytes_at_time(timestamp: float | None) -> bytes: ...
-def migrate_states_context_ids(instance: Recorder) -> bool: ...
-def migrate_events_context_ids(instance: Recorder) -> bool: ...
-def migrate_event_type_ids(instance: Recorder) -> bool: ...
-def migrate_entity_ids(instance: Recorder) -> bool: ...
 def post_migrate_entity_ids(instance: Recorder) -> bool: ...
 def cleanup_legacy_states_event_ids(instance: Recorder) -> bool: ...
 def _initialize_database(session: Session) -> bool: ...
 def initialize_database(session_maker: Callable[[], Session]) -> bool: ...
 
+@dataclass(slots=True)
+class MigrationTask(RecorderTask):
+    migrator: BaseRunTimeMigration
+    commit_before = ...
+    def run(self, instance: Recorder) -> None: ...
+    def __init__(self, migrator) -> None: ...
+
+@dataclass(slots=True)
+class CommitBeforeMigrationTask(MigrationTask):
+    commit_before = ...
+    def __init__(self, migrator) -> None: ...
+
+@dataclass(frozen=True, kw_only=True)
+class NeedsMigrateResult:
+    needs_migrate: bool
+    migration_done: bool
+    def __init__(self, *, needs_migrate, migration_done) -> None: ...
+
 class BaseRunTimeMigration(ABC, metaclass=abc.ABCMeta):
     required_schema_version: int
     migration_version: int
     migration_id: str
-    task: Callable[[], RecorderTask]
+    task = MigrationTask
     schema_version: Incomplete
-    session: Incomplete
     migration_changes: Incomplete
-    def __init__(self, session: Session, schema_version: int, migration_changes: dict[str, int]) -> None: ...
+    def __init__(self, schema_version: int, migration_changes: dict[str, int]) -> None: ...
+    def do_migrate(self, instance: Recorder, session: Session) -> None: ...
+    @staticmethod
+    @abstractmethod
+    def migrate_data(instance: Recorder) -> bool: ...
+    def migration_done(self, instance: Recorder, session: Session | None) -> None: ...
+    @abstractmethod
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate(self, instance: Recorder, session: Session) -> bool: ...
+
+class BaseRunTimeMigrationWithQuery(BaseRunTimeMigration, metaclass=abc.ABCMeta):
     @abstractmethod
     def needs_migrate_query(self) -> StatementLambdaElement: ...
-    def needs_migrate(self) -> bool: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
 
-class StatesContextIDMigration(BaseRunTimeMigration):
+class StatesContextIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = CONTEXT_ID_AS_BINARY_SCHEMA_VERSION
     migration_id: str
-    task = StatesContextIDMigrationTask
+    @staticmethod
+    def migrate_data(instance: Recorder) -> bool: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
-class EventsContextIDMigration(BaseRunTimeMigration):
+class EventsContextIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = CONTEXT_ID_AS_BINARY_SCHEMA_VERSION
     migration_id: str
-    task = EventsContextIDMigrationTask
+    @staticmethod
+    def migrate_data(instance: Recorder) -> bool: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
-class EventTypeIDMigration(BaseRunTimeMigration):
+class EventTypeIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = EVENT_TYPE_IDS_SCHEMA_VERSION
     migration_id: str
-    task = EventTypeIDMigrationTask
+    task = CommitBeforeMigrationTask
+    @staticmethod
+    def migrate_data(instance: Recorder) -> bool: ...
+    def migration_done(self, instance: Recorder, session: Session | None) -> None: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
-class EntityIDMigration(BaseRunTimeMigration):
+class EntityIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = STATES_META_SCHEMA_VERSION
     migration_id: str
-    task = EntityIDMigrationTask
+    task = CommitBeforeMigrationTask
+    @staticmethod
+    def migrate_data(instance: Recorder) -> bool: ...
+    def migration_done(self, instance: Recorder, _session: Session | None) -> None: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
+class EventIDPostMigration(BaseRunTimeMigration):
+    migration_id: str
+    task = MigrationTask
+    @staticmethod
+    def migrate_data(instance: Recorder) -> bool: ...
+    @staticmethod
+    def _legacy_event_id_foreign_key_exists(instance: Recorder) -> bool: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+
 def _mark_migration_done(session: Session, migration: type[BaseRunTimeMigration]) -> None: ...
 def rebuild_sqlite_table(session_maker: Callable[[], Session], engine: Engine, table: type[Base]) -> None: ...

homeassistant-stubs/components/recorder/models/context.pyi

@@ -1,9 +1,7 @@
 from _typeshed import Incomplete
-from homeassistant.util.ulid import bytes_to_ulid as bytes_to_ulid, ulid_to_bytes as ulid_to_bytes
+from homeassistant.util.ulid import bytes_to_ulid as bytes_to_ulid, bytes_to_ulid_or_none as bytes_to_ulid_or_none, ulid_to_bytes as ulid_to_bytes, ulid_to_bytes_or_none as ulid_to_bytes_or_none
 
 _LOGGER: Incomplete
 
-def ulid_to_bytes_or_none(ulid: str | None) -> bytes | None: ...
-def bytes_to_ulid_or_none(_bytes: bytes | None) -> str | None: ...
 def uuid_hex_to_bytes_or_none(uuid_hex: str | None) -> bytes | None: ...
 def bytes_to_uuid_hex_or_none(_bytes: bytes | None) -> str | None: ...

homeassistant-stubs/components/recorder/statistics.pyi

@@ -36,7 +36,7 @@ class ShortTermStatisticsRunCache:
     def get_latest_ids(self, metadata_ids: set[int]) -> dict[int, int]: ...
     def set_latest_id_for_metadata_id(self, metadata_id: int, id_: int) -> None: ...
     def set_latest_ids_for_metadata_ids(self, metadata_id_to_id: dict[int, int]) -> None: ...
-    def __init__(self, _latest_id_by_metadata_id) -> None: ...
+    def __init__(self, _latest_id_by_metadata_id=...) -> None: ...
 
 class BaseStatisticsRow(TypedDict, total=False):
     start: float
@@ -76,7 +76,7 @@ class ValidationIssue:
     type: str
     data: dict[str, str | None] | None = ...
     def as_dict(self) -> dict: ...
-    def __init__(self, type, data) -> None: ...
+    def __init__(self, type, data=...) -> None: ...
 
 def get_start_time() -> datetime: ...
 def _compile_hourly_statistics_summary_mean_stmt(start_time_ts: float, end_time_ts: float) -> StatementLambdaElement: ...

homeassistant-stubs/components/recorder/table_managers/__init__.pyi

@@ -6,7 +6,6 @@ from typing import Any
 
 class BaseTableManager:
     _id_map: LRU[EventType[Any] | str, int]
-    active: bool
     recorder: Incomplete
     _pending: Incomplete
     def __init__(self, recorder: Recorder) -> None: ...

homeassistant-stubs/components/recorder/table_managers/event_data.pyi

@@ -14,7 +14,6 @@ CACHE_SIZE: int
 _LOGGER: Incomplete
 
 class EventDataManager(BaseLRUTableManager[EventData]):
-    active: bool
     def __init__(self, recorder: Recorder) -> None: ...
     def serialize_from_event(self, event: Event) -> bytes | None: ...
     def load(self, events: list[Event], session: Session) -> None: ...

homeassistant-stubs/components/recorder/table_managers/event_types.pyi

@@ -15,6 +15,7 @@ from typing import Any
 CACHE_SIZE: int
 
 class EventTypeManager(BaseLRUTableManager[EventTypes]):
+    active: bool
     _non_existent_event_types: Incomplete
     def __init__(self, recorder: Recorder) -> None: ...
     def load(self, events: list[Event], session: Session) -> None: ...

homeassistant-stubs/components/recorder/table_managers/state_attributes.pyi

@@ -14,7 +14,6 @@ CACHE_SIZE: int
 _LOGGER: Incomplete
 
 class StateAttributesManager(BaseLRUTableManager[StateAttributes]):
-    active: bool
     def __init__(self, recorder: Recorder) -> None: ...
     def serialize_from_event(self, event: Event[EventStateChangedData]) -> bytes | None: ...
     def load(self, events: list[Event[EventStateChangedData]], session: Session) -> None: ...

homeassistant-stubs/components/recorder/table_managers/states_meta.pyi

@@ -11,6 +11,7 @@ from sqlalchemy.orm.session import Session as Session
 CACHE_SIZE: int
 
 class StatesMetaManager(BaseLRUTableManager[StatesMeta]):
+    active: bool
     _did_first_load: bool
     def __init__(self, recorder: Recorder) -> None: ...
     def load(self, events: list[Event[EventStateChangedData]], session: Session) -> None: ...

homeassistant-stubs/components/recorder/tasks.pyi

@@ -157,34 +157,10 @@ class AdjustLRUSizeTask(RecorderTask):
     commit_before = ...
     def run(self, instance: Recorder) -> None: ...
 
-@dataclass(slots=True)
-class StatesContextIDMigrationTask(RecorderTask):
-    commit_before = ...
-    def run(self, instance: Recorder) -> None: ...
-
-@dataclass(slots=True)
-class EventsContextIDMigrationTask(RecorderTask):
-    commit_before = ...
-    def run(self, instance: Recorder) -> None: ...
-
-@dataclass(slots=True)
-class EventTypeIDMigrationTask(RecorderTask):
-    commit_before = ...
-    def run(self, instance: Recorder) -> None: ...
-
-@dataclass(slots=True)
-class EntityIDMigrationTask(RecorderTask):
-    commit_before = ...
-    def run(self, instance: Recorder) -> None: ...
-
 @dataclass(slots=True)
 class EntityIDPostMigrationTask(RecorderTask):
     def run(self, instance: Recorder) -> None: ...
 
-@dataclass(slots=True)
-class EventIdMigrationTask(RecorderTask):
-    def run(self, instance: Recorder) -> None: ...
-
 @dataclass(slots=True)
 class RefreshEventTypesTask(RecorderTask):
     event_types: list[EventType[Any] | str]

homeassistant-stubs/components/recorder/util.pyi

@@ -1,12 +1,13 @@
 from . import Recorder as Recorder
-from .const import DATA_INSTANCE as DATA_INSTANCE, DEFAULT_MAX_BIND_VARS as DEFAULT_MAX_BIND_VARS, DOMAIN as DOMAIN, SQLITE_MAX_BIND_VARS as SQLITE_MAX_BIND_VARS, SQLITE_MODERN_MAX_BIND_VARS as SQLITE_MODERN_MAX_BIND_VARS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
+from .const import DEFAULT_MAX_BIND_VARS as DEFAULT_MAX_BIND_VARS, DOMAIN as DOMAIN, SQLITE_MAX_BIND_VARS as SQLITE_MAX_BIND_VARS, SQLITE_MODERN_MAX_BIND_VARS as SQLITE_MODERN_MAX_BIND_VARS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
 from .db_schema import RecorderRuns as RecorderRuns, TABLES_TO_CHECK as TABLES_TO_CHECK, TABLE_RECORDER_RUNS as TABLE_RECORDER_RUNS, TABLE_SCHEMA_CHANGES as TABLE_SCHEMA_CHANGES
 from .models import DatabaseEngine as DatabaseEngine, DatabaseOptimizer as DatabaseOptimizer, StatisticPeriod as StatisticPeriod, UnsupportedDialect as UnsupportedDialect, process_timestamp as process_timestamp
 from _typeshed import Incomplete
 from awesomeversion import AwesomeVersion
-from collections.abc import Callable, Sequence
+from collections.abc import Callable, Generator, Sequence
 from datetime import date, datetime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.recorder import DATA_INSTANCE as DATA_INSTANCE, get_instance as get_instance, session_scope as session_scope
 from sqlalchemy.engine import Result as Result, Row as Row
 from sqlalchemy.engine.interfaces import DBAPIConnection as DBAPIConnection
 from sqlalchemy.exc import OperationalError
@@ -15,7 +16,6 @@ from sqlalchemy.orm.session import Session as Session
 from sqlalchemy.sql.lambdas import StatementLambdaElement as StatementLambdaElement
 from sqlite3.dbapi2 import Cursor as SQLiteCursor
 from typing import Any, NoReturn
-from typing_extensions import Generator
 
 _LOGGER: Incomplete
 RETRIES: int
@@ -40,6 +40,7 @@ MARIADB_WITH_FIXED_IN_QUERIES_108: Incomplete
 MIN_VERSION_MYSQL: Incomplete
 MIN_VERSION_PGSQL: Incomplete
 MIN_VERSION_SQLITE: Incomplete
+UPCOMING_MIN_VERSION_SQLITE: Incomplete
 MIN_VERSION_SQLITE_MODERN_BIND_VARS: Incomplete
 MAX_RESTART_TIME: Incomplete
 RETRYABLE_MYSQL_ERRORS: Incomplete
@@ -47,7 +48,6 @@ FIRST_POSSIBLE_SUNDAY: int
 SUNDAY_WEEKDAY: int
 DAYS_IN_WEEK: int
 
-def session_scope(*, hass: HomeAssistant | None = None, session: Session | None = None, exception_filter: Callable[[Exception], bool] | None = None, read_only: bool = False) -> Generator[Session]: ...
 def execute(qry: Query, to_native: bool = False, validate_entity_ids: bool = True) -> list[Row]: ...
 def execute_stmt_lambda_element(session: Session, stmt: StatementLambdaElement, start_time: datetime | None = None, end_time: datetime | None = None, yield_per: int = ..., orm_rows: bool = True) -> Sequence[Row] | Result: ...
 def validate_or_move_away_sqlite_database(dburl: str) -> bool: ...
@@ -60,7 +60,10 @@ def move_away_broken_database(dbfile: str) -> None: ...
 def execute_on_connection(dbapi_connection: DBAPIConnection, statement: str) -> None: ...
 def query_on_connection(dbapi_connection: DBAPIConnection, statement: str) -> Any: ...
 def _fail_unsupported_dialect(dialect_name: str) -> NoReturn: ...
-def _fail_unsupported_version(server_version: str, dialect_name: str, minimum_version: str) -> NoReturn: ...
+def _raise_if_version_unsupported(server_version: str, dialect_name: str, minimum_version: str) -> NoReturn: ...
+def _async_delete_issue_deprecated_version(hass: HomeAssistant, dialect_name: str) -> None: ...
+def _async_create_issue_deprecated_version(hass: HomeAssistant, server_version: AwesomeVersion, dialect_name: str, min_version: AwesomeVersion) -> None: ...
+def _extract_version_from_server_response_or_raise(server_response: str) -> AwesomeVersion: ...
 def _extract_version_from_server_response(server_response: str) -> AwesomeVersion | None: ...
 def _datetime_or_none(value: str) -> datetime | None: ...
 def build_mysqldb_conv() -> dict: ...
@@ -80,7 +83,6 @@ def async_migration_in_progress(hass: HomeAssistant) -> bool: ...
 def async_migration_is_live(hass: HomeAssistant) -> bool: ...
 def second_sunday(year: int, month: int) -> date: ...
 def is_second_sunday(date_time: datetime) -> bool: ...
-def get_instance(hass: HomeAssistant) -> Recorder: ...
 
 PERIOD_SCHEMA: Incomplete
 

homeassistant-stubs/components/recorder/websocket_api.pyi

@@ -29,4 +29,3 @@ def ws_update_statistics_metadata(hass: HomeAssistant, connection: websocket_api
 def ws_change_statistics_unit(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def ws_adjust_sum_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def ws_import_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
-def ws_info(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/remote/__init__.pyi

@@ -38,9 +38,9 @@ DEFAULT_DELAY_SECS: float
 DEFAULT_HOLD_SECS: int
 
 class RemoteEntityFeature(IntFlag):
-    LEARN_COMMAND: int
-    DELETE_COMMAND: int
-    ACTIVITY: int
+    LEARN_COMMAND = 1
+    DELETE_COMMAND = 2
+    ACTIVITY = 4
 
 _DEPRECATED_SUPPORT_LEARN_COMMAND: Incomplete
 _DEPRECATED_SUPPORT_DELETE_COMMAND: Incomplete
@@ -53,7 +53,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ..
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 
 class RemoteEntityDescription(ToggleEntityDescription, frozen_or_thawed=True):
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=...) -> None: ...
 
 CACHED_PROPERTIES_WITH_ATTR_: Incomplete
 

homeassistant-stubs/components/renault/binary_sensor.pyi

@@ -11,7 +11,7 @@ from renault_api.kamereon.models import KamereonVehicleBatteryStatusData
 class RenaultBinarySensorEntityDescription(BinarySensorEntityDescription, RenaultDataEntityDescription):
     on_key: str
     on_value: StateType
-    def __init__(self, coordinator, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, on_key, on_value) -> None: ...
+    def __init__(self, coordinator, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., on_key, on_value) -> None: ...
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/renault/button.pyi

@@ -11,7 +11,7 @@ from typing import Any
 class RenaultButtonEntityDescription(ButtonEntityDescription):
     async_press: Callable[[RenaultButtonEntity], Coroutine[Any, Any, Any]]
     requires_electricity: bool = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, async_press, requires_electricity) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., async_press, requires_electricity=...) -> None: ...
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/renault/entity.pyi

@@ -13,7 +13,7 @@ class RenaultDataRequiredKeysMixin:
 
 @dataclass(frozen=True)
 class RenaultDataEntityDescription(EntityDescription, RenaultDataRequiredKeysMixin):
-    def __init__(self, coordinator, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement) -> None: ...
+    def __init__(self, coordinator, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=...) -> None: ...
 
 class RenaultEntity(Entity):
     _attr_has_entity_name: bool

homeassistant-stubs/components/renault/renault_vehicle.pyi

@@ -21,7 +21,7 @@ class RenaultCoordinatorDescription:
     key: str
     update_method: Callable[[RenaultVehicle], Callable[[], Awaitable[models.KamereonVehicleDataAttributes]]]
     requires_electricity: bool = ...
-    def __init__(self, endpoint, key, update_method, requires_electricity) -> None: ...
+    def __init__(self, endpoint, key, update_method, requires_electricity=...) -> None: ...
 
 class RenaultVehicleProxy:
     hass: Incomplete

homeassistant-stubs/components/renault/select.pyi

@@ -10,7 +10,7 @@ from renault_api.kamereon.models import KamereonVehicleBatteryStatusData
 @dataclass(frozen=True, kw_only=True)
 class RenaultSelectEntityDescription(SelectEntityDescription, RenaultDataEntityDescription):
     data_key: str
-    def __init__(self, coordinator, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, options, data_key) -> None: ...
+    def __init__(self, coordinator, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., options=..., data_key) -> None: ...
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/renault/sensor.pyi

@@ -20,7 +20,7 @@ class RenaultSensorEntityDescription(SensorEntityDescription, RenaultDataEntityD
     condition_lambda: Callable[[RenaultVehicleProxy], bool] | None = ...
     requires_fuel: bool = ...
     value_lambda: Callable[[RenaultSensor[T]], StateType | datetime] | None = ...
-    def __init__(self, coordinator, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, data_key, entity_class, condition_lambda, requires_fuel, value_lambda) -> None: ...
+    def __init__(self, coordinator, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., data_key, entity_class, condition_lambda=..., requires_fuel=..., value_lambda=...) -> None: ...
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/repairs/websocket_api.pyi

@@ -23,5 +23,5 @@ class RepairsFlowIndexView(FlowManagerIndexView):
 class RepairsFlowResourceView(FlowManagerResourceView):
     url: str
     name: str
-    async def get(self, request: web.Request, flow_id: str) -> web.Response: ...
+    async def get(self, request: web.Request, /, flow_id: str) -> web.Response: ...
     async def post(self, request: web.Request, flow_id: str) -> web.Response: ...

homeassistant-stubs/components/rfxtrx/sensor.pyi

@@ -23,7 +23,7 @@ def _rssi_convert(value: int | None) -> str | None: ...
 @dataclass(frozen=True)
 class RfxtrxSensorEntityDescription(SensorEntityDescription):
     convert: Callable[[Any], StateType | date | datetime | Decimal] = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, convert) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., convert=...) -> None: ...
 
 SENSOR_TYPES: Incomplete
 SENSOR_TYPES_DICT: Incomplete

homeassistant-stubs/components/ring/binary_sensor.pyi

@@ -15,7 +15,7 @@ from typing import Any
 @dataclass(frozen=True, kw_only=True)
 class RingBinarySensorEntityDescription(BinarySensorEntityDescription):
     exists_fn: Callable[[RingGeneric], bool]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, exists_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., exists_fn) -> None: ...
 
 BINARY_SENSOR_TYPES: tuple[RingBinarySensorEntityDescription, ...]
 

homeassistant-stubs/components/ring/light.pyi

@@ -15,8 +15,8 @@ _LOGGER: Incomplete
 SKIP_UPDATES_DELAY: Incomplete
 
 class OnOffState(StrEnum):
-    ON: Incomplete
-    OFF: Incomplete
+    ON = ...
+    OFF = ...
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...