homeassistant-stubs 2024.9.3__py3-none-any.whl → 2024.10.0__py3-none-any.whl

homeassistant-stubs/components/rainmachine/__init__.pyi

@@ -1,7 +1,6 @@
 from .config_flow import get_client_controller as get_client_controller
 from .const import CONF_ALLOW_INACTIVE_ZONES_TO_RUN as CONF_ALLOW_INACTIVE_ZONES_TO_RUN, CONF_DEFAULT_ZONE_RUN_TIME as CONF_DEFAULT_ZONE_RUN_TIME, CONF_DURATION as CONF_DURATION, CONF_USE_APP_RUN_TIMES as CONF_USE_APP_RUN_TIMES, DATA_API_VERSIONS as DATA_API_VERSIONS, DATA_MACHINE_FIRMWARE_UPDATE_STATUS as DATA_MACHINE_FIRMWARE_UPDATE_STATUS, DATA_PROGRAMS as DATA_PROGRAMS, DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_CURRENT as DATA_RESTRICTIONS_CURRENT, DATA_RESTRICTIONS_UNIVERSAL as DATA_RESTRICTIONS_UNIVERSAL, DATA_ZONES as DATA_ZONES, DEFAULT_ZONE_RUN as DEFAULT_ZONE_RUN, DOMAIN as DOMAIN, LOGGER as LOGGER
 from .coordinator import RainMachineDataUpdateCoordinator as RainMachineDataUpdateCoordinator
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
@@ -10,8 +9,7 @@ from homeassistant.const import CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_IP_ADDRES
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady, HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import aiohttp_client as aiohttp_client
-from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
-from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity, UpdateFailed as UpdateFailed
+from homeassistant.helpers.update_coordinator import UpdateFailed as UpdateFailed
 from homeassistant.util.dt import as_timestamp as as_timestamp, utcnow as utcnow
 from homeassistant.util.network import is_ip_address as is_ip_address
 from regenmaschine.controller import Controller as Controller
@@ -70,18 +68,3 @@ async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry)
 async def async_unload_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> bool: ...
 async def async_migrate_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> bool: ...
 async def async_reload_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> None: ...
-
-class RainMachineEntity(CoordinatorEntity[RainMachineDataUpdateCoordinator]):
-    _attr_has_entity_name: bool
-    _attr_extra_state_attributes: Incomplete
-    _attr_unique_id: Incomplete
-    _entry: Incomplete
-    _data: Incomplete
-    _version_coordinator: Incomplete
-    entity_description: Incomplete
-    def __init__(self, entry: RainMachineConfigEntry, data: RainMachineData, description: RainMachineEntityDescription) -> None: ...
-    @property
-    def device_info(self) -> DeviceInfo: ...
-    def _handle_coordinator_update(self) -> None: ...
-    async def async_added_to_hass(self) -> None: ...
-    def update_from_latest_data(self) -> None: ...

homeassistant-stubs/components/rainmachine/binary_sensor.pyi

@@ -1,6 +1,6 @@
-from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineEntity as RainMachineEntity
+from . import RainMachineConfigEntry as RainMachineConfigEntry
 from .const import DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_CURRENT as DATA_RESTRICTIONS_CURRENT
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
+from .entity import RainMachineEntity as RainMachineEntity, RainMachineEntityDescription as RainMachineEntityDescription
 from .util import EntityDomainReplacementStrategy as EntityDomainReplacementStrategy, async_finish_entity_domain_replacements as async_finish_entity_domain_replacements, key_exists as key_exists
 from _typeshed import Incomplete
 from dataclasses import dataclass

homeassistant-stubs/components/rainmachine/button.pyi

@@ -1,6 +1,6 @@
-from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineEntity as RainMachineEntity
+from . import RainMachineConfigEntry as RainMachineConfigEntry
 from .const import DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
+from .entity import RainMachineEntity as RainMachineEntity, RainMachineEntityDescription as RainMachineEntityDescription
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass

homeassistant-stubs/components/rainmachine/entity.pyi

@@ -0,0 +1,30 @@
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData
+from .const import DATA_API_VERSIONS as DATA_API_VERSIONS, DOMAIN as DOMAIN
+from .coordinator import RainMachineDataUpdateCoordinator as RainMachineDataUpdateCoordinator
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.const import CONF_IP_ADDRESS as CONF_IP_ADDRESS, CONF_PORT as CONF_PORT
+from homeassistant.core import callback as callback
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity import EntityDescription as EntityDescription
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+
+@dataclass(frozen=True, kw_only=True)
+class RainMachineEntityDescription(EntityDescription):
+    api_category: str
+    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=..., api_category) -> None: ...
+
+class RainMachineEntity(CoordinatorEntity[RainMachineDataUpdateCoordinator]):
+    _attr_has_entity_name: bool
+    _attr_extra_state_attributes: Incomplete
+    _attr_unique_id: Incomplete
+    _entry: Incomplete
+    _data: Incomplete
+    _version_coordinator: Incomplete
+    entity_description: Incomplete
+    def __init__(self, entry: RainMachineConfigEntry, data: RainMachineData, description: RainMachineEntityDescription) -> None: ...
+    @property
+    def device_info(self) -> DeviceInfo: ...
+    def _handle_coordinator_update(self) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    def update_from_latest_data(self) -> None: ...

homeassistant-stubs/components/rainmachine/select.pyi

