homeassistant-stubs 2025.9.0b0__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/components/portainer/sensor.pyi

@@ -0,0 +1,40 @@
+from .coordinator import PortainerConfigEntry as PortainerConfigEntry, PortainerContainerData as PortainerContainerData, PortainerCoordinator as PortainerCoordinator
+from .entity import PortainerContainerEntity as PortainerContainerEntity, PortainerCoordinatorData as PortainerCoordinatorData, PortainerEndpointEntity as PortainerEndpointEntity
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.sensor import EntityCategory as EntityCategory, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass, StateType as StateType
+from homeassistant.const import PERCENTAGE as PERCENTAGE, UnitOfInformation as UnitOfInformation
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+
+@dataclass(frozen=True, kw_only=True)
+class PortainerContainerSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[PortainerContainerData], StateType]
+
+@dataclass(frozen=True, kw_only=True)
+class PortainerEndpointSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[PortainerCoordinatorData], StateType]
+
+CONTAINER_SENSORS: tuple[PortainerContainerSensorEntityDescription, ...]
+ENDPOINT_SENSORS: tuple[PortainerEndpointSensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PortainerConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class PortainerContainerSensor(PortainerContainerEntity, SensorEntity):
+    entity_description: PortainerContainerSensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: PortainerCoordinator, entity_description: PortainerContainerSensorEntityDescription, device_info: PortainerContainerData, via_device: PortainerCoordinatorData) -> None: ...
+    @property
+    def available(self) -> bool: ...
+    @property
+    def native_value(self) -> StateType: ...
+
+class PortainerEndpointSensor(PortainerEndpointEntity, SensorEntity):
+    entity_description: PortainerEndpointSensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: PortainerCoordinator, entity_description: PortainerEndpointSensorEntityDescription, device_info: PortainerCoordinatorData) -> None: ...
+    @property
+    def available(self) -> bool: ...
+    @property
+    def native_value(self) -> StateType: ...

homeassistant-stubs/components/portainer/switch.pyi

@@ -0,0 +1,34 @@
+from . import PortainerConfigEntry as PortainerConfigEntry
+from .const import DOMAIN as DOMAIN
+from .coordinator import PortainerContainerData as PortainerContainerData, PortainerCoordinator as PortainerCoordinator
+from .entity import PortainerContainerEntity as PortainerContainerEntity, PortainerCoordinatorData as PortainerCoordinatorData
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable, Coroutine
+from dataclasses import dataclass
+from homeassistant.components.switch import SwitchDeviceClass as SwitchDeviceClass, SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from pyportainer import Portainer as Portainer
+from typing import Any
+
+@dataclass(frozen=True, kw_only=True)
+class PortainerSwitchEntityDescription(SwitchEntityDescription):
+    is_on_fn: Callable[[PortainerContainerData], bool | None]
+    turn_on_fn: Callable[[str, Portainer, int, str], Coroutine[Any, Any, None]]
+    turn_off_fn: Callable[[str, Portainer, int, str], Coroutine[Any, Any, None]]
+
+async def perform_action(action: str, portainer: Portainer, endpoint_id: int, container_id: str) -> None: ...
+
+SWITCHES: tuple[PortainerSwitchEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PortainerConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class PortainerContainerSwitch(PortainerContainerEntity, SwitchEntity):
+    entity_description: PortainerSwitchEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: PortainerCoordinator, entity_description: PortainerSwitchEntityDescription, device_info: PortainerContainerData, via_device: PortainerCoordinatorData) -> None: ...
+    @property
+    def is_on(self) -> bool | None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/prometheus/__init__.pyi

@@ -1,6 +1,6 @@
 from _typeshed import Incomplete
 from aiohttp import web
-from collections.abc import Callable as Callable
+from collections.abc import Callable as Callable, Sequence
 from dataclasses import dataclass
 from homeassistant.components.alarm_control_panel import AlarmControlPanelState as AlarmControlPanelState
 from homeassistant.components.climate import ATTR_CURRENT_TEMPERATURE as ATTR_CURRENT_TEMPERATURE, ATTR_FAN_MODE as ATTR_FAN_MODE, ATTR_FAN_MODES as ATTR_FAN_MODES, ATTR_HVAC_ACTION as ATTR_HVAC_ACTION, ATTR_HVAC_MODES as ATTR_HVAC_MODES, ATTR_TARGET_TEMP_HIGH as ATTR_TARGET_TEMP_HIGH, ATTR_TARGET_TEMP_LOW as ATTR_TARGET_TEMP_LOW, HVACAction as HVACAction
@@ -12,9 +12,12 @@ from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass
 from homeassistant.const import ATTR_BATTERY_LEVEL as ATTR_BATTERY_LEVEL, ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, ATTR_MODE as ATTR_MODE, ATTR_TEMPERATURE as ATTR_TEMPERATURE, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, CONTENT_TYPE_TEXT_PLAIN as CONTENT_TYPE_TEXT_PLAIN, EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, PERCENTAGE as PERCENTAGE, STATE_CLOSED as STATE_CLOSED, STATE_CLOSING as STATE_CLOSING, STATE_ON as STATE_ON, STATE_OPEN as STATE_OPEN, STATE_OPENING as STATE_OPENING, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, State as State
