homeassistant-stubs 2024.7.3__py3-none-any.whl → 2024.8.0__py3-none-any.whl

homeassistant-stubs/components/rainmachine/binary_sensor.pyi

@@ -1,11 +1,10 @@
-from . import RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
-from .const import DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_CURRENT as DATA_RESTRICTIONS_CURRENT, DOMAIN as DOMAIN
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineEntity as RainMachineEntity
+from .const import DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_CURRENT as DATA_RESTRICTIONS_CURRENT
 from .model import 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
 from homeassistant.components.binary_sensor import BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -21,11 +20,11 @@ TYPE_WEEKDAY: str
 @dataclass(frozen=True, kw_only=True)
 class RainMachineBinarySensorDescription(BinarySensorEntityDescription, RainMachineEntityDescription):
     data_key: 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, data_key) -> 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=..., api_category, data_key) -> None: ...
 
 BINARY_SENSOR_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class CurrentRestrictionsBinarySensor(RainMachineEntity, BinarySensorEntity):
     entity_description: RainMachineBinarySensorDescription

homeassistant-stubs/components/rainmachine/button.pyi

@@ -1,11 +1,10 @@
-from . import RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
-from .const import DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DOMAIN as DOMAIN
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineEntity as RainMachineEntity
+from .const import DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS
 from .model import RainMachineEntityDescription as RainMachineEntityDescription
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -16,7 +15,7 @@ from regenmaschine.controller import Controller as Controller
 @dataclass(frozen=True, kw_only=True)
 class RainMachineButtonDescription(ButtonEntityDescription, RainMachineEntityDescription):
     push_action: Callable[[Controller], Awaitable]
-    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, push_action) -> 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=..., api_category, push_action) -> None: ...
 
 BUTTON_KIND_REBOOT: str
 
@@ -24,7 +23,7 @@ async def _async_reboot(controller: Controller) -> None: ...
 
 BUTTON_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RainMachineButton(RainMachineEntity, ButtonEntity):
     _attr_device_class: Incomplete

homeassistant-stubs/components/rainmachine/coordinator.pyi

@@ -1,8 +1,8 @@
+from . import RainMachineConfigEntry as RainMachineConfigEntry
 from .const import LOGGER as LOGGER
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
 from datetime import timedelta
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
@@ -12,11 +12,11 @@ SIGNAL_REBOOT_COMPLETED: str
 SIGNAL_REBOOT_REQUESTED: str
 
 class RainMachineDataUpdateCoordinator(DataUpdateCoordinator[dict]):
-    config_entry: ConfigEntry
+    config_entry: RainMachineConfigEntry
     _rebooting: bool
     _signal_handler_unsubs: Incomplete
     signal_reboot_completed: Incomplete
     signal_reboot_requested: Incomplete
-    def __init__(self, hass: HomeAssistant, *, entry: ConfigEntry, name: str, api_category: str, update_interval: timedelta, update_method: Callable[[], Coroutine[Any, Any, dict]]) -> None: ...
+    def __init__(self, hass: HomeAssistant, *, entry: RainMachineConfigEntry, name: str, api_category: str, update_interval: timedelta, update_method: Callable[[], Coroutine[Any, Any, dict]]) -> None: ...
     last_update_success: bool
     def async_initialize(self) -> None: ...

homeassistant-stubs/components/rainmachine/diagnostics.pyi

@@ -1,8 +1,7 @@
-from . import RainMachineData as RainMachineData
-from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
+from . import RainMachineConfigEntry as RainMachineConfigEntry
+from .const import LOGGER as LOGGER
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ELEVATION as CONF_ELEVATION, CONF_LATITUDE as CONF_LATITUDE, CONF_LONGITUDE as CONF_LONGITUDE, CONF_PASSWORD as CONF_PASSWORD, CONF_UNIQUE_ID as CONF_UNIQUE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
@@ -13,4 +12,4 @@ CONF_STATION_SOURCE: str
 CONF_TIMEZONE: str
 TO_REDACT: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: RainMachineConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/rainmachine/model.pyi