@@ -1,6 +1,6 @@
-from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData
 from .const import DATA_RESTRICTIONS_UNIVERSAL as DATA_RESTRICTIONS_UNIVERSAL
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
+from .entity import RainMachineEntity as RainMachineEntity, RainMachineEntityDescription as RainMachineEntityDescription
 from .util import key_exists as key_exists
 from _typeshed import Incomplete
 from dataclasses import dataclass

homeassistant-stubs/components/rainmachine/sensor.pyi

@@ -1,6 +1,6 @@
-from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData
 from .const import DATA_PROGRAMS as DATA_PROGRAMS, DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_ZONES as DATA_ZONES
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
+from .entity import RainMachineEntity as RainMachineEntity, RainMachineEntityDescription as RainMachineEntityDescription
 from .util import EntityDomainReplacementStrategy as EntityDomainReplacementStrategy, RUN_STATE_MAP as RUN_STATE_MAP, RunStates as RunStates, async_finish_entity_domain_replacements as async_finish_entity_domain_replacements, key_exists as key_exists
 from _typeshed import Incomplete
 from dataclasses import dataclass

homeassistant-stubs/components/rainmachine/switch.pyi

@@ -1,6 +1,6 @@
-from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity, async_update_programs_and_zones as async_update_programs_and_zones
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, async_update_programs_and_zones as async_update_programs_and_zones
 from .const import CONF_ALLOW_INACTIVE_ZONES_TO_RUN as CONF_ALLOW_INACTIVE_ZONES_TO_RUN, CONF_DEFAULT_ZONE_RUN_TIME as CONF_DEFAULT_ZONE_RUN_TIME, CONF_DURATION as CONF_DURATION, CONF_USE_APP_RUN_TIMES as CONF_USE_APP_RUN_TIMES, DATA_PROGRAMS as DATA_PROGRAMS, DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_UNIVERSAL as DATA_RESTRICTIONS_UNIVERSAL, DATA_ZONES as DATA_ZONES, DEFAULT_ZONE_RUN as DEFAULT_ZONE_RUN
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
+from .entity import RainMachineEntity as RainMachineEntity, RainMachineEntityDescription as RainMachineEntityDescription
 from .util import RUN_STATE_MAP as RUN_STATE_MAP, key_exists as key_exists
 from _typeshed import Incomplete
 from collections.abc import Awaitable as Awaitable, Callable as Callable, Coroutine

homeassistant-stubs/components/rainmachine/update.pyi

@@ -1,6 +1,6 @@
-from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineEntity as RainMachineEntity
+from . import RainMachineConfigEntry as RainMachineConfigEntry
 from .const import DATA_MACHINE_FIRMWARE_UPDATE_STATUS as DATA_MACHINE_FIRMWARE_UPDATE_STATUS
-from .model import RainMachineEntityDescription as RainMachineEntityDescription
+from .entity import RainMachineEntity as RainMachineEntity, RainMachineEntityDescription as RainMachineEntityDescription
 from _typeshed import Incomplete
 from enum import Enum
 from homeassistant.components.update import UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityFeature as UpdateEntityFeature

homeassistant-stubs/components/recorder/const.pyi

@@ -22,15 +22,15 @@ ATTR_KEEP_DAYS: str
 ATTR_REPACK: str
 ATTR_APPLY_FILTER: str
 KEEPALIVE_TIME: int
-STATISTICS_ROWS_SCHEMA_VERSION: int
 CONTEXT_ID_AS_BINARY_SCHEMA_VERSION: int
 EVENT_TYPE_IDS_SCHEMA_VERSION: int
 STATES_META_SCHEMA_VERSION: int
 LAST_REPORTED_SCHEMA_VERSION: int
 LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION: int
 INTEGRATION_PLATFORM_COMPILE_STATISTICS: str
-INTEGRATION_PLATFORM_VALIDATE_STATISTICS: str
 INTEGRATION_PLATFORM_LIST_STATISTIC_IDS: str
+INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES: str
+INTEGRATION_PLATFORM_VALIDATE_STATISTICS: str
 INTEGRATION_PLATFORM_METHODS: Incomplete
 
 class SupportedDialect(StrEnum):

homeassistant-stubs/components/recorder/core.pyi

@@ -1,7 +1,7 @@
 import asyncio
 import threading
 from . import migration as migration, statistics as statistics