-from homeassistant.helpers import entityfilter as entityfilter
+from homeassistant.helpers import area_registry as ar, device_registry as dr, entity_registry as er, entityfilter as entityfilter, floor_registry as fr
+from homeassistant.helpers.area_registry import AreaEntry as AreaEntry, EVENT_AREA_REGISTRY_UPDATED as EVENT_AREA_REGISTRY_UPDATED, EventAreaRegistryUpdatedData as EventAreaRegistryUpdatedData
+from homeassistant.helpers.device_registry import EVENT_DEVICE_REGISTRY_UPDATED as EVENT_DEVICE_REGISTRY_UPDATED, EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
 from homeassistant.helpers.entity_registry import EVENT_ENTITY_REGISTRY_UPDATED as EVENT_ENTITY_REGISTRY_UPDATED, EventEntityRegistryUpdatedData as EventEntityRegistryUpdatedData
 from homeassistant.helpers.entity_values import EntityValues as EntityValues
+from homeassistant.helpers.floor_registry import EVENT_FLOOR_REGISTRY_UPDATED as EVENT_FLOOR_REGISTRY_UPDATED, EventFloorRegistryUpdatedData as EventFloorRegistryUpdatedData, FloorEntry as FloorEntry
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.util.dt import as_timestamp as as_timestamp
 from homeassistant.util.unit_conversion import TemperatureConverter as TemperatureConverter
@@ -55,10 +58,21 @@ class PrometheusMetrics:
     _metrics: dict[str, MetricWrapperBase]
     _metrics_by_entity_id: dict[str, set[MetricNameWithLabelValues]]
     _climate_units: Incomplete
-    def __init__(self, entity_filter: entityfilter.EntityFilter, namespace: str, climate_units: UnitOfTemperature, component_config: EntityValues, override_metric: str | None, default_metric: str | None) -> None: ...
+    _area_info_metrics: dict[str, MetricNameWithLabelValues]
+    _floor_info_metrics: dict[str, MetricNameWithLabelValues]
+    area_registry: Incomplete
+    device_registry: Incomplete
+    entity_registry: Incomplete
+    floor_registry: Incomplete
+    def __init__(self, entity_filter: entityfilter.EntityFilter, namespace: str, climate_units: UnitOfTemperature, component_config: EntityValues, override_metric: str | None, default_metric: str | None, area_registry: ar.AreaRegistry, device_registry: dr.DeviceRegistry, entity_registry: er.EntityRegistry, floor_registry: fr.FloorRegistry) -> None: ...
     def handle_state_changed_event(self, event: Event[EventStateChangedData]) -> None: ...
     def handle_state(self, state: State) -> None: ...
     def handle_entity_registry_updated(self, event: Event[EventEntityRegistryUpdatedData]) -> None: ...
+    def handle_device_registry_updated(self, event: Event[EventDeviceRegistryUpdatedData]) -> None: ...
+    def handle_area_registry_updated(self, event: Event[EventAreaRegistryUpdatedData]) -> None: ...
+    def handle_area(self, area: AreaEntry) -> None: ...
+    def handle_floor_registry_updated(self, event: Event[EventFloorRegistryUpdatedData]) -> None: ...
+    def handle_floor(self, floor: FloorEntry) -> None: ...
     def _remove_labelsets(self, entity_id: str, ignored_metric_names: set[str] | None = None) -> None: ...
     def _handle_attributes(self, state: State) -> None: ...
     def _metric[_MetricBaseT: MetricWrapperBase](self, metric_name: str, factory: type[_MetricBaseT], documentation: str, labels: dict[str, str]) -> _MetricBaseT: ...
@@ -68,20 +82,34 @@ class PrometheusMetrics:
     def state_as_number(state: State) -> float | None: ...
     @staticmethod
     def _labels(state: State, extra_labels: dict[str, str] | None = None) -> dict[str, Any]: ...
-    def _battery(self, state: State) -> None: ...
+    def _remove_entity_info(self, entity_id: str) -> None: ...
+    def _add_entity_info(self, entity_id: str, area_id: str) -> None: ...
+    def _find_area_id(self, entity_id: str) -> str | None: ...
+    def _battery_metric(self, state: State) -> None: ...
+    def _temperature_metric(self, state: State, attr: str, metric_name: str, metric_description: str) -> None: ...
+    def _bool_metric(self, state: State, attr: str, metric_name: str, metric_description: str, true_values: set[Any] | None = None) -> None: ...
+    def _float_metric(self, state: State, attr: str, metric_name: str, metric_description: str) -> None: ...
+    def _enum_metric(self, state: State, current_value: Any | None, values: Sequence[str] | None, metric_name: str, metric_description: str, enum_label_name: str) -> None: ...
+    def _numeric_metric(self, state: State, domain: str, title: str) -> None: ...
     def _handle_binary_sensor(self, state: State) -> None: ...
     def _handle_input_boolean(self, state: State) -> None: ...
-    def _numeric_handler(self, state: State, domain: str, title: str) -> None: ...
     def _handle_input_number(self, state: State) -> None: ...
     def _handle_number(self, state: State) -> None: ...
     def _handle_device_tracker(self, state: State) -> None: ...
     def _handle_person(self, state: State) -> None: ...