@@ -4,4 +4,4 @@ from homeassistant.helpers.entity import EntityDescription as EntityDescription
 @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: ...
+    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: ...

homeassistant-stubs/components/rainmachine/select.pyi

@@ -1,5 +1,5 @@
-from . import RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
-from .const import DATA_RESTRICTIONS_UNIVERSAL as DATA_RESTRICTIONS_UNIVERSAL, DOMAIN as DOMAIN
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
+from .const import DATA_RESTRICTIONS_UNIVERSAL as DATA_RESTRICTIONS_UNIVERSAL
 from .model import RainMachineEntityDescription as RainMachineEntityDescription
 from .util import key_exists as key_exists
 from _typeshed import Incomplete
@@ -15,7 +15,7 @@ from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM as US_CUSTOMARY_S
 @dataclass(frozen=True, kw_only=True)
 class RainMachineSelectDescription(SelectEntityDescription, RainMachineEntityDescription):
     data_key: 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, options, data_key) -> 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=..., api_category, options=..., data_key) -> None: ...
 
 @dataclass
 class FreezeProtectionSelectOption:
@@ -27,12 +27,12 @@ class FreezeProtectionSelectOption:
 @dataclass(frozen=True, kw_only=True)
 class FreezeProtectionSelectDescription(RainMachineSelectDescription):
     extended_options: list[FreezeProtectionSelectOption]
-    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, options, data_key, extended_options) -> 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=..., api_category, options=..., data_key, extended_options) -> None: ...
 
 TYPE_FREEZE_PROTECTION_TEMPERATURE: str
 SELECT_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class FreezeProtectionTemperatureSelect(RainMachineEntity, SelectEntity):
     entity_description: FreezeProtectionSelectDescription

homeassistant-stubs/components/rainmachine/sensor.pyi

@@ -1,5 +1,5 @@
-from . import RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
-from .const import DATA_PROGRAMS as DATA_PROGRAMS, DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_ZONES as DATA_ZONES, DOMAIN as DOMAIN
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
+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 .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
@@ -27,16 +27,16 @@ TYPE_ZONE_RUN_COMPLETION_TIME: str
 @dataclass(frozen=True, kw_only=True)
 class RainMachineSensorDataDescription(SensorEntityDescription, RainMachineEntityDescription):
     data_key: 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, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, data_key) -> 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=..., api_category, last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., data_key) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class RainMachineSensorCompletionTimerDescription(SensorEntityDescription, RainMachineEntityDescription):
     uid: int
-    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, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, uid) -> 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=..., api_category, last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., uid) -> None: ...
 
 SENSOR_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TimeRemainingSensor(RainMachineEntity, RestoreSensor):
     entity_description: RainMachineSensorCompletionTimerDescription

homeassistant-stubs/components/rainmachine/switch.pyi

@@ -1,5 +1,5 @@
-from . import RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity, 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, DOMAIN as DOMAIN
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity, 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 .util import RUN_STATE_MAP as RUN_STATE_MAP, key_exists as key_exists
 from _typeshed import Incomplete
@@ -47,24 +47,24 @@ def raise_on_request_error(func: Callable[Concatenate[_T, _P], Awaitable[None]])
 
 @dataclass(frozen=True, kw_only=True)
 class RainMachineSwitchDescription(SwitchEntityDescription, RainMachineEntityDescription):
-    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: ...
+    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: ...
 
 @dataclass(frozen=True, kw_only=True)
 class RainMachineActivitySwitchDescription(RainMachineSwitchDescription):
     kind: str
     uid: int
-    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, kind, uid) -> 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=..., api_category, kind, uid) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class RainMachineRestrictionSwitchDescription(RainMachineSwitchDescription):
     data_key: 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, data_key) -> 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=..., api_category, data_key) -> None: ...
 
 TYPE_RESTRICTIONS_FREEZE_PROTECT_ENABLED: str
 TYPE_RESTRICTIONS_HOT_DAYS_EXTRA_WATERING: str
 RESTRICTIONS_SWITCH_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RainMachineBaseSwitch(RainMachineEntity, SwitchEntity):
     entity_description: RainMachineSwitchDescription