-from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, DOMAIN as DOMAIN, KEEPALIVE_TIME as KEEPALIVE_TIME, LAST_REPORTED_SCHEMA_VERSION as LAST_REPORTED_SCHEMA_VERSION, MARIADB_PYMYSQL_URL_PREFIX as MARIADB_PYMYSQL_URL_PREFIX, MARIADB_URL_PREFIX as MARIADB_URL_PREFIX, MAX_QUEUE_BACKLOG_MIN_VALUE as MAX_QUEUE_BACKLOG_MIN_VALUE, MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG as MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG, MYSQLDB_PYMYSQL_URL_PREFIX as MYSQLDB_PYMYSQL_URL_PREFIX, MYSQLDB_URL_PREFIX as MYSQLDB_URL_PREFIX, SQLITE_MAX_BIND_VARS as SQLITE_MAX_BIND_VARS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, STATISTICS_ROWS_SCHEMA_VERSION as STATISTICS_ROWS_SCHEMA_VERSION, SupportedDialect as SupportedDialect
+from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, DOMAIN as DOMAIN, KEEPALIVE_TIME as KEEPALIVE_TIME, LAST_REPORTED_SCHEMA_VERSION as LAST_REPORTED_SCHEMA_VERSION, MARIADB_PYMYSQL_URL_PREFIX as MARIADB_PYMYSQL_URL_PREFIX, MARIADB_URL_PREFIX as MARIADB_URL_PREFIX, MAX_QUEUE_BACKLOG_MIN_VALUE as MAX_QUEUE_BACKLOG_MIN_VALUE, MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG as MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG, MYSQLDB_PYMYSQL_URL_PREFIX as MYSQLDB_PYMYSQL_URL_PREFIX, MYSQLDB_URL_PREFIX as MYSQLDB_URL_PREFIX, SQLITE_MAX_BIND_VARS as SQLITE_MAX_BIND_VARS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
 from .db_schema import Base as Base, EventData as EventData, EventTypes as EventTypes, Events as Events, SCHEMA_VERSION as SCHEMA_VERSION, StateAttributes as StateAttributes, States as States, StatesMeta as StatesMeta, Statistics as Statistics, StatisticsShortTerm as StatisticsShortTerm
 from .executor import DBInterruptibleThreadPoolExecutor as DBInterruptibleThreadPoolExecutor
 from .migration import EntityIDMigration as EntityIDMigration, EventIDPostMigration as EventIDPostMigration, EventTypeIDMigration as EventTypeIDMigration, EventsContextIDMigration as EventsContextIDMigration, StatesContextIDMigration as StatesContextIDMigration
@@ -87,7 +87,6 @@ class Recorder(threading.Thread):
     event_session: Incomplete
     _get_session: Incomplete
     _completed_first_database_setup: Incomplete
-    async_migration_event: Incomplete
     migration_in_progress: bool
     migration_is_live: bool
     use_legacy_events_index: bool
@@ -138,8 +137,8 @@ class Recorder(threading.Thread):
     def _adjust_lru_size(self) -> None: ...
     def async_periodic_statistics(self) -> None: ...
     def async_adjust_statistics(self, statistic_id: str, start_time: datetime, sum_adjustment: float, adjustment_unit: str) -> None: ...
-    def async_clear_statistics(self, statistic_ids: list[str]) -> None: ...
-    def async_update_statistics_metadata(self, statistic_id: str, *, new_statistic_id: str | UndefinedType = ..., new_unit_of_measurement: str | None | UndefinedType = ...) -> None: ...
+    def async_clear_statistics(self, statistic_ids: list[str], *, on_done: Callable[[], None] | None = None) -> None: ...
+    def async_update_statistics_metadata(self, statistic_id: str, *, new_statistic_id: str | UndefinedType = ..., new_unit_of_measurement: str | None | UndefinedType = ..., on_done: Callable[[], None] | None = None) -> None: ...
     def async_update_states_metadata(self, entity_id: str, new_entity_id: str) -> None: ...
     def async_change_statistics_unit(self, statistic_id: str, *, new_unit_of_measurement: str, old_unit_of_measurement: str) -> None: ...
     def async_import_statistics(self, metadata: StatisticMetaData, stats: Iterable[StatisticData], table: type[Statistics | StatisticsShortTerm]) -> None: ...
@@ -157,7 +156,6 @@ class Recorder(threading.Thread):
     def _guarded_process_one_task_or_event_or_recover(self, task: RecorderTask | Event) -> None: ...
     def _process_one_task_or_event_or_recover(self, task: RecorderTask | Event) -> None: ...
     def _setup_recorder(self) -> bool: ...
-    def _async_migration_started(self) -> None: ...
     def _migrate_schema_offline(self, schema_status: migration.SchemaValidationStatus) -> tuple[bool, migration.SchemaValidationStatus]: ...
     def _migrate_schema_live(self, schema_status: migration.SchemaValidationStatus) -> tuple[bool, migration.SchemaValidationStatus]: ...
     def _migrate_schema(self, schema_status: migration.SchemaValidationStatus, live: bool) -> tuple[bool, migration.SchemaValidationStatus]: ...
@@ -172,8 +170,6 @@ class Recorder(threading.Thread):
     def _close_event_session(self) -> None: ...
     def _reopen_event_session(self) -> None: ...
     def _open_event_session(self) -> None: ...
-    def _post_schema_migration(self, old_version: int, new_version: int) -> None: ...
-    def _post_migrate_entity_ids(self) -> bool: ...
     def _send_keep_alive(self) -> None: ...
     async def async_block_till_done(self) -> None: ...
     def block_till_done(self) -> None: ...

homeassistant-stubs/components/recorder/history/legacy.pyi

@@ -1,16 +1,15 @@
-from ... import recorder as recorder
 from ..db_schema import RecorderRuns as RecorderRuns, StateAttributes as StateAttributes, States as States
 from ..filters import Filters as Filters
-from ..models import process_datetime_to_timestamp as process_datetime_to_timestamp, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat
-from ..models.legacy import LegacyLazyState as LegacyLazyState, LegacyLazyStatePreSchema31 as LegacyLazyStatePreSchema31, legacy_row_to_compressed_state as legacy_row_to_compressed_state, legacy_row_to_compressed_state_pre_schema_31 as legacy_row_to_compressed_state_pre_schema_31
+from ..models import process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat
+from ..models.legacy import LegacyLazyState as LegacyLazyState, legacy_row_to_compressed_state as legacy_row_to_compressed_state
 from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element, session_scope as session_scope