+    def _handle_lock(self, state: State) -> None: ...
     def _handle_cover(self, state: State) -> None: ...
     def _handle_light(self, state: State) -> None: ...
-    def _handle_lock(self, state: State) -> None: ...
-    def _handle_climate_temp(self, state: State, attr: str, metric_name: str, metric_description: str) -> None: ...
     def _handle_climate(self, state: State) -> None: ...
     def _handle_humidifier(self, state: State) -> None: ...
+    def _handle_water_heater(self, state: State) -> None: ...
+    def _handle_switch(self, state: State) -> None: ...
+    def _handle_fan(self, state: State) -> None: ...
+    def _handle_zwave(self, state: State) -> None: ...
+    def _handle_automation(self, state: State) -> None: ...
+    def _handle_counter(self, state: State) -> None: ...
+    def _handle_update(self, state: State) -> None: ...
+    def _handle_alarm_control_panel(self, state: State) -> None: ...
     def _handle_sensor(self, state: State) -> None: ...
     def _sensor_default_metric(self, state: State, unit: str | None) -> str | None: ...
     @staticmethod
@@ -94,13 +122,6 @@ class PrometheusMetrics:
     def _sensor_fallback_metric(state: State, unit: str | None) -> str | None: ...
     @staticmethod
     def _unit_string(unit: str | None) -> str | None: ...
-    def _handle_switch(self, state: State) -> None: ...
-    def _handle_fan(self, state: State) -> None: ...
-    def _handle_zwave(self, state: State) -> None: ...
-    def _handle_automation(self, state: State) -> None: ...
-    def _handle_counter(self, state: State) -> None: ...
-    def _handle_update(self, state: State) -> None: ...
-    def _handle_alarm_control_panel(self, state: State) -> None: ...
 
 class PrometheusView(HomeAssistantView):
     url = API_ENDPOINT

homeassistant-stubs/components/prusalink/coordinator.pyi

@@ -5,11 +5,13 @@ from abc import ABC, abstractmethod
 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.debounce import Debouncer as Debouncer
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 from pyprusalink import JobInfo, LegacyPrinterStatus, PrinterInfo, PrinterStatus, PrusaLink as PrusaLink
 from typing import TypeVar
 
 _LOGGER: Incomplete
+_MINIMUM_REFRESH_INTERVAL: float
 T = TypeVar('T', PrinterStatus, LegacyPrinterStatus, JobInfo)
 
 class PrusaLinkUpdateCoordinator(DataUpdateCoordinator[T], ABC, metaclass=abc.ABCMeta):

homeassistant-stubs/components/pyload/config_flow.pyi

@@ -12,6 +12,7 @@ from typing import Any
 _LOGGER: Incomplete
 STEP_USER_DATA_SCHEMA: Incomplete
 REAUTH_SCHEMA: Incomplete
+PLACEHOLDER: Incomplete
 
 async def validate_input(hass: HomeAssistant, user_input: dict[str, Any]) -> None: ...
 

homeassistant-stubs/components/qbus/entity.pyi

@@ -9,10 +9,8 @@ from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo, form
 from homeassistant.helpers.entity import Entity as Entity
 from qbusmqttapi.discovery import QbusMqttDevice as QbusMqttDevice, QbusMqttOutput as QbusMqttOutput
 from qbusmqttapi.state import QbusMqttState
-from typing import Generic, TypeVar
 
 _REFID_REGEX: Incomplete
-StateT = TypeVar('StateT', bound=QbusMqttState)
 
 def create_new_entities(coordinator: QbusControllerCoordinator, added_outputs: list[QbusMqttOutput], filter_fn: Callable[[QbusMqttOutput], bool], entity_type: type[QbusEntity]) -> list[QbusEntity]: ...
 def determine_new_outputs(coordinator: QbusControllerCoordinator, added_outputs: list[QbusMqttOutput], filter_fn: Callable[[QbusMqttOutput], bool]) -> list[QbusMqttOutput]: ...
@@ -20,7 +18,7 @@ def format_ref_id(ref_id: str) -> str | None: ...
 def create_device_identifier(mqtt_device: QbusMqttDevice) -> tuple[str, str]: ...
 def create_unique_id(serial_number: str, suffix: str) -> str: ...
 
-class QbusEntity(Entity, ABC, Generic[StateT], metaclass=abc.ABCMeta):
+class QbusEntity[StateT: QbusMqttState](Entity, ABC, metaclass=abc.ABCMeta):
     _state_cls: type[StateT]
     _attr_has_entity_name: bool
     _attr_should_poll: bool

homeassistant-stubs/components/radio_browser/media_source.pyi

@@ -4,6 +4,7 @@ from _typeshed import Incomplete
 from homeassistant.components.media_player import MediaClass as MediaClass, MediaType as MediaType
 from homeassistant.components.media_source import BrowseMediaSource as BrowseMediaSource, MediaSource as MediaSource, MediaSourceItem as MediaSourceItem, PlayMedia as PlayMedia, Unresolvable as Unresolvable
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.util.location import vincenty as vincenty
 from radios import RadioBrowser as RadioBrowser, Station as Station
 
 CODEC_TO_MIMETYPE: Incomplete