homeassistant-stubs/components/rainmachine/update.pyi

@@ -1,27 +1,26 @@
-from . import RainMachineData as RainMachineData, RainMachineEntity as RainMachineEntity
-from .const import DATA_MACHINE_FIRMWARE_UPDATE_STATUS as DATA_MACHINE_FIRMWARE_UPDATE_STATUS, DOMAIN as DOMAIN
+from . import RainMachineConfigEntry as RainMachineConfigEntry, RainMachineEntity as RainMachineEntity
+from .const import DATA_MACHINE_FIRMWARE_UPDATE_STATUS as DATA_MACHINE_FIRMWARE_UPDATE_STATUS
 from .model import 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
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
 
 class UpdateStates(Enum):
-    IDLE: int
-    CHECKING: int
-    DOWNLOADING: int
-    UPGRADING: int
-    ERROR: int
-    REBOOT: int
+    IDLE = 1
+    CHECKING = 2
+    DOWNLOADING = 3
+    UPGRADING = 4
+    ERROR = 5
+    REBOOT = 6
 
 UPDATE_STATE_MAP: Incomplete
 UPDATE_DESCRIPTION: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RainMachineUpdateEntity(RainMachineEntity, UpdateEntity):
     _attr_device_class: Incomplete

homeassistant-stubs/components/rainmachine/util.pyi

@@ -8,9 +8,9 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from typing import Any
 
 class RunStates(StrEnum):
-    NOT_RUNNING: str
-    QUEUED: str
-    RUNNING: str
+    NOT_RUNNING = 'Not Running'
+    QUEUED = 'Queued'
+    RUNNING = 'Running'
 
 RUN_STATE_MAP: Incomplete
 
@@ -21,7 +21,7 @@ class EntityDomainReplacementStrategy:
     replacement_entity_id: str
     breaks_in_ha_version: str
     remove_old_entity: bool = ...
-    def __init__(self, old_domain, old_unique_id, replacement_entity_id, breaks_in_ha_version, remove_old_entity) -> None: ...
+    def __init__(self, old_domain, old_unique_id, replacement_entity_id, breaks_in_ha_version, remove_old_entity=...) -> None: ...
 
 def async_finish_entity_domain_replacements(hass: HomeAssistant, entry: ConfigEntry, entity_replacement_strategies: Iterable[EntityDomainReplacementStrategy]) -> None: ...
 def key_exists(data: dict[str, Any], search_key: str) -> bool: ...

homeassistant-stubs/components/rdw/binary_sensor.pyi

@@ -13,7 +13,7 @@ from vehicle import Vehicle
 @dataclass(frozen=True, kw_only=True)
 class RDWBinarySensorEntityDescription(BinarySensorEntityDescription):
     is_on_fn: Callable[[Vehicle], bool | 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, is_on_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=..., is_on_fn) -> None: ...
 
 BINARY_SENSORS: tuple[RDWBinarySensorEntityDescription, ...]
 

homeassistant-stubs/components/rdw/sensor.pyi

@@ -14,7 +14,7 @@ from vehicle import Vehicle
 @dataclass(frozen=True, kw_only=True)
 class RDWSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[Vehicle], date | str | float | 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, value_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=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 SENSORS: tuple[RDWSensorEntityDescription, ...]
 

homeassistant-stubs/components/recorder/__init__.pyi

@@ -1,5 +1,5 @@
 from . import entity_registry as entity_registry, websocket_api as websocket_api
-from .const import CONF_DB_INTEGRITY_CHECK as CONF_DB_INTEGRITY_CHECK, DATA_INSTANCE as DATA_INSTANCE, DOMAIN as DOMAIN, INTEGRATION_PLATFORMS_LOAD_IN_RECORDER_THREAD as INTEGRATION_PLATFORMS_LOAD_IN_RECORDER_THREAD, INTEGRATION_PLATFORM_COMPILE_STATISTICS as INTEGRATION_PLATFORM_COMPILE_STATISTICS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
+from .const import CONF_DB_INTEGRITY_CHECK as CONF_DB_INTEGRITY_CHECK, DOMAIN as DOMAIN, INTEGRATION_PLATFORM_COMPILE_STATISTICS as INTEGRATION_PLATFORM_COMPILE_STATISTICS, INTEGRATION_PLATFORM_METHODS as INTEGRATION_PLATFORM_METHODS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
 from .core import Recorder as Recorder
 from .services import async_register_services as async_register_services
 from .tasks import AddRecorderPlatformTask as AddRecorderPlatformTask