-from .common import _schema_version as _schema_version
 from .const import LAST_CHANGED_KEY as LAST_CHANGED_KEY, NEED_ATTRIBUTE_DOMAINS as NEED_ATTRIBUTE_DOMAINS, SIGNIFICANT_DOMAINS as SIGNIFICANT_DOMAINS, SIGNIFICANT_DOMAINS_ENTITY_ID_LIKE as SIGNIFICANT_DOMAINS_ENTITY_ID_LIKE, STATE_KEY as STATE_KEY
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable
 from datetime import datetime
 from homeassistant.const import COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, split_entity_id as split_entity_id
+from homeassistant.helpers.recorder import get_instance as get_instance
 from sqlalchemy import Column as Column
 from sqlalchemy.engine.row import Row as Row
 from sqlalchemy.orm.session import Session as Session
@@ -34,16 +33,16 @@ _QUERY_STATES_NO_LAST_CHANGED: Incomplete
 _FIELD_MAP: Incomplete
 _FIELD_MAP_PRE_SCHEMA_31: Incomplete
 
-def _lambda_stmt_and_join_attributes(schema_version: int, no_attributes: bool, include_last_changed: bool = True) -> tuple[StatementLambdaElement, bool]: ...
+def _lambda_stmt_and_join_attributes(no_attributes: bool, include_last_changed: bool = True) -> tuple[StatementLambdaElement, bool]: ...
 def get_significant_states(hass: HomeAssistant, start_time: datetime, end_time: datetime | None = None, entity_ids: list[str] | None = None, filters: Filters | None = None, include_start_time_state: bool = True, significant_changes_only: bool = True, minimal_response: bool = False, no_attributes: bool = False, compressed_state_format: bool = False) -> dict[str, list[State | dict[str, Any]]]: ...
-def _significant_states_stmt(schema_version: int, start_time: datetime, end_time: datetime | None, entity_ids: list[str], significant_changes_only: bool, no_attributes: bool) -> StatementLambdaElement: ...
+def _significant_states_stmt(start_time: datetime, end_time: datetime | None, entity_ids: list[str], significant_changes_only: bool, no_attributes: bool) -> StatementLambdaElement: ...
 def get_significant_states_with_session(hass: HomeAssistant, session: Session, start_time: datetime, end_time: datetime | None = None, entity_ids: list[str] | None = None, filters: Filters | None = None, include_start_time_state: bool = True, significant_changes_only: bool = True, minimal_response: bool = False, no_attributes: bool = False, compressed_state_format: bool = False) -> dict[str, list[State | dict[str, Any]]]: ...
 def get_full_significant_states_with_session(hass: HomeAssistant, session: Session, start_time: datetime, end_time: datetime | None = None, entity_ids: list[str] | None = None, filters: Filters | None = None, include_start_time_state: bool = True, significant_changes_only: bool = True, no_attributes: bool = False) -> dict[str, list[State]]: ...
-def _state_changed_during_period_stmt(schema_version: int, start_time: datetime, end_time: datetime | None, entity_id: str, no_attributes: bool, descending: bool, limit: int | None) -> StatementLambdaElement: ...
+def _state_changed_during_period_stmt(start_time: datetime, end_time: datetime | None, entity_id: str, no_attributes: bool, descending: bool, limit: int | None) -> StatementLambdaElement: ...
 def state_changes_during_period(hass: HomeAssistant, start_time: datetime, end_time: datetime | None = None, entity_id: str | None = None, no_attributes: bool = False, descending: bool = False, limit: int | None = None, include_start_time_state: bool = True) -> dict[str, list[State]]: ...
-def _get_last_state_changes_stmt(schema_version: int, number_of_states: int, entity_id: str) -> StatementLambdaElement: ...
+def _get_last_state_changes_stmt(number_of_states: int, entity_id: str) -> StatementLambdaElement: ...
 def get_last_state_changes(hass: HomeAssistant, number_of_states: int, entity_id: str) -> dict[str, list[State]]: ...
-def _get_states_for_entities_stmt(schema_version: int, run_start: datetime, utc_point_in_time: datetime, entity_ids: list[str], no_attributes: bool) -> StatementLambdaElement: ...
+def _get_states_for_entities_stmt(run_start: datetime, utc_point_in_time: datetime, entity_ids: list[str], no_attributes: bool) -> StatementLambdaElement: ...
 def _get_rows_with_session(hass: HomeAssistant, session: Session, utc_point_in_time: datetime, entity_ids: list[str], run: RecorderRuns | None = None, no_attributes: bool = False) -> Iterable[Row]: ...
-def _get_single_entity_states_stmt(schema_version: int, utc_point_in_time: datetime, entity_id: str, no_attributes: bool = False) -> StatementLambdaElement: ...
+def _get_single_entity_states_stmt(utc_point_in_time: datetime, entity_id: str, no_attributes: bool = False) -> StatementLambdaElement: ...
 def _sorted_states_to_dict(hass: HomeAssistant, session: Session, states: Iterable[Row], start_time: datetime, entity_ids: list[str], include_start_time_state: bool = True, minimal_response: bool = False, no_attributes: bool = False, compressed_state_format: bool = False) -> dict[str, list[State | dict[str, Any]]]: ...

homeassistant-stubs/components/recorder/history/modern.pyi

