homeassistant-stubs 2025.10.4__py3-none-any.whl → 2025.11.0b1__py3-none-any.whl

homeassistant-stubs/components/onkyo/receiver.pyi

@@ -31,5 +31,5 @@ class ReceiverManager:
     async def write(self, message: Instruction) -> None: ...
     def start_unloading(self) -> None: ...
 
-async def async_interview(host: str) -> ReceiverInfo | None: ...
+async def async_interview(host: str) -> ReceiverInfo: ...
 async def async_discover(hass: HomeAssistant) -> Iterable[ReceiverInfo]: ...

homeassistant-stubs/components/open_router/entity.pyi

@@ -6,11 +6,13 @@ from collections.abc import AsyncGenerator, Callable as Callable
 from homeassistant.components import conversation as conversation
 from homeassistant.config_entries import ConfigSubentry as ConfigSubentry
 from homeassistant.const import CONF_MODEL as CONF_MODEL
+from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import llm as llm
 from homeassistant.helpers.entity import Entity as Entity
-from openai.types.chat import ChatCompletionFunctionToolParam, ChatCompletionMessage as ChatCompletionMessage, ChatCompletionMessageParam as ChatCompletionMessageParam
+from openai.types.chat import ChatCompletionContentPartImageParam as ChatCompletionContentPartImageParam, ChatCompletionFunctionToolParam, ChatCompletionMessage as ChatCompletionMessage, ChatCompletionMessageParam as ChatCompletionMessageParam
 from openai.types.shared_params.response_format_json_schema import JSONSchema as JSONSchema
+from pathlib import Path
 from typing import Any
 
 MAX_TOOL_ITERATIONS: int
@@ -21,6 +23,7 @@ def _format_tool(tool: llm.Tool, custom_serializer: Callable[[Any], Any] | None)
 def _convert_content_to_chat_message(content: conversation.Content) -> ChatCompletionMessageParam | None: ...
 def _decode_tool_arguments(arguments: str) -> Any: ...
 async def _transform_response(message: ChatCompletionMessage) -> AsyncGenerator[conversation.AssistantContentDeltaDict]: ...
+async def async_prepare_files_for_prompt(hass: HomeAssistant, files: list[tuple[Path, str | None]]) -> list[ChatCompletionContentPartImageParam]: ...
 
 class OpenRouterEntity(Entity):
     _attr_has_entity_name: bool

homeassistant-stubs/components/openai_conversation/ai_task.pyi

@@ -1,5 +1,5 @@
 from . import OpenAIConfigEntry as OpenAIConfigEntry
-from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, UNSUPPORTED_IMAGE_MODELS as UNSUPPORTED_IMAGE_MODELS
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_IMAGE_MODEL as CONF_IMAGE_MODEL, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_IMAGE_MODEL as RECOMMENDED_IMAGE_MODEL, UNSUPPORTED_IMAGE_MODELS as UNSUPPORTED_IMAGE_MODELS
 from .entity import OpenAIBaseLLMEntity as OpenAIBaseLLMEntity
 from _typeshed import Incomplete
 from homeassistant.components import ai_task as ai_task, conversation as conversation