@@ -9,6 +9,7 @@ from homeassistant.const import CONF_EXCLUDE as CONF_EXCLUDE, EVENT_RECORDER_5MI
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entityfilter import INCLUDE_EXCLUDE_BASE_FILTER_SCHEMA as INCLUDE_EXCLUDE_BASE_FILTER_SCHEMA, INCLUDE_EXCLUDE_FILTER_SCHEMA_INNER as INCLUDE_EXCLUDE_FILTER_SCHEMA_INNER, convert_include_exclude_filter as convert_include_exclude_filter
 from homeassistant.helpers.integration_platform import async_process_integration_platforms as async_process_integration_platforms
+from homeassistant.helpers.recorder import DATA_INSTANCE as DATA_INSTANCE
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.event_type import EventType as EventType

homeassistant-stubs/components/recorder/basic_websocket_api.pyi

@@ -0,0 +1,7 @@
+from .util import get_instance as get_instance
+from homeassistant.components import websocket_api as websocket_api
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from typing import Any
+
+def async_setup(hass: HomeAssistant) -> None: ...
+def ws_info(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/recorder/const.pyi

@@ -3,9 +3,7 @@ from _typeshed import Incomplete
 from enum import StrEnum
 from homeassistant.const import ATTR_ATTRIBUTION as ATTR_ATTRIBUTION, ATTR_RESTORED as ATTR_RESTORED, ATTR_SUPPORTED_FEATURES as ATTR_SUPPORTED_FEATURES, EVENT_RECORDER_5MIN_STATISTICS_GENERATED as EVENT_RECORDER_5MIN_STATISTICS_GENERATED, EVENT_RECORDER_HOURLY_STATISTICS_GENERATED as EVENT_RECORDER_HOURLY_STATISTICS_GENERATED
 from homeassistant.helpers.json import JSON_DUMP as JSON_DUMP
-from homeassistant.util.hass_dict import HassKey as HassKey
 
-DATA_INSTANCE: HassKey[Recorder]
 SQLITE_URL_PREFIX: str
 MARIADB_URL_PREFIX: str
 MARIADB_PYMYSQL_URL_PREFIX: str
@@ -14,8 +12,7 @@ MYSQLDB_PYMYSQL_URL_PREFIX: str
 DOMAIN: str
 CONF_DB_INTEGRITY_CHECK: str
 MAX_QUEUE_BACKLOG_MIN_VALUE: int
-ESTIMATED_QUEUE_ITEM_SIZE: int
-QUEUE_PERCENTAGE_ALLOWED_AVAILABLE_MEMORY: float
+MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG: Incomplete
 SQLITE_MAX_BIND_VARS: int
 SQLITE_MODERN_MAX_BIND_VARS: int
 DEFAULT_MAX_BIND_VARS: int
@@ -34,9 +31,9 @@ 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_PLATFORMS_LOAD_IN_RECORDER_THREAD: Incomplete
+INTEGRATION_PLATFORM_METHODS: Incomplete
 
 class SupportedDialect(StrEnum):
-    SQLITE: str
-    MYSQL: str
-    POSTGRESQL: str
+    SQLITE = 'sqlite'
+    MYSQL = 'mysql'
+    POSTGRESQL = 'postgresql'

homeassistant-stubs/components/recorder/core.pyi

@@ -1,10 +1,10 @@
 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, ESTIMATED_QUEUE_ITEM_SIZE as ESTIMATED_QUEUE_ITEM_SIZE, KEEPALIVE_TIME as KEEPALIVE_TIME, LAST_REPORTED_SCHEMA_VERSION as LAST_REPORTED_SCHEMA_VERSION, LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION as LEGACY_STATES_EVENT_ID_INDEX_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, MYSQLDB_PYMYSQL_URL_PREFIX as MYSQLDB_PYMYSQL_URL_PREFIX, MYSQLDB_URL_PREFIX as MYSQLDB_URL_PREFIX, QUEUE_PERCENTAGE_ALLOWED_AVAILABLE_MEMORY as QUEUE_PERCENTAGE_ALLOWED_AVAILABLE_MEMORY, 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 .db_schema import Base as Base, EventData as EventData, 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, SCHEMA_VERSION as SCHEMA_VERSION, StateAttributes as StateAttributes, States as States, StatesMeta as StatesMeta, Statistics as Statistics, StatisticsShortTerm as StatisticsShortTerm, TABLE_STATES as TABLE_STATES
+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 .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 BaseRunTimeMigration as BaseRunTimeMigration, EntityIDMigration as EntityIDMigration, EventTypeIDMigration as EventTypeIDMigration, EventsContextIDMigration as EventsContextIDMigration, StatesContextIDMigration as StatesContextIDMigration
+from .migration import EntityIDMigration as EntityIDMigration, EventIDPostMigration as EventIDPostMigration, EventTypeIDMigration as EventTypeIDMigration, EventsContextIDMigration as EventsContextIDMigration, StatesContextIDMigration as StatesContextIDMigration
 from .models import DatabaseEngine as DatabaseEngine, StatisticData as StatisticData, StatisticMetaData as StatisticMetaData, UnsupportedDialect as UnsupportedDialect
 from .pool import MutexPool as MutexPool, POOL_SIZE as POOL_SIZE, RecorderPool as RecorderPool
 from .queries import get_migration_changes as get_migration_changes
@@ -15,8 +15,8 @@ from .table_managers.state_attributes import StateAttributesManager as StateAttr
 from .table_managers.states import StatesManager as StatesManager
 from .table_managers.states_meta import StatesMetaManager as StatesMetaManager
 from .table_managers.statistics_meta import StatisticsMetaManager as StatisticsMetaManager
-from .tasks import AdjustLRUSizeTask as AdjustLRUSizeTask, AdjustStatisticsTask as AdjustStatisticsTask, ChangeStatisticsUnitTask as ChangeStatisticsUnitTask, ClearStatisticsTask as ClearStatisticsTask, CommitTask as CommitTask, CompileMissingStatisticsTask as CompileMissingStatisticsTask, DatabaseLockTask as DatabaseLockTask, EntityIDPostMigrationTask as EntityIDPostMigrationTask, EventIdMigrationTask as EventIdMigrationTask, ImportStatisticsTask as ImportStatisticsTask, KeepAliveTask as KeepAliveTask, PerodicCleanupTask as PerodicCleanupTask, PurgeTask as PurgeTask, RecorderTask as RecorderTask, StatisticsTask as StatisticsTask, StopTask as StopTask, SynchronizeTask as SynchronizeTask, UpdateStatesMetadataTask as UpdateStatesMetadataTask, UpdateStatisticsMetadataTask as UpdateStatisticsMetadataTask, WaitTask as WaitTask
-from .util import async_create_backup_failure_issue as async_create_backup_failure_issue, build_mysqldb_conv as build_mysqldb_conv, dburl_to_path as dburl_to_path, end_incomplete_runs as end_incomplete_runs, execute_stmt_lambda_element as execute_stmt_lambda_element, get_index_by_name as get_index_by_name, is_second_sunday as is_second_sunday, move_away_broken_database as move_away_broken_database, session_scope as session_scope, setup_connection_for_dialect as setup_connection_for_dialect, validate_or_move_away_sqlite_database as validate_or_move_away_sqlite_database, write_lock_db_sqlite as write_lock_db_sqlite
+from .tasks import AdjustLRUSizeTask as AdjustLRUSizeTask, AdjustStatisticsTask as AdjustStatisticsTask, ChangeStatisticsUnitTask as ChangeStatisticsUnitTask, ClearStatisticsTask as ClearStatisticsTask, CommitTask as CommitTask, CompileMissingStatisticsTask as CompileMissingStatisticsTask, DatabaseLockTask as DatabaseLockTask, ImportStatisticsTask as ImportStatisticsTask, KeepAliveTask as KeepAliveTask, PerodicCleanupTask as PerodicCleanupTask, PurgeTask as PurgeTask, RecorderTask as RecorderTask, StatisticsTask as StatisticsTask, StopTask as StopTask, SynchronizeTask as SynchronizeTask, UpdateStatesMetadataTask as UpdateStatesMetadataTask, UpdateStatisticsMetadataTask as UpdateStatisticsMetadataTask, WaitTask as WaitTask
+from .util import async_create_backup_failure_issue as async_create_backup_failure_issue, build_mysqldb_conv as build_mysqldb_conv, dburl_to_path as dburl_to_path, end_incomplete_runs as end_incomplete_runs, execute_stmt_lambda_element as execute_stmt_lambda_element, is_second_sunday as is_second_sunday, move_away_broken_database as move_away_broken_database, session_scope as session_scope, setup_connection_for_dialect as setup_connection_for_dialect, validate_or_move_away_sqlite_database as validate_or_move_away_sqlite_database, write_lock_db_sqlite as write_lock_db_sqlite
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable
 from datetime import datetime
@@ -44,6 +44,7 @@ WAIT_TASK: Incomplete
 ADJUST_LRU_SIZE_TASK: Incomplete
 DB_LOCK_TIMEOUT: int
 DB_LOCK_QUEUE_CHECK_TIMEOUT: int
+QUEUE_CHECK_INTERVAL: Incomplete
 INVALIDATED_ERR: str
 CONNECTIVITY_ERR: str
 MAX_DB_EXECUTOR_WORKERS: Incomplete
@@ -122,14 +123,14 @@ class Recorder(threading.Thread):
     def _stop_executor(self) -> None: ...
     def _async_check_queue(self, *_: Any) -> None: ...
     def _available_memory(self) -> int: ...
-    def _reached_max_backlog_percentage(self, percentage: int) -> bool: ...
+    def _reached_max_backlog(self) -> bool: ...
     def _async_stop_queue_watcher_and_event_listener(self) -> None: ...
     def _async_stop_listeners(self) -> None: ...
     async def _async_close(self, event: Event) -> None: ...
     async def _async_shutdown(self, event: Event) -> None: ...
     def _async_hass_started(self, hass: HomeAssistant) -> None: ...
     def async_register(self) -> None: ...
-    def _async_startup_failed(self) -> None: ...
+    def _async_startup_done(self, startup_failed: bool) -> None: ...
     def async_connection_success(self) -> None: ...
     def async_set_db_ready(self) -> None: ...
     def _async_set_recorder_ready_migration_done(self) -> None: ...
@@ -148,33 +149,32 @@ class Recorder(threading.Thread):
     def _wait_startup_or_shutdown(self) -> object | None: ...
     def run(self) -> None: ...
     def _add_to_session(self, session: Session, obj: object) -> None: ...
+    def _notify_migration_failed(self) -> None: ...
+    def _dismiss_migration_in_progress(self) -> None: ...
     def _run(self) -> None: ...
-    def _activate_and_set_db_ready(self) -> None: ...
+    def _activate_and_set_db_ready(self, schema_status: migration.SchemaValidationStatus) -> None: ...
     def _run_event_loop(self) -> None: ...
     def _pre_process_startup_events(self, startup_task_or_events: list[RecorderTask | Event[Any]]) -> None: ...
     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_and_setup_run(self, schema_status: migration.SchemaValidationStatus) -> bool: ...
+    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]: ...
     def _lock_database(self, task: DatabaseLockTask) -> None: ...
     def _process_one_event(self, event: Event[Any]) -> None: ...
     def _process_non_state_changed_event_into_session(self, event: Event) -> None: ...
     def _process_state_changed_event_into_session(self, event: Event[EventStateChangedData]) -> None: ...