@@ -1,4 +1,3 @@
-from ... import recorder as recorder
 from ..const import LAST_REPORTED_SCHEMA_VERSION as LAST_REPORTED_SCHEMA_VERSION
 from ..db_schema import SHARED_ATTR_OR_LEGACY_ATTRIBUTES as SHARED_ATTR_OR_LEGACY_ATTRIBUTES, StateAttributes as StateAttributes, States as States
 from ..filters import Filters as Filters
@@ -10,6 +9,7 @@ from collections.abc import Callable as Callable, Iterable
 from datetime import datetime
 from homeassistant.const import COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, split_entity_id as split_entity_id
+from homeassistant.helpers.recorder import get_instance as get_instance
 from sqlalchemy import CompoundSelect as CompoundSelect, Select as Select, Subquery as Subquery
 from sqlalchemy.engine.row import Row as Row
 from sqlalchemy.orm.session import Session as Session

homeassistant-stubs/components/recorder/migration.pyi

@@ -6,9 +6,9 @@ from .db_schema import BIG_INTEGER_SQL as BIG_INTEGER_SQL, Base as Base, CONTEXT
 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, 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 .statistics import cleanup_statistics_timestamp_migration as cleanup_statistics_timestamp_migration, get_start_time as get_start_time
+from .tasks import RecorderTask as RecorderTask
+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, retryable_database_job_method as retryable_database_job_method, session_scope as session_scope
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
 from collections.abc import Callable as Callable, Iterable
@@ -24,6 +24,8 @@ from typing import Any
 
 LIVE_MIGRATION_MIN_SCHEMA_VERSION: int
 MIGRATION_NOTE_OFFLINE: str
+MIGRATION_NOTE_MINUTES: str
+MIGRATION_NOTE_WHILE: str
 _EMPTY_ENTITY_ID: str
 _EMPTY_EVENT_TYPE: str
 _LOGGER: Incomplete
@@ -233,8 +235,6 @@ class _SchemaVersion47Migrator(_SchemaVersionMigrator, target_version=47):
 
 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: ...
-def _wipe_old_string_statistics_columns(instance: Recorder) -> None: ...
 def _wipe_old_string_time_columns(instance: Recorder, engine: Engine, session: Session) -> None: ...
 def _migrate_columns_to_timestamp(instance: Recorder, session_maker: Callable[[], Session], engine: Engine) -> None: ...
 def _migrate_statistics_columns_to_timestamp_one_by_one(instance: Recorder, session_maker: Callable[[], Session]) -> None: ...