@@ -28,3 +29,5 @@ class RadioMediaSource(MediaSource):
     async def _async_build_by_language(self, radios: RadioBrowser, item: MediaSourceItem) -> list[BrowseMediaSource]: ...
     async def _async_build_popular(self, radios: RadioBrowser, item: MediaSourceItem) -> list[BrowseMediaSource]: ...
     async def _async_build_by_tag(self, radios: RadioBrowser, item: MediaSourceItem) -> list[BrowseMediaSource]: ...
+    def _filter_local_stations(self, stations: list[Station], latitude: float, longitude: float) -> list[Station]: ...
+    async def _async_build_local(self, radios: RadioBrowser, item: MediaSourceItem) -> list[BrowseMediaSource]: ...

homeassistant-stubs/components/rainmachine/__init__.pyi

@@ -14,6 +14,7 @@ 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
 
+API_URL_REFERENCE: str
 DEFAULT_SSL: bool
 PLATFORMS: Incomplete
 CONF_CONDITION: str

homeassistant-stubs/components/raspberry_pi/hardware.pyi

@@ -1,6 +1,6 @@
 from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.components.hardware.models import BoardInfo as BoardInfo, HardwareInfo as HardwareInfo
+from homeassistant.components.hardware import BoardInfo as BoardInfo, HardwareInfo as HardwareInfo
 from homeassistant.components.hassio import get_os_info as get_os_info
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError

homeassistant-stubs/components/recorder/auto_repairs/schema.pyi

@@ -1,6 +1,6 @@
 from .. import Recorder as Recorder
 from ..const import SupportedDialect as SupportedDialect
-from ..db_schema import DOUBLE_PRECISION_TYPE_SQL as DOUBLE_PRECISION_TYPE_SQL, DOUBLE_TYPE as DOUBLE_TYPE
+from ..db_schema import DOUBLE_PRECISION_TYPE_SQL as DOUBLE_PRECISION_TYPE_SQL, DOUBLE_TYPE as DOUBLE_TYPE, MYSQL_COLLATE as MYSQL_COLLATE
 from ..util import session_scope as session_scope
 from _typeshed import Incomplete
 from collections.abc import Iterable, Mapping

homeassistant-stubs/components/recorder/const.pyi

@@ -24,8 +24,8 @@ KEEPALIVE_TIME: int
 CONTEXT_ID_AS_BINARY_SCHEMA_VERSION: int
 EVENT_TYPE_IDS_SCHEMA_VERSION: int
 STATES_META_SCHEMA_VERSION: int
-LAST_REPORTED_SCHEMA_VERSION: int
 CIRCULAR_MEAN_SCHEMA_VERSION: int
+UNIT_CLASS_SCHEMA_VERSION: int
 LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION: int
 LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION: int
 INTEGRATION_PLATFORM_COMPILE_STATISTICS: str

homeassistant-stubs/components/recorder/core.pyi

@@ -3,10 +3,10 @@ import psutil_home_assistant as ha_psutil
 import queue
 import threading
 from . import migration as migration, statistics as statistics
-from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, DEFAULT_MAX_BIND_VARS as DEFAULT_MAX_BIND_VARS, 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_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
+from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, DEFAULT_MAX_BIND_VARS as DEFAULT_MAX_BIND_VARS, DOMAIN as DOMAIN, KEEPALIVE_TIME as KEEPALIVE_TIME, 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_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 .models import DatabaseEngine as DatabaseEngine, StatisticData as StatisticData, StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData, UnsupportedDialect as UnsupportedDialect
+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 .table_managers.event_data import EventDataManager as EventDataManager
 from .table_managers.event_types import EventTypeManager as EventTypeManager
@@ -158,7 +158,7 @@ class Recorder(threading.Thread):
     @callback
     def async_clear_statistics(self, statistic_ids: list[str], *, on_done: Callable[[], None] | None = None) -> None: ...
     @callback
-    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_statistics_metadata(self, statistic_id: str, *, new_statistic_id: str | UndefinedType = ..., new_unit_class: str | None | UndefinedType = ..., new_unit_of_measurement: str | None | UndefinedType = ..., on_done: Callable[[], None] | None = None) -> None: ...
     @callback
     def async_update_states_metadata(self, entity_id: str, new_entity_id: str) -> None: ...
     @callback

homeassistant-stubs/components/recorder/db_schema.pyi

@@ -1,13 +1,12 @@
 from .const import ALL_DOMAIN_EXCLUDE_ATTRS as ALL_DOMAIN_EXCLUDE_ATTRS, SupportedDialect as SupportedDialect
-from .models import StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData, bytes_to_ulid_or_none as bytes_to_ulid_or_none, bytes_to_uuid_hex_or_none as bytes_to_uuid_hex_or_none, datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp, ulid_to_bytes_or_none as ulid_to_bytes_or_none, uuid_hex_to_bytes_or_none as uuid_hex_to_bytes_or_none
+from .models import StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData, datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp, ulid_to_bytes_or_none as ulid_to_bytes_or_none, uuid_hex_to_bytes_or_none as uuid_hex_to_bytes_or_none
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from datetime import datetime, timedelta
 from homeassistant.components.sensor import ATTR_STATE_CLASS as ATTR_STATE_CLASS
 from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, MATCH_ALL as MATCH_ALL, MAX_LENGTH_EVENT_EVENT_TYPE as MAX_LENGTH_EVENT_EVENT_TYPE, MAX_LENGTH_STATE_ENTITY_ID as MAX_LENGTH_STATE_ENTITY_ID, MAX_LENGTH_STATE_STATE as MAX_LENGTH_STATE_STATE