-    def _handle_database_error(self, err: Exception) -> bool: ...
+    def _handle_database_error(self, err: Exception, *, setup_run: bool) -> bool: ...
     def _commit_event_session_or_retry(self) -> None: ...
     def _commit_event_session(self) -> None: ...
-    def _handle_sqlite_corruption(self) -> None: ...
+    def _handle_sqlite_corruption(self, setup_run: bool) -> None: ...
     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 _migrate_states_context_ids(self) -> bool: ...
-    def _migrate_events_context_ids(self) -> bool: ...
-    def _migrate_event_type_ids(self) -> bool: ...
-    def _migrate_entity_ids(self) -> bool: ...
     def _post_migrate_entity_ids(self) -> bool: ...
-    def _cleanup_legacy_states_event_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/db_schema.pyi

@@ -16,6 +16,7 @@ from sqlalchemy.types import TypeDecorator as TypeDecorator
 from typing import Any, Self
 
 class Base(DeclarativeBase): ...
+class LegacyBase(DeclarativeBase): ...
 
 SCHEMA_VERSION: int
 _LOGGER: Incomplete
@@ -58,11 +59,12 @@ def compile_char_zero(type_: TypeDecorator, compiler: Any, **kw: Any) -> str: ..
 def compile_char_one(type_: TypeDecorator, compiler: Any, **kw: Any) -> str: ...
 
 class FAST_PYSQLITE_DATETIME(sqlite.DATETIME):