@@ -242,7 +242,6 @@ def _migrate_statistics_columns_to_timestamp(instance: Recorder, session_maker:
 def _context_id_to_bytes(context_id: str | None) -> bytes | None: ...
 def _generate_ulid_bytes_at_time(timestamp: float | None) -> bytes: ...
 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: ...
 
@@ -259,12 +258,13 @@ class CommitBeforeMigrationTask(MigrationTask):
     def __init__(self, migrator) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
-class NeedsMigrateResult:
+class DataMigrationStatus:
     needs_migrate: bool
     migration_done: bool
     def __init__(self, *, needs_migrate, migration_done) -> None: ...
 
 class BaseRunTimeMigration(ABC, metaclass=abc.ABCMeta):
+    index_to_drop: tuple[str, str] | None
     required_schema_version: int
     migration_version: int
     migration_id: str
@@ -273,60 +273,63 @@ class BaseRunTimeMigration(ABC, metaclass=abc.ABCMeta):
     migration_changes: Incomplete
     def __init__(self, schema_version: int, migration_changes: dict[str, int]) -> None: ...
     def do_migrate(self, instance: Recorder, session: Session) -> None: ...
-    @staticmethod
+    def migrate_data(self, instance: Recorder) -> bool: ...
     @abstractmethod
-    def migrate_data(instance: Recorder) -> bool: ...
-    def migration_done(self, instance: Recorder, session: Session | None) -> None: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
+    def migration_done(self, instance: Recorder, session: Session) -> None: ...
     @abstractmethod
-    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> DataMigrationStatus: ...
     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_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> DataMigrationStatus: ...
 
 class StatesContextIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = CONTEXT_ID_AS_BINARY_SCHEMA_VERSION
     migration_id: str
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
+    migration_version: int
+    index_to_drop: Incomplete
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EventsContextIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = CONTEXT_ID_AS_BINARY_SCHEMA_VERSION
     migration_id: str
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
+    migration_version: int
+    index_to_drop: Incomplete
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EventTypeIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = EVENT_TYPE_IDS_SCHEMA_VERSION
     migration_id: str
     task = CommitBeforeMigrationTask
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
-    def migration_done(self, instance: Recorder, session: Session | None) -> None: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
+    def migration_done(self, instance: Recorder, session: Session) -> None: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EntityIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = STATES_META_SCHEMA_VERSION
     migration_id: str
     task = CommitBeforeMigrationTask
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
-    def migration_done(self, instance: Recorder, _session: Session | None) -> None: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
+    def migration_done(self, instance: Recorder, session: Session) -> None: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EventIDPostMigration(BaseRunTimeMigration):
     migration_id: str
     task = MigrationTask
     migration_version: int
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
     @staticmethod
     def _legacy_event_id_foreign_key_exists(instance: Recorder) -> bool: ...
-    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> DataMigrationStatus: ...
+
+@dataclass(slots=True)
+class EntityIDPostMigrationTask(RecorderTask):
+    def run(self, instance: Recorder) -> None: ...
 
 def _mark_migration_done(session: Session, migration: type[BaseRunTimeMigration]) -> None: ...
 def rebuild_sqlite_table(session_maker: Callable[[], Session], engine: Engine, table: type[Base]) -> bool: ...

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

@@ -3,6 +3,6 @@ from .database import DatabaseEngine as DatabaseEngine, DatabaseOptimizer as Dat
 from .event import extract_event_type_ids as extract_event_type_ids
 from .state import LazyState as LazyState, extract_metadata_ids as extract_metadata_ids, row_to_compressed_state as row_to_compressed_state
 from .statistics import CalendarStatisticPeriod as CalendarStatisticPeriod, FixedStatisticPeriod as FixedStatisticPeriod, RollingWindowStatisticPeriod as RollingWindowStatisticPeriod, StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMetaData as StatisticMetaData, StatisticPeriod as StatisticPeriod, StatisticResult as StatisticResult
-from .time import datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_datetime_to_timestamp as process_datetime_to_timestamp, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat, timestamp_to_datetime_or_none as timestamp_to_datetime_or_none
+from .time import datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat, timestamp_to_datetime_or_none as timestamp_to_datetime_or_none
 
-__all__ = ['CalendarStatisticPeriod', 'DatabaseEngine', 'DatabaseOptimizer', 'FixedStatisticPeriod', 'LazyState', 'RollingWindowStatisticPeriod', 'StatisticData', 'StatisticDataTimestamp', 'StatisticMetaData', 'StatisticPeriod', 'StatisticResult', 'UnsupportedDialect', 'bytes_to_ulid_or_none', 'bytes_to_uuid_hex_or_none', 'datetime_to_timestamp_or_none', 'extract_event_type_ids', 'extract_metadata_ids', 'process_datetime_to_timestamp', 'process_timestamp', 'process_timestamp_to_utc_isoformat', 'row_to_compressed_state', 'timestamp_to_datetime_or_none', 'ulid_to_bytes_or_none', 'uuid_hex_to_bytes_or_none']
+__all__ = ['CalendarStatisticPeriod', 'DatabaseEngine', 'DatabaseOptimizer', 'FixedStatisticPeriod', 'LazyState', 'RollingWindowStatisticPeriod', 'StatisticData', 'StatisticDataTimestamp', 'StatisticMetaData', 'StatisticPeriod', 'StatisticResult', 'UnsupportedDialect', 'bytes_to_ulid_or_none', 'bytes_to_uuid_hex_or_none', 'datetime_to_timestamp_or_none', 'extract_event_type_ids', 'extract_metadata_ids', 'process_timestamp', 'process_timestamp_to_utc_isoformat', 'row_to_compressed_state', 'timestamp_to_datetime_or_none', 'ulid_to_bytes_or_none', 'uuid_hex_to_bytes_or_none']

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

@@ -1,5 +1,5 @@
 from .state_attributes import decode_attributes_from_source as decode_attributes_from_source
-from .time import process_datetime_to_timestamp as process_datetime_to_timestamp, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat
+from .time import process_timestamp as process_timestamp
 from _typeshed import Incomplete
 from datetime import datetime
 from homeassistant.const import COMPRESSED_STATE_ATTRIBUTES as COMPRESSED_STATE_ATTRIBUTES, COMPRESSED_STATE_LAST_CHANGED as COMPRESSED_STATE_LAST_CHANGED, COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
@@ -7,42 +7,6 @@ from homeassistant.core import Context as Context, State as State
 from sqlalchemy.engine.row import Row as Row
 from typing import Any
 
-class LegacyLazyStatePreSchema31(State):
-    __slots__: Incomplete
-    _row: Incomplete
-    entity_id: Incomplete
-    state: Incomplete
-    _attributes: Incomplete
-    _last_changed: Incomplete
-    _last_reported: Incomplete
-    _last_updated: Incomplete
-    _context: Incomplete
-    attr_cache: Incomplete
-    def __init__(self, row: Row, attr_cache: dict[str, dict[str, Any]], start_time: datetime | None) -> None: ...
-    @property
-    def attributes(self) -> dict[str, Any]: ...
-    @attributes.setter
-    def attributes(self, value: dict[str, Any]) -> None: ...
-    @property
-    def context(self) -> Context: ...
-    @context.setter
-    def context(self, value: Context) -> None: ...
-    @property
-    def last_changed(self) -> datetime: ...
-    @last_changed.setter
-    def last_changed(self, value: datetime) -> None: ...
-    @property
-    def last_reported(self) -> datetime: ...
-    @last_reported.setter
-    def last_reported(self, value: datetime) -> None: ...
-    @property
-    def last_updated(self) -> datetime: ...
-    @last_updated.setter
-    def last_updated(self, value: datetime) -> None: ...
-    def as_dict(self) -> dict[str, Any]: ...
-
-def legacy_row_to_compressed_state_pre_schema_31(row: Row, attr_cache: dict[str, dict[str, Any]], start_time: datetime | None) -> dict[str, Any]: ...
-
 class LegacyLazyState(State):
     __slots__: Incomplete
     _row: Incomplete

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

@@ -14,6 +14,5 @@ def process_timestamp(ts: datetime) -> datetime: ...
 def process_timestamp_to_utc_isoformat(ts: None) -> None: ...
 @overload
 def process_timestamp_to_utc_isoformat(ts: datetime) -> str: ...
-def process_datetime_to_timestamp(ts: datetime) -> float: ...
 def datetime_to_timestamp_or_none(dt: datetime | None) -> float | None: ...
 def timestamp_to_datetime_or_none(ts: float | None) -> datetime | None: ...

homeassistant-stubs/components/recorder/statistics.pyi

@@ -1,6 +1,6 @@
 import dataclasses
 from . import Recorder as Recorder
-from .const import DOMAIN as DOMAIN, EVENT_RECORDER_5MIN_STATISTICS_GENERATED as EVENT_RECORDER_5MIN_STATISTICS_GENERATED, EVENT_RECORDER_HOURLY_STATISTICS_GENERATED as EVENT_RECORDER_HOURLY_STATISTICS_GENERATED, INTEGRATION_PLATFORM_COMPILE_STATISTICS as INTEGRATION_PLATFORM_COMPILE_STATISTICS, INTEGRATION_PLATFORM_LIST_STATISTIC_IDS as INTEGRATION_PLATFORM_LIST_STATISTIC_IDS, INTEGRATION_PLATFORM_VALIDATE_STATISTICS as INTEGRATION_PLATFORM_VALIDATE_STATISTICS, SupportedDialect as SupportedDialect
+from .const import DOMAIN as DOMAIN, EVENT_RECORDER_5MIN_STATISTICS_GENERATED as EVENT_RECORDER_5MIN_STATISTICS_GENERATED, EVENT_RECORDER_HOURLY_STATISTICS_GENERATED as EVENT_RECORDER_HOURLY_STATISTICS_GENERATED, INTEGRATION_PLATFORM_COMPILE_STATISTICS as INTEGRATION_PLATFORM_COMPILE_STATISTICS, INTEGRATION_PLATFORM_LIST_STATISTIC_IDS as INTEGRATION_PLATFORM_LIST_STATISTIC_IDS, INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES as INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES, INTEGRATION_PLATFORM_VALIDATE_STATISTICS as INTEGRATION_PLATFORM_VALIDATE_STATISTICS, SupportedDialect as SupportedDialect
 from .db_schema import STATISTICS_TABLES as STATISTICS_TABLES, Statistics as Statistics, StatisticsBase as StatisticsBase, StatisticsRuns as StatisticsRuns, StatisticsShortTerm as StatisticsShortTerm
 from .models import StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMetaData as StatisticMetaData, StatisticResult as StatisticResult, datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp
 from .util import execute as execute, execute_stmt_lambda_element as execute_stmt_lambda_element, filter_unique_constraint_integrity_error as filter_unique_constraint_integrity_error, get_instance as get_instance, retryable_database_job as retryable_database_job, session_scope as session_scope
@@ -138,6 +138,7 @@ def _build_stats(db_rows: list[Row], table_duration_seconds: float, start_ts_idx
 def _build_converted_stats(db_rows: list[Row], table_duration_seconds: float, start_ts_idx: int, row_mapping: tuple[tuple[str, int], ...], convert: Callable[[float | None], float | None] | Callable[[float], float]) -> list[StatisticsRow]: ...
 def _sorted_statistics_to_dict(hass: HomeAssistant, stats: Sequence[Row[Any]], statistic_ids: set[str] | None, _metadata: dict[str, tuple[int, StatisticMetaData]], convert_units: bool, table: type[StatisticsBase], units: dict[str, str] | None, types: set[Literal['last_reset', 'max', 'mean', 'min', 'state', 'sum']]) -> dict[str, list[StatisticsRow]]: ...
 def validate_statistics(hass: HomeAssistant) -> dict[str, list[ValidationIssue]]: ...
+def update_statistics_issues(hass: HomeAssistant) -> None: ...
 def _statistics_exists(session: Session, table: type[StatisticsBase], metadata_id: int, start: datetime) -> int | None: ...
 def _async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...
 def async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...

homeassistant-stubs/components/recorder/tasks.pyi

@@ -33,17 +33,19 @@ class ChangeStatisticsUnitTask(RecorderTask):
 
 @dataclass(slots=True)
 class ClearStatisticsTask(RecorderTask):
+    on_done: Callable[[], None] | None
     statistic_ids: list[str]
     def run(self, instance: Recorder) -> None: ...
-    def __init__(self, statistic_ids) -> None: ...
+    def __init__(self, on_done, statistic_ids) -> None: ...
 
 @dataclass(slots=True)
 class UpdateStatisticsMetadataTask(RecorderTask):
+    on_done: Callable[[], None] | None
     statistic_id: str
     new_statistic_id: str | None | UndefinedType
     new_unit_of_measurement: str | None | UndefinedType
     def run(self, instance: Recorder) -> None: ...
-    def __init__(self, statistic_id, new_statistic_id, new_unit_of_measurement) -> None: ...
+    def __init__(self, on_done, statistic_id, new_statistic_id, new_unit_of_measurement) -> None: ...
 
 @dataclass(slots=True)
 class UpdateStatesMetadataTask(RecorderTask):
@@ -141,26 +143,11 @@ class SynchronizeTask(RecorderTask):
     def run(self, instance: Recorder) -> None: ...
     def __init__(self, event) -> None: ...
 
-@dataclass(slots=True)
-class PostSchemaMigrationTask(RecorderTask):
-    old_version: int
-    new_version: int
-    def run(self, instance: Recorder) -> None: ...
-    def __init__(self, old_version, new_version) -> None: ...
-
-@dataclass(slots=True)
-class StatisticsTimestampMigrationCleanupTask(RecorderTask):
-    def run(self, instance: Recorder) -> None: ...
-
 @dataclass(slots=True)
 class AdjustLRUSizeTask(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 RefreshEventTypesTask(RecorderTask):
     event_types: list[EventType[Any] | str]

homeassistant-stubs/components/recorder/util.pyi

@@ -74,9 +74,13 @@ def end_incomplete_runs(session: Session, start_time: datetime) -> None: ...
 def _is_retryable_error(instance: Recorder, err: OperationalError) -> bool: ...
 
 _FuncType: Incomplete
+_MethType: Incomplete
+_FuncOrMethType: Incomplete
 
-def retryable_database_job(description: str) -> Callable[[_FuncType[_RecorderT, _P, bool]], _FuncType[_RecorderT, _P, bool]]: ...
-def database_job_retry_wrapper(description: str, attempts: int = 5) -> Callable[[_FuncType[_RecorderT, _P, None]], _FuncType[_RecorderT, _P, None]]: ...
+def retryable_database_job(description: str) -> Callable[[_FuncType[_P, bool]], _FuncType[_P, bool]]: ...
+def retryable_database_job_method(description: str) -> Callable[[_MethType[_Self, _P, bool]], _MethType[_Self, _P, bool]]: ...
+def _wrap_func_or_meth(job: _FuncOrMethType[_P, bool], description: str, method: bool) -> _FuncOrMethType[_P, bool]: ...
+def database_job_retry_wrapper(description: str, attempts: int = 5) -> Callable[[_FuncType[_P, None]], _FuncType[_P, None]]: ...
 def periodic_db_cleanups(instance: Recorder) -> None: ...
 def write_lock_db_sqlite(instance: Recorder) -> Generator[None]: ...
 def async_migration_in_progress(hass: HomeAssistant) -> bool: ...

homeassistant-stubs/components/recorder/websocket_api.pyi

@@ -1,5 +1,5 @@
 from .models import StatisticPeriod as StatisticPeriod
-from .statistics import STATISTIC_UNIT_TO_UNIT_CONVERTER as STATISTIC_UNIT_TO_UNIT_CONVERTER, async_add_external_statistics as async_add_external_statistics, async_change_statistics_unit as async_change_statistics_unit, async_import_statistics as async_import_statistics, async_list_statistic_ids as async_list_statistic_ids, list_statistic_ids as list_statistic_ids, statistic_during_period as statistic_during_period, statistics_during_period as statistics_during_period, validate_statistics as validate_statistics
+from .statistics import STATISTIC_UNIT_TO_UNIT_CONVERTER as STATISTIC_UNIT_TO_UNIT_CONVERTER, async_add_external_statistics as async_add_external_statistics, async_change_statistics_unit as async_change_statistics_unit, async_import_statistics as async_import_statistics, async_list_statistic_ids as async_list_statistic_ids, list_statistic_ids as list_statistic_ids, statistic_during_period as statistic_during_period, statistics_during_period as statistics_during_period, update_statistics_issues as update_statistics_issues, validate_statistics as validate_statistics
 from .util import PERIOD_SCHEMA as PERIOD_SCHEMA, get_instance as get_instance, resolve_period as resolve_period
 from _typeshed import Incomplete
 from datetime import datetime as dt
@@ -11,6 +11,8 @@ from homeassistant.helpers.json import json_bytes as json_bytes
 from homeassistant.util.unit_conversion import ConductivityConverter as ConductivityConverter, DataRateConverter as DataRateConverter, DistanceConverter as DistanceConverter, DurationConverter as DurationConverter, ElectricCurrentConverter as ElectricCurrentConverter, ElectricPotentialConverter as ElectricPotentialConverter, EnergyConverter as EnergyConverter, InformationConverter as InformationConverter, MassConverter as MassConverter, PowerConverter as PowerConverter, PressureConverter as PressureConverter, SpeedConverter as SpeedConverter, TemperatureConverter as TemperatureConverter, UnitlessRatioConverter as UnitlessRatioConverter, VolumeConverter as VolumeConverter, VolumeFlowRateConverter as VolumeFlowRateConverter
 from typing import Any, Literal
 
+CLEAR_STATISTICS_TIME_OUT: int
+UPDATE_STATISTICS_METADATA_TIME_OUT: int
 UNIT_SCHEMA: Incomplete
 
 def async_setup(hass: HomeAssistant) -> None: ...
@@ -23,9 +25,10 @@ def _ws_get_list_statistic_ids(hass: HomeAssistant, msg_id: int, statistic_type:
 async def ws_handle_list_statistic_ids(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
 async def ws_list_statistic_ids(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def ws_validate_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
-def ws_clear_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def ws_update_statistics_issues(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def ws_clear_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def ws_get_statistics_metadata(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
-def ws_update_statistics_metadata(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def ws_update_statistics_metadata(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 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: ...

homeassistant-stubs/components/remote/__init__.pyi

@@ -10,10 +10,12 @@ from homeassistant.helpers.entity import ToggleEntity as ToggleEntity, ToggleEnt
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any
 
 _LOGGER: Incomplete
 DOMAIN: str
+DATA_COMPONENT: HassKey[EntityComponent[RemoteEntity]]
 ENTITY_ID_FORMAT: Incomplete
 PLATFORM_SCHEMA: Incomplete
 PLATFORM_SCHEMA_BASE: Incomplete