-from homeassistant.core import Context as Context, Event as Event, EventOrigin as EventOrigin, EventStateChangedData as EventStateChangedData, State as State
+from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData
 from homeassistant.helpers.json import JSON_DUMP as JSON_DUMP, json_bytes as json_bytes, json_bytes_strip_null as json_bytes_strip_null
-from homeassistant.util.json import JSON_DECODE_EXCEPTIONS as JSON_DECODE_EXCEPTIONS, json_loads as json_loads, json_loads_object as json_loads_object
 from sqlalchemy import CHAR, ColumnElement as ColumnElement, DateTime, JSON, LargeBinary
 from sqlalchemy.dialects import sqlite
 from sqlalchemy.engine.interfaces import Dialect as Dialect
@@ -85,8 +84,6 @@ class _LiteralProcessorType(Protocol):
 class JSONLiteral(JSON):
     def literal_processor(self, dialect: Dialect) -> _LiteralProcessorType: ...
 
-EVENT_ORIGIN_ORDER: Incomplete
-
 class Events(Base):
     __table_args__: Incomplete
     __tablename__ = TABLE_EVENTS
@@ -112,7 +109,6 @@ class Events(Base):
     def _time_fired_isotime(self) -> str | None: ...
     @staticmethod
     def from_event(event: Event) -> Events: ...
-    def to_native(self, validate_entity_id: bool = True) -> Event | None: ...
 
 class LegacyEvents(LegacyBase):
     __table_args__: Incomplete
@@ -131,7 +127,6 @@ class EventData(Base):
     def shared_data_bytes_from_event(event: Event, dialect: SupportedDialect | None) -> bytes: ...
     @staticmethod
     def hash_shared_data_bytes(shared_data_bytes: bytes) -> int: ...
-    def to_native(self) -> dict[str, Any]: ...
 
 class EventTypes(Base):
     __table_args__: Incomplete
@@ -171,7 +166,6 @@ class States(Base):
     def _last_updated_isotime(self) -> str | None: ...
     @staticmethod
     def from_event(event: Event[EventStateChangedData]) -> States: ...
-    def to_native(self, validate_entity_id: bool = True) -> State | None: ...
 
 class LegacyStates(LegacyBase):
     __table_args__: Incomplete
@@ -192,7 +186,6 @@ class StateAttributes(Base):
     def shared_attrs_bytes_from_event(event: Event[EventStateChangedData], dialect: SupportedDialect | None) -> bytes: ...
     @staticmethod
     def hash_shared_attrs_bytes(shared_attrs_bytes: bytes) -> int: ...
-    def to_native(self) -> dict[str, Any]: ...
 
 class StatesMeta(Base):
     __table_args__: Incomplete
@@ -245,6 +238,7 @@ class _StatisticsMeta:
     statistic_id: Mapped[str | None]
     source: Mapped[str | None]
     unit_of_measurement: Mapped[str | None]
+    unit_class: Mapped[str | None]
     has_mean: Mapped[bool | None]
     has_sum: Mapped[bool | None]
     name: Mapped[str | None]
@@ -266,7 +260,6 @@ class RecorderRuns(Base):
     closed_incorrect: Mapped[bool]
     created: Mapped[datetime]
     def __repr__(self) -> str: ...
-    def to_native(self, validate_entity_id: bool = True) -> Self: ...
 
 class MigrationChanges(Base):
     __tablename__ = TABLE_MIGRATION_CHANGES

homeassistant-stubs/components/recorder/entity_registry.pyi

@@ -1,4 +1,5 @@
 from .core import Recorder as Recorder
+from .statistics import async_update_statistics_metadata as async_update_statistics_metadata
 from .util import filter_unique_constraint_integrity_error as filter_unique_constraint_integrity_error, get_instance as get_instance, session_scope as session_scope
 from _typeshed import Incomplete
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback

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

@@ -7,8 +7,8 @@ from typing import Any
 
 __all__ = ['NEED_ATTRIBUTE_DOMAINS', 'SIGNIFICANT_DOMAINS', 'get_full_significant_states_with_session', 'get_last_state_changes', 'get_significant_states', 'get_significant_states_with_session', 'state_changes_during_period']
 
-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 get_last_state_changes(hass: HomeAssistant, number_of_states: int, entity_id: str) -> dict[str, list[State]]: ...
 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 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_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(hass: HomeAssistant, number_of_states: int, entity_id: str) -> dict[str, list[State]]: ...

homeassistant-stubs/components/recorder/migration.pyi

@@ -6,7 +6,7 @@ from .db_schema import BIG_INTEGER_SQL as BIG_INTEGER_SQL, Base as Base, CONTEXT
 from .models import StatisticMeanType as StatisticMeanType, 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, get_migration_changes as get_migration_changes, 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_entity_ids as has_used_states_entity_ids, 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 cleanup_statistics_timestamp_migration as cleanup_statistics_timestamp_migration, get_start_time as get_start_time