homeassistant-stubs/components/openai_conversation/config_flow.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_CODE_INTERPRETER as CONF_CODE_INTERPRETER, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_RECOMMENDED as CONF_RECOMMENDED, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_VERBOSITY as CONF_VERBOSITY, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DEFAULT_AI_TASK_NAME as DEFAULT_AI_TASK_NAME, DEFAULT_CONVERSATION_NAME as DEFAULT_CONVERSATION_NAME, DOMAIN as DOMAIN, RECOMMENDED_AI_TASK_OPTIONS as RECOMMENDED_AI_TASK_OPTIONS, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_CODE_INTERPRETER as RECOMMENDED_CODE_INTERPRETER, RECOMMENDED_CONVERSATION_OPTIONS as RECOMMENDED_CONVERSATION_OPTIONS, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_VERBOSITY as RECOMMENDED_VERBOSITY, RECOMMENDED_WEB_SEARCH as RECOMMENDED_WEB_SEARCH, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE, RECOMMENDED_WEB_SEARCH_USER_LOCATION as RECOMMENDED_WEB_SEARCH_USER_LOCATION, UNSUPPORTED_MODELS as UNSUPPORTED_MODELS, UNSUPPORTED_WEB_SEARCH_MODELS as UNSUPPORTED_WEB_SEARCH_MODELS
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_CODE_INTERPRETER as CONF_CODE_INTERPRETER, CONF_IMAGE_MODEL as CONF_IMAGE_MODEL, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_RECOMMENDED as CONF_RECOMMENDED, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_VERBOSITY as CONF_VERBOSITY, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DEFAULT_AI_TASK_NAME as DEFAULT_AI_TASK_NAME, DEFAULT_CONVERSATION_NAME as DEFAULT_CONVERSATION_NAME, DOMAIN as DOMAIN, RECOMMENDED_AI_TASK_OPTIONS as RECOMMENDED_AI_TASK_OPTIONS, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_CODE_INTERPRETER as RECOMMENDED_CODE_INTERPRETER, RECOMMENDED_CONVERSATION_OPTIONS as RECOMMENDED_CONVERSATION_OPTIONS, RECOMMENDED_IMAGE_MODEL as RECOMMENDED_IMAGE_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_VERBOSITY as RECOMMENDED_VERBOSITY, RECOMMENDED_WEB_SEARCH as RECOMMENDED_WEB_SEARCH, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE, RECOMMENDED_WEB_SEARCH_USER_LOCATION as RECOMMENDED_WEB_SEARCH_USER_LOCATION, UNSUPPORTED_IMAGE_MODELS as UNSUPPORTED_IMAGE_MODELS, UNSUPPORTED_MODELS as UNSUPPORTED_MODELS, UNSUPPORTED_WEB_SEARCH_MODELS as UNSUPPORTED_WEB_SEARCH_MODELS
 from _typeshed import Incomplete
 from homeassistant.components.zone import ENTITY_ID_HOME as ENTITY_ID_HOME
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, ConfigSubentryFlow as ConfigSubentryFlow, SubentryFlowResult as SubentryFlowResult

homeassistant-stubs/components/openai_conversation/const.pyi

@@ -9,6 +9,7 @@ DEFAULT_CONVERSATION_NAME: str
 DEFAULT_AI_TASK_NAME: str
 DEFAULT_NAME: str
 CONF_CHAT_MODEL: str
+CONF_IMAGE_MODEL: str
 CONF_CODE_INTERPRETER: str
 CONF_FILENAMES: str
 CONF_MAX_TOKENS: str
@@ -27,6 +28,7 @@ CONF_WEB_SEARCH_COUNTRY: str
 CONF_WEB_SEARCH_TIMEZONE: str
 RECOMMENDED_CODE_INTERPRETER: bool
 RECOMMENDED_CHAT_MODEL: str
+RECOMMENDED_IMAGE_MODEL: str
 RECOMMENDED_MAX_TOKENS: int
 RECOMMENDED_REASONING_EFFORT: str
 RECOMMENDED_TEMPERATURE: float

homeassistant-stubs/components/openai_conversation/entity.pyi

@@ -1,6 +1,6 @@
 import voluptuous as vol
 from . import OpenAIConfigEntry as OpenAIConfigEntry
-from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_CODE_INTERPRETER as CONF_CODE_INTERPRETER, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_VERBOSITY as CONF_VERBOSITY, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DOMAIN as DOMAIN, LOGGER as LOGGER, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_VERBOSITY as RECOMMENDED_VERBOSITY, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_CODE_INTERPRETER as CONF_CODE_INTERPRETER, CONF_IMAGE_MODEL as CONF_IMAGE_MODEL, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_VERBOSITY as CONF_VERBOSITY, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DOMAIN as DOMAIN, LOGGER as LOGGER, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_IMAGE_MODEL as RECOMMENDED_IMAGE_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_VERBOSITY as RECOMMENDED_VERBOSITY, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE
 from _typeshed import Incomplete
 from collections.abc import AsyncGenerator, Callable as Callable, Iterable
 from homeassistant.components import conversation as conversation

homeassistant-stubs/components/opower/coordinator.pyi