-    def result_processor(self, dialect, coltype): ...
+    def result_processor(self, dialect: Dialect, coltype: Any) -> Callable | None: ...
 
 class NativeLargeBinary(LargeBinary):
-    def result_processor(self, dialect, coltype) -> None: ...
+    def result_processor(self, dialect: Dialect, coltype: Any) -> Callable | None: ...
 
+ID_TYPE: Incomplete
 UINT_32_TYPE: Incomplete
 JSON_VARIANT_CAST: Incomplete
 JSONB_VARIANT_CAST: Incomplete
@@ -72,6 +74,7 @@ UNUSED_LEGACY_COLUMN: Incomplete
 UNUSED_LEGACY_DATETIME_COLUMN: Incomplete
 UNUSED_LEGACY_INTEGER_COLUMN: Incomplete
 DOUBLE_PRECISION_TYPE_SQL: str
+BIG_INTEGER_SQL: str
 CONTEXT_BINARY_TYPE: Incomplete
 TIMESTAMP_TYPE = DOUBLE_TYPE
 
@@ -79,7 +82,6 @@ class JSONLiteral(JSON):
     def literal_processor(self, dialect: Dialect) -> Callable[[Any], str]: ...
 
 EVENT_ORIGIN_ORDER: Incomplete
-EVENT_ORIGIN_TO_IDX: Incomplete
 
 class Events(Base):
     __table_args__: Incomplete