+from .statistics import _PRIMARY_UNIT_CONVERTERS as _PRIMARY_UNIT_CONVERTERS, 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, database_job_retry_wrapper_method as database_job_retry_wrapper_method, execute_stmt_lambda_element as execute_stmt_lambda_element, get_index_by_name as get_index_by_name, retryable_database_job_method as retryable_database_job_method, session_scope as session_scope
 from _typeshed import Incomplete
@@ -259,6 +259,17 @@ class _SchemaVersion49Migrator(_SchemaVersionMigrator, target_version=49):
 class _SchemaVersion50Migrator(_SchemaVersionMigrator, target_version=50):
     def _apply_update(self) -> None: ...
 
+class _SchemaVersion51Migrator(_SchemaVersionMigrator, target_version=51):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion52Migrator(_SchemaVersionMigrator, target_version=52):
+    def _apply_update(self) -> None: ...
+    def _apply_update_mysql(self) -> None: ...
+    def _apply_update_postgresql_sqlite(self) -> None: ...
+
+class _SchemaVersion53Migrator(_SchemaVersionMigrator, target_version=53):
+    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 _wipe_old_string_time_columns(instance: Recorder, engine: Engine, session: Session) -> None: ...

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

@@ -38,11 +38,13 @@ class StatisticMetaData(TypedDict):
     name: str | None
     source: str
     statistic_id: str
+    unit_class: str | None
     unit_of_measurement: str | None
 
 class CalendarStatisticPeriod(TypedDict, total=False):
     period: Literal['hour', 'day', 'week', 'month', 'year']
     offset: int
+    first_weekday: Literal['mon', 'tue', 'wed', 'thu', 'fri', 'sat', 'sun']
 
 class FixedStatisticPeriod(TypedDict, total=False):
     end_time: datetime

homeassistant-stubs/components/recorder/statistics.pyi

@@ -10,12 +10,13 @@ from datetime import datetime, timedelta
 from homeassistant.const import ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback, valid_entity_id as valid_entity_id
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.frame import report_usage as report_usage
 from homeassistant.helpers.recorder import DATA_RECORDER as DATA_RECORDER
 from homeassistant.helpers.singleton import singleton as singleton
 from homeassistant.helpers.typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from homeassistant.util.collection import chunked_or_all as chunked_or_all
 from homeassistant.util.enum import try_parse_enum as try_parse_enum
-from homeassistant.util.unit_conversion import ApparentPowerConverter as ApparentPowerConverter, AreaConverter as AreaConverter, BaseUnitConverter as BaseUnitConverter, BloodGlucoseConcentrationConverter as BloodGlucoseConcentrationConverter, ConductivityConverter as ConductivityConverter, DataRateConverter as DataRateConverter, DistanceConverter as DistanceConverter, DurationConverter as DurationConverter, ElectricCurrentConverter as ElectricCurrentConverter, ElectricPotentialConverter as ElectricPotentialConverter, EnergyConverter as EnergyConverter, EnergyDistanceConverter as EnergyDistanceConverter, InformationConverter as InformationConverter, MassConverter as MassConverter, MassVolumeConcentrationConverter as MassVolumeConcentrationConverter, PowerConverter as PowerConverter, PressureConverter as PressureConverter, ReactiveEnergyConverter as ReactiveEnergyConverter, ReactivePowerConverter as ReactivePowerConverter, SpeedConverter as SpeedConverter, TemperatureConverter as TemperatureConverter, UnitlessRatioConverter as UnitlessRatioConverter, VolumeConverter as VolumeConverter, VolumeFlowRateConverter as VolumeFlowRateConverter
+from homeassistant.util.unit_conversion import ApparentPowerConverter as ApparentPowerConverter, AreaConverter as AreaConverter, BaseUnitConverter as BaseUnitConverter, BloodGlucoseConcentrationConverter as BloodGlucoseConcentrationConverter, CarbonMonoxideConcentrationConverter as CarbonMonoxideConcentrationConverter, ConductivityConverter as ConductivityConverter, DataRateConverter as DataRateConverter, DistanceConverter as DistanceConverter, DurationConverter as DurationConverter, ElectricCurrentConverter as ElectricCurrentConverter, ElectricPotentialConverter as ElectricPotentialConverter, EnergyConverter as EnergyConverter, EnergyDistanceConverter as EnergyDistanceConverter, InformationConverter as InformationConverter, MassConverter as MassConverter, MassVolumeConcentrationConverter as MassVolumeConcentrationConverter, PowerConverter as PowerConverter, PressureConverter as PressureConverter, ReactiveEnergyConverter as ReactiveEnergyConverter, ReactivePowerConverter as ReactivePowerConverter, SpeedConverter as SpeedConverter, TemperatureConverter as TemperatureConverter, TemperatureDeltaConverter as TemperatureDeltaConverter, UnitlessRatioConverter as UnitlessRatioConverter, VolumeConverter as VolumeConverter, VolumeFlowRateConverter as VolumeFlowRateConverter
 from sqlalchemy import Label as Label, Select as Select
 from sqlalchemy.engine.row import Row
 from sqlalchemy.orm.session import Session as Session
@@ -29,8 +30,10 @@ def query_circular_mean(table: type[StatisticsBase]) -> tuple[Label, Label]: ...
 
 QUERY_STATISTICS_SUMMARY_MEAN: Incomplete
 QUERY_STATISTICS_SUMMARY_SUM: Incomplete