@@ -9,6 +9,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+from homeassistant.util.unit_conversion import EnergyConverter as EnergyConverter, VolumeConverter as VolumeConverter
 from opower import Account as Account, CostRead as CostRead, Forecast
 
 _LOGGER: Incomplete

homeassistant-stubs/components/ping/__init__.pyi

@@ -12,6 +12,7 @@ CONFIG_SCHEMA: Incomplete
 PLATFORMS: Incomplete
 DATA_PRIVILEGED_KEY: HassKey[bool | None]
 
+async def async_migrate_entry(hass: HomeAssistant, entry: PingConfigEntry) -> bool: ...
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, entry: PingConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: PingConfigEntry) -> bool: ...

homeassistant-stubs/components/ping/config_flow.pyi

@@ -14,6 +14,7 @@ def _clean_user_input(user_input: dict[str, Any]) -> dict[str, Any]: ...
 
 class PingConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
+    MINOR_VERSION: int
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     @staticmethod
     @callback

homeassistant-stubs/components/ping/device_tracker.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_IMPORTED_BY as CONF_IMPORTED_BY
+from .const import CONF_IMPORTED_BY as CONF_IMPORTED_BY, DOMAIN as DOMAIN
 from .coordinator import PingConfigEntry as PingConfigEntry, PingUpdateCoordinator as PingUpdateCoordinator
 from _typeshed import Incomplete
 from datetime import datetime
@@ -14,7 +14,8 @@ class PingDeviceTracker(CoordinatorEntity[PingUpdateCoordinator], ScannerEntity)
     _attr_name: Incomplete
     config_entry: Incomplete
     _consider_home_interval: Incomplete
-    def __init__(self, config_entry: PingConfigEntry, coordinator: PingUpdateCoordinator) -> None: ...
+    device_entry: Incomplete
+    def __init__(self, hass: HomeAssistant, config_entry: PingConfigEntry, coordinator: PingUpdateCoordinator) -> None: ...
     @property
     def ip_address(self) -> str: ...
     @property

homeassistant-stubs/components/ping/entity.pyi

@@ -1,3 +1,4 @@
+from .const import DOMAIN as DOMAIN
 from .coordinator import PingConfigEntry as PingConfigEntry, PingUpdateCoordinator as PingUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo

homeassistant-stubs/components/portainer/button.pyi

@@ -0,0 +1,29 @@
+from . import PortainerConfigEntry as PortainerConfigEntry
+from .const import DOMAIN as DOMAIN
+from .coordinator import PortainerCoordinator as PortainerCoordinator, PortainerCoordinatorData as PortainerCoordinatorData
+from .entity import PortainerContainerEntity as PortainerContainerEntity
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable, Coroutine
+from dataclasses import dataclass
+from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
+from homeassistant.const import EntityCategory as EntityCategory
+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 pyportainer.models.docker import DockerContainer as DockerContainer
+from typing import Any
+
+@dataclass(frozen=True, kw_only=True)
+class PortainerButtonDescription(ButtonEntityDescription):
+    press_action: Callable[[Portainer, int, str], Coroutine[Any, Any, None]]
+
+BUTTONS: tuple[PortainerButtonDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PortainerConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class PortainerButton(PortainerContainerEntity, ButtonEntity):
+    entity_description: PortainerButtonDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: PortainerCoordinator, entity_description: PortainerButtonDescription, device_info: DockerContainer, via_device: PortainerCoordinatorData) -> None: ...
+    async def async_press(self) -> None: ...

homeassistant-stubs/components/portainer/config_flow.pyi

@@ -1,5 +1,6 @@
 from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
+from collections.abc import Mapping
 from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import CONF_API_TOKEN as CONF_API_TOKEN, CONF_URL as CONF_URL, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant
@@ -15,6 +16,9 @@ async def _validate_input(hass: HomeAssistant, data: dict[str, Any]) -> None: ..
 class PortainerConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
+    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
 
 class CannotConnect(HomeAssistantError): ...
 class InvalidAuth(HomeAssistantError): ...

homeassistant-stubs/components/portainer/const.pyi

@@ -1,2 +1,3 @@
 DOMAIN: str
 DEFAULT_NAME: str
+ENDPOINT_STATUS_DOWN: int

homeassistant-stubs/components/portainer/coordinator.pyi