@@ -206,12 +208,18 @@ class Statistics(Base, StatisticsBase):
     __table_args__: Incomplete
     __tablename__ = TABLE_STATISTICS
 
-class StatisticsShortTerm(Base, StatisticsBase):
+class _StatisticsShortTerm(StatisticsBase):
     duration: Incomplete
-    __table_args__: Incomplete
     __tablename__ = TABLE_STATISTICS_SHORT_TERM
 
-class StatisticsMeta(Base):
+class StatisticsShortTerm(Base, _StatisticsShortTerm):
+    __table_args__: Incomplete
+
+class LegacyStatisticsShortTerm(LegacyBase, _StatisticsShortTerm):
+    __table_args__: Incomplete
+    metadata_id: Mapped[int | None]
+
+class _StatisticsMeta:
     __table_args__: Incomplete
     __tablename__ = TABLE_STATISTICS_META
     id: Mapped[int]
@@ -224,6 +232,11 @@ class StatisticsMeta(Base):
     @staticmethod
     def from_meta(meta: StatisticMetaData) -> StatisticsMeta: ...
 
+class StatisticsMeta(Base, _StatisticsMeta): ...
+
+class LegacyStatisticsMeta(LegacyBase, _StatisticsMeta):
+    id: Mapped[int]
+
 class RecorderRuns(Base):
     __table_args__: Incomplete
     __tablename__ = TABLE_RECORDER_RUNS