+_PRIMARY_UNIT_CONVERTERS: list[type[BaseUnitConverter]]
+_SECONDARY_UNIT_CONVERTERS: list[type[BaseUnitConverter]]
 STATISTIC_UNIT_TO_UNIT_CONVERTER: dict[str | None, type[BaseUnitConverter]]
-UNIT_CLASSES: Incomplete
+UNIT_CLASS_TO_UNIT_CONVERTER: dict[str | None, type[BaseUnitConverter]]
 DATA_SHORT_TERM_STATISTICS_RUN_CACHE: str
 
 def mean(values: list[float]) -> float | None: ...
@@ -63,11 +66,12 @@ class StatisticsRow(BaseStatisticsRow, total=False):
     mean_weight: float | None
     change: float | None
 
-def get_display_unit(hass: HomeAssistant, statistic_id: str, statistic_unit: str | None) -> str | None: ...
-def _get_statistic_to_display_unit_converter(statistic_unit: str | None, state_unit: str | None, requested_units: dict[str, str] | None, allow_none: bool = True) -> Callable[[float | None], float | None] | Callable[[float], float] | None: ...
-def _get_display_to_statistic_unit_converter(display_unit: str | None, statistic_unit: str | None) -> Callable[[float], float] | None: ...
-def _get_unit_converter(from_unit: str, to_unit: str) -> Callable[[float | None], float | None] | None: ...
-def can_convert_units(from_unit: str | None, to_unit: str | None) -> bool: ...
+def _get_unit_converter(unit_class: str | None, from_unit: str | None) -> type[BaseUnitConverter] | None: ...
+def get_display_unit(hass: HomeAssistant, statistic_id: str, unit_class: str | None, statistic_unit: str | None) -> str | None: ...
+def _get_statistic_to_display_unit_converter(unit_class: str | None, statistic_unit: str | None, state_unit: str | None, requested_units: dict[str, str] | None, allow_none: bool = True) -> Callable[[float | None], float | None] | Callable[[float], float] | None: ...
+def _get_display_to_statistic_unit_converter_func(unit_class: str | None, display_unit: str | None, statistic_unit: str | None) -> Callable[[float], float] | None: ...
+def _get_unit_converter_func(unit_class: str | None, from_unit: str, to_unit: str) -> Callable[[float | None], float | None] | None: ...
+def can_convert_units(unit_class: str | None, from_unit: str | None, to_unit: str | None) -> bool: ...
 
 @dataclasses.dataclass
 class PlatformCompiledStatistics:
@@ -101,7 +105,9 @@ def _update_statistics(session: Session, table: type[StatisticsBase], stat_id: i
 def get_metadata_with_session(instance: Recorder, session: Session, *, statistic_ids: set[str] | None = None, statistic_type: Literal['mean', 'sum'] | None = None, statistic_source: str | None = None) -> dict[str, tuple[int, StatisticMetaData]]: ...
 def get_metadata(hass: HomeAssistant, *, statistic_ids: set[str] | None = None, statistic_type: Literal['mean', 'sum'] | None = None, statistic_source: str | None = None) -> dict[str, tuple[int, StatisticMetaData]]: ...
 def clear_statistics(instance: Recorder, statistic_ids: list[str]) -> None: ...
-def update_statistics_metadata(instance: Recorder, statistic_id: str, new_statistic_id: str | None | UndefinedType, new_unit_of_measurement: str | None | UndefinedType) -> None: ...
+@callback
+def async_update_statistics_metadata(hass: HomeAssistant, statistic_id: str, *, new_statistic_id: str | UndefinedType = ..., new_unit_class: str | None | UndefinedType = ..., new_unit_of_measurement: str | None | UndefinedType = ..., on_done: Callable[[], None] | None = None, _called_from_ws_api: bool = False) -> None: ...
+def update_statistics_metadata(instance: Recorder, statistic_id: str, new_statistic_id: str | None | UndefinedType, new_unit_class: str | None | UndefinedType, new_unit_of_measurement: str | None | UndefinedType) -> None: ...
 async def async_list_statistic_ids(hass: HomeAssistant, statistic_ids: set[str] | None = None, statistic_type: Literal['mean', 'sum'] | None = None) -> list[dict]: ...
 def _statistic_by_id_from_metadata(hass: HomeAssistant, metadata: dict[str, tuple[int, StatisticMetaData]]) -> dict[str, dict[str, Any]]: ...
 def _flatten_list_statistic_ids_metadata_result(result: dict[str, dict[str, Any]]) -> list[dict]: ...
@@ -161,9 +167,9 @@ def _statistics_exists(session: Session, table: type[StatisticsBase], metadata_i
 @callback
 def _async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...
 @callback
-def async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...
+def async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData], *, _called_from_ws_api: bool = False) -> None: ...
 @callback
-def async_add_external_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...
+def async_add_external_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData], *, _called_from_ws_api: bool = False) -> None: ...
 def _import_statistics_with_session(instance: Recorder, session: Session, metadata: StatisticMetaData, statistics: Iterable[StatisticData], table: type[StatisticsBase]) -> bool: ...
 def get_short_term_statistics_run_cache(hass: HomeAssistant) -> ShortTermStatisticsRunCache: ...
 def cache_latest_short_term_statistic_id_for_metadata_id(run_cache: ShortTermStatisticsRunCache, session: Session, metadata_id: int) -> int | None: ...