@@ -1,13 +1,15 @@
-from .const import DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN, ENDPOINT_STATUS_DOWN as ENDPOINT_STATUS_DOWN
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_URL as CONF_URL
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.exceptions import ConfigEntryError as ConfigEntryError, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 from pyportainer import Portainer as Portainer
 from pyportainer.models.docker import DockerContainer as DockerContainer
+from pyportainer.models.docker_inspect import DockerInfo as DockerInfo, DockerVersion as DockerVersion
 from pyportainer.models.portainer import Endpoint as Endpoint
 
 type PortainerConfigEntry = ConfigEntry[PortainerCoordinator]
@@ -20,10 +22,17 @@ class PortainerCoordinatorData:
     name: str | None
     endpoint: Endpoint
     containers: dict[str, DockerContainer]
+    docker_version: DockerVersion
+    docker_info: DockerInfo
 
 class PortainerCoordinator(DataUpdateCoordinator[dict[int, PortainerCoordinatorData]]):
     config_entry: PortainerConfigEntry
     portainer: Incomplete
+    known_endpoints: set[int]
+    known_containers: set[tuple[int, str]]
+    new_endpoints_callbacks: list[Callable[[list[PortainerCoordinatorData]], None]]
+    new_containers_callbacks: list[Callable[[list[tuple[PortainerCoordinatorData, DockerContainer]]], None]]
     def __init__(self, hass: HomeAssistant, config_entry: PortainerConfigEntry, portainer: Portainer) -> None: ...
     async def _async_setup(self) -> None: ...
     async def _async_update_data(self) -> dict[int, PortainerCoordinatorData]: ...
+    def _async_add_remove_endpoints(self, mapped_endpoints: dict[int, PortainerCoordinatorData]) -> None: ...

homeassistant-stubs/components/portainer/diagnostics.pyi

@@ -0,0 +1,12 @@
+from . import PortainerConfigEntry as PortainerConfigEntry
+from .coordinator import PortainerCoordinator as PortainerCoordinator
+from _typeshed import Incomplete
+from homeassistant.components.diagnostics import async_redact_data as async_redact_data
+from homeassistant.const import CONF_API_TOKEN as CONF_API_TOKEN
+from homeassistant.core import HomeAssistant as HomeAssistant
+from typing import Any
+
+TO_REDACT: Incomplete
+
+def _serialize_coordinator(coordinator: PortainerCoordinator) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: PortainerConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/portainer/entity.pyi

@@ -1,6 +1,7 @@
 from .const import DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN
 from .coordinator import PortainerCoordinator as PortainerCoordinator, PortainerCoordinatorData as PortainerCoordinatorData
 from _typeshed import Incomplete
+from homeassistant.const import CONF_URL as CONF_URL
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from pyportainer.models.docker import DockerContainer as DockerContainer
@@ -18,5 +19,6 @@ class PortainerContainerEntity(PortainerCoordinatorEntity):
     _device_info: Incomplete
     device_id: Incomplete
     endpoint_id: Incomplete
+    device_name: Incomplete
     _attr_device_info: Incomplete
     def __init__(self, device_info: DockerContainer, coordinator: PortainerCoordinator, via_device: PortainerCoordinatorData) -> None: ...

homeassistant-stubs/components/portainer/sensor.pyi

@@ -0,0 +1,41 @@
+from .coordinator import PortainerConfigEntry as PortainerConfigEntry, 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 UnitOfInformation as UnitOfInformation
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from pyportainer.models.docker import DockerContainer as DockerContainer
+
+@dataclass(frozen=True, kw_only=True)
+class PortainerContainerSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[DockerContainer], 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: DockerContainer, 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,35 @@
+from . import PortainerConfigEntry as PortainerConfigEntry
+from .const import DOMAIN as DOMAIN
+from .coordinator import 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 pyportainer.models.docker import DockerContainer as DockerContainer
+from typing import Any
+
+@dataclass(frozen=True, kw_only=True)
+class PortainerSwitchEntityDescription(SwitchEntityDescription):
+    is_on_fn: Callable[[DockerContainer], 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: DockerContainer, 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/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/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/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,9 @@ 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: ...
+
 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,6 +38,7 @@ 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):

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: ...