@@ -172,6 +178,5 @@ def import_statistics(instance: Recorder, metadata: StatisticMetaData, statistic
 def adjust_statistics(instance: Recorder, statistic_id: str, start_time: datetime, sum_adjustment: float, adjustment_unit: str) -> bool: ...
 def _change_statistics_unit_for_table(session: Session, table: type[StatisticsBase], metadata_id: int, convert: Callable[[float | None], float | None]) -> None: ...
 def change_statistics_unit(instance: Recorder, statistic_id: str, new_unit: str, old_unit: str) -> None: ...
-@callback
-def async_change_statistics_unit(hass: HomeAssistant, statistic_id: str, *, new_unit_of_measurement: str, old_unit_of_measurement: str) -> None: ...
+async def async_change_statistics_unit(hass: HomeAssistant, statistic_id: str, *, new_unit_of_measurement: str, old_unit_of_measurement: str) -> None: ...
 def cleanup_statistics_timestamp_migration(instance: Recorder) -> bool: ...

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

@@ -11,7 +11,6 @@ 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/table_managers/statistics_meta.pyi

@@ -1,7 +1,8 @@
-from ..const import CIRCULAR_MEAN_SCHEMA_VERSION as CIRCULAR_MEAN_SCHEMA_VERSION
+from ..const import CIRCULAR_MEAN_SCHEMA_VERSION as CIRCULAR_MEAN_SCHEMA_VERSION, UNIT_CLASS_SCHEMA_VERSION as UNIT_CLASS_SCHEMA_VERSION
 from ..core import Recorder as Recorder
 from ..db_schema import StatisticsMeta as StatisticsMeta
 from ..models import StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData
+from ..statistics import STATISTIC_UNIT_TO_UNIT_CONVERTER as STATISTIC_UNIT_TO_UNIT_CONVERTER
 from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element
 from _typeshed import Incomplete
 from lru import LRU
@@ -12,7 +13,7 @@ from typing import Final, Literal
 
 CACHE_SIZE: int
 _LOGGER: Incomplete
-QUERY_STATISTIC_META: Incomplete
+QUERY_STATISTICS_META: Incomplete
 INDEX_ID: Final[int]
 INDEX_STATISTIC_ID: Final[int]
 INDEX_SOURCE: Final[int]
@@ -20,6 +21,7 @@ INDEX_UNIT_OF_MEASUREMENT: Final[int]
 INDEX_HAS_SUM: Final[int]
 INDEX_NAME: Final[int]
 INDEX_MEAN_TYPE: Final[int]
+INDEX_UNIT_CLASS: Final[int]
 
 def _generate_get_metadata_stmt(statistic_ids: set[str] | None = None, statistic_type: Literal['mean', 'sum'] | None = None, statistic_source: str | None = None, schema_version: int = 0) -> StatementLambdaElement: ...
 
@@ -37,7 +39,7 @@ class StatisticsMetaManager:
     def get_many(self, session: Session, statistic_ids: set[str] | None = None, statistic_type: Literal['mean', 'sum'] | None = None, statistic_source: str | None = None) -> dict[str, tuple[int, StatisticMetaData]]: ...
     def get_from_cache_threadsafe(self, statistic_ids: set[str]) -> dict[str, tuple[int, StatisticMetaData]]: ...
     def update_or_add(self, session: Session, new_metadata: StatisticMetaData, old_metadata_dict: dict[str, tuple[int, StatisticMetaData]]) -> tuple[str | None, int]: ...
-    def update_unit_of_measurement(self, session: Session, statistic_id: str, new_unit: str | None) -> None: ...
+    def update_unit_of_measurement(self, session: Session, statistic_id: str, new_unit_class: str | None, new_unit: str | None) -> None: ...
     def update_statistic_id(self, session: Session, source: str, old_statistic_id: str, new_statistic_id: str) -> None: ...
     def delete(self, session: Session, statistic_ids: list[str]) -> None: ...
     def reset(self) -> None: ...

homeassistant-stubs/components/recorder/tasks.pyi

@@ -41,6 +41,7 @@ class UpdateStatisticsMetadataTask(RecorderTask):
     on_done: Callable[[], None] | None
     statistic_id: str
     new_statistic_id: str | None | UndefinedType
+    new_unit_class: str | None | UndefinedType
     new_unit_of_measurement: str | None | UndefinedType
     def run(self, instance: Recorder) -> None: ...
 

homeassistant-stubs/components/recorder/util.pyi

@@ -7,6 +7,7 @@ from awesomeversion import AwesomeVersion
 from collections.abc import Callable, Generator, Sequence
 from contextlib import contextmanager
 from datetime import date, datetime
+from homeassistant.const import WEEKDAYS as WEEKDAYS
 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
@@ -47,7 +48,7 @@ FIRST_POSSIBLE_SUNDAY: int
 SUNDAY_WEEKDAY: int
 DAYS_IN_WEEK: int
 
-def execute(qry: Query, to_native: bool = False, validate_entity_ids: bool = True) -> list[Row]: ...
+def execute(qry: Query) -> 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: ...
 def dburl_to_path(dburl: str) -> str: ...