homeassistant-stubs 2024.3.3__py3-none-any.whl → 2024.4.0__py3-none-any.whl

homeassistant-stubs/helpers/label_registry.pyi

@@ -1,10 +1,11 @@
+from .normalized_name_base_registry import NormalizedNameBaseRegistryEntry as NormalizedNameBaseRegistryEntry, NormalizedNameBaseRegistryItems as NormalizedNameBaseRegistryItems, normalize_name as normalize_name
+from .registry import BaseRegistry as BaseRegistry
 from .storage import Store as Store
-from .typing import EventType as EventType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
+from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
-from collections import UserDict
-from collections.abc import Iterable, ValuesView
+from collections.abc import Iterable
 from dataclasses import dataclass
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util import slugify as slugify
 from typing import Literal, TypedDict
 
@@ -12,33 +13,22 @@ DATA_REGISTRY: str
 EVENT_LABEL_REGISTRY_UPDATED: str
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
-SAVE_DELAY: int
 
 class EventLabelRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     label_id: str
-EventLabelRegistryUpdated = EventType[EventLabelRegistryUpdatedData]
+EventLabelRegistryUpdated = Event[EventLabelRegistryUpdatedData]
 
-@dataclass(slots=True, frozen=True)
-class LabelEntry:
+@dataclass(slots=True, frozen=True, kw_only=True)
+class LabelEntry(NormalizedNameBaseRegistryEntry):
     label_id: str
-    name: str
-    normalized_name: str
     description: str | None = ...
     color: str | None = ...
     icon: str | None = ...
-    def __init__(self, label_id, name, normalized_name, description, color, icon) -> None: ...
+    def __init__(self, *, name, normalized_name, label_id, description, color, icon) -> None: ...
 
-class LabelRegistryItems(UserDict[str, LabelEntry]):
-    _normalized_names: Incomplete
-    def __init__(self) -> None: ...
-    def values(self) -> ValuesView[LabelEntry]: ...
-    def __setitem__(self, key: str, entry: LabelEntry) -> None: ...
-    def __delitem__(self, key: str) -> None: ...
-    def get_label_by_name(self, name: str) -> LabelEntry | None: ...
-
-class LabelRegistry:
-    labels: LabelRegistryItems
+class LabelRegistry(BaseRegistry):
+    labels: NormalizedNameBaseRegistryItems[LabelEntry]
     _label_data: dict[str, LabelEntry]
     hass: Incomplete
     _store: Incomplete
@@ -51,9 +41,7 @@ class LabelRegistry:
     def async_delete(self, label_id: str) -> None: ...
     def async_update(self, label_id: str, *, color: str | None | UndefinedType = ..., description: str | None | UndefinedType = ..., icon: str | None | UndefinedType = ..., name: str | UndefinedType = ...) -> LabelEntry: ...
     async def async_load(self) -> None: ...
-    def async_schedule_save(self) -> None: ...
     def _data_to_save(self) -> dict[str, list[dict[str, str | None]]]: ...
 
 def async_get(hass: HomeAssistant) -> LabelRegistry: ...
 async def async_load(hass: HomeAssistant) -> None: ...
-def _normalize_label_name(label_name: str) -> str: ...

homeassistant-stubs/helpers/normalized_name_base_registry.pyi

@@ -0,0 +1,22 @@
+from _typeshed import Incomplete
+from collections import UserDict
+from collections.abc import ValuesView
+from dataclasses import dataclass
+from typing import TypeVar
+
+@dataclass(slots=True, frozen=True, kw_only=True)
+class NormalizedNameBaseRegistryEntry:
+    name: str
+    normalized_name: str
+    def __init__(self, *, name, normalized_name) -> None: ...
+_VT = TypeVar('_VT', bound=NormalizedNameBaseRegistryEntry)
+
+def normalize_name(name: str) -> str: ...
+
+class NormalizedNameBaseRegistryItems(UserDict[str, _VT]):
+    _normalized_names: Incomplete
+    def __init__(self) -> None: ...
+    def values(self) -> ValuesView[_VT]: ...
+    def __setitem__(self, key: str, entry: _VT) -> None: ...
+    def __delitem__(self, key: str) -> None: ...
+    def get_by_name(self, name: str) -> _VT | None: ...

homeassistant-stubs/helpers/ratelimit.pyi

@@ -1,6 +1,5 @@
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Hashable
-from datetime import datetime, timedelta
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from typing import TypeVarTuple
 
@@ -13,7 +12,7 @@ class KeyedRateLimit:
     _rate_limit_timers: Incomplete
     def __init__(self, hass: HomeAssistant) -> None: ...
     def async_has_timer(self, key: Hashable) -> bool: ...
-    def async_triggered(self, key: Hashable, now: datetime | None = None) -> None: ...
+    def async_triggered(self, key: Hashable, now: float | None = None) -> None: ...
     def async_cancel_timer(self, key: Hashable) -> None: ...
     def async_remove(self) -> None: ...
-    def async_schedule_action(self, key: Hashable, rate_limit: timedelta | None, now: datetime, action: Callable[[Unpack[_Ts]], None], *args: Unpack[_Ts]) -> datetime | None: ...
+    def async_schedule_action(self, key: Hashable, rate_limit: float | None, now: float, action: Callable[[Unpack[_Ts]], None], *args: Unpack[_Ts]) -> float | None: ...

homeassistant-stubs/helpers/registry.pyi

@@ -0,0 +1,15 @@
+import abc
+from .storage import Store as Store
+from abc import ABC, abstractmethod
+from homeassistant.core import CoreState as CoreState, HomeAssistant as HomeAssistant, callback as callback
+from typing import Any
+
+SAVE_DELAY: int
+SAVE_DELAY_LONG: int
+
+class BaseRegistry(ABC, metaclass=abc.ABCMeta):
+    hass: HomeAssistant
+    _store: Store
+    def async_schedule_save(self) -> None: ...
+    @abstractmethod
+    def _data_to_save(self) -> dict[str, Any]: ...

homeassistant-stubs/helpers/schema_config_entry_flow.pyi

@@ -6,9 +6,9 @@ from _typeshed import Incomplete
 from abc import ABC, abstractmethod
 from collections.abc import Callable as Callable, Coroutine, Mapping
 from dataclasses import dataclass
-from homeassistant import config_entries as config_entries
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, OptionsFlowWithConfigEntry as OptionsFlowWithConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback, split_entity_id as split_entity_id
-from homeassistant.data_entry_flow import FlowResult as FlowResult, UnknownHandler as UnknownHandler
+from homeassistant.data_entry_flow import UnknownHandler as UnknownHandler
 from typing import Any
 
 class SchemaFlowError(Exception): ...
@@ -41,15 +41,15 @@ class SchemaCommonFlowHandler:
     def options(self) -> dict[str, Any]: ...
     @property
     def flow_state(self) -> dict[str, Any]: ...
-    async def async_step(self, step_id: str, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def async_step(self, step_id: str, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def _get_schema(self, form_step: SchemaFlowFormStep) -> vol.Schema | None: ...
-    async def _async_form_step(self, step_id: str, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def _async_form_step(self, step_id: str, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     def _update_and_remove_omitted_optional_keys(self, values: dict[str, Any], user_input: dict[str, Any], data_schema: vol.Schema | None) -> None: ...
-    async def _show_next_step_or_create_entry(self, form_step: SchemaFlowFormStep) -> FlowResult: ...
-    async def _show_next_step(self, next_step_id: str, error: SchemaFlowError | None = None, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def _async_menu_step(self, step_id: str, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def _show_next_step_or_create_entry(self, form_step: SchemaFlowFormStep) -> ConfigFlowResult: ...
+    async def _show_next_step(self, next_step_id: str, error: SchemaFlowError | None = None, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def _async_menu_step(self, step_id: str, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
 
-class SchemaConfigFlowHandler(config_entries.ConfigFlow, ABC, metaclass=abc.ABCMeta):
+class SchemaConfigFlowHandler(ConfigFlow, ABC, metaclass=abc.ABCMeta):
     config_flow: Mapping[str, SchemaFlowStep]
     options_flow: Mapping[str, SchemaFlowStep] | None
     VERSION: int
@@ -59,23 +59,23 @@ class SchemaConfigFlowHandler(config_entries.ConfigFlow, ABC, metaclass=abc.ABCM
     @staticmethod
     async def async_setup_preview(hass: HomeAssistant) -> None: ...
     @classmethod
-    def async_supports_options_flow(cls, config_entry: config_entries.ConfigEntry) -> bool: ...
+    def async_supports_options_flow(cls, config_entry: ConfigEntry) -> bool: ...
     @staticmethod
-    def _async_step(step_id: str) -> Callable[[SchemaConfigFlowHandler, dict[str, Any] | None], Coroutine[Any, Any, FlowResult]]: ...
+    def _async_step(step_id: str) -> Callable[[SchemaConfigFlowHandler, dict[str, Any] | None], Coroutine[Any, Any, ConfigFlowResult]]: ...
     @abstractmethod
     def async_config_entry_title(self, options: Mapping[str, Any]) -> str: ...
     def async_config_flow_finished(self, options: Mapping[str, Any]) -> None: ...
     @staticmethod
     def async_options_flow_finished(hass: HomeAssistant, options: Mapping[str, Any]) -> None: ...
-    def async_create_entry(self, data: Mapping[str, Any], **kwargs: Any) -> FlowResult: ...
+    def async_create_entry(self, data: Mapping[str, Any], **kwargs: Any) -> ConfigFlowResult: ...
 
-class SchemaOptionsFlowHandler(config_entries.OptionsFlowWithConfigEntry):
+class SchemaOptionsFlowHandler(OptionsFlowWithConfigEntry):
     _common_handler: Incomplete
     _async_options_flow_finished: Incomplete
-    def __init__(self, config_entry: config_entries.ConfigEntry, options_flow: Mapping[str, SchemaFlowStep], async_options_flow_finished: Callable[[HomeAssistant, Mapping[str, Any]], None] | None = None, async_setup_preview: Callable[[HomeAssistant], Coroutine[Any, Any, None]] | None = None) -> None: ...
+    def __init__(self, config_entry: ConfigEntry, options_flow: Mapping[str, SchemaFlowStep], async_options_flow_finished: Callable[[HomeAssistant, Mapping[str, Any]], None] | None = None, async_setup_preview: Callable[[HomeAssistant], Coroutine[Any, Any, None]] | None = None) -> None: ...
     @staticmethod
-    def _async_step(step_id: str) -> Callable[[SchemaConfigFlowHandler, dict[str, Any] | None], Coroutine[Any, Any, FlowResult]]: ...
-    def async_create_entry(self, data: Mapping[str, Any], **kwargs: Any) -> FlowResult: ...
+    def _async_step(step_id: str) -> Callable[[SchemaConfigFlowHandler, dict[str, Any] | None], Coroutine[Any, Any, ConfigFlowResult]]: ...
+    def async_create_entry(self, data: Mapping[str, Any], **kwargs: Any) -> ConfigFlowResult: ...
 
 def wrapped_entity_config_entry_title(hass: HomeAssistant, entity_id_or_uuid: str) -> str: ...
 def entity_selector_without_own_entities(handler: SchemaOptionsFlowHandler, entity_selector_config: selector.EntitySelectorConfig) -> vol.Schema: ...

homeassistant-stubs/helpers/script.pyi

@@ -3,7 +3,7 @@ import logging
 import voluptuous as vol
 from . import condition as condition, service as service, template as template
 from .condition import ConditionCheckerType as ConditionCheckerType, trace_condition_function as trace_condition_function
-from .dispatcher import SignalType as SignalType, async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
+from .dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
 from .event import async_call_later as async_call_later, async_track_template as async_track_template
 from .script_variables import ScriptVariables as ScriptVariables
 from .trace import TraceElement as TraceElement, async_trace_path as async_trace_path, script_execution_set as script_execution_set, trace_append_element as trace_append_element, trace_id_get as trace_id_get, trace_path as trace_path, trace_path_get as trace_path_get, trace_path_stack_cv as trace_path_stack_cv, trace_set_result as trace_set_result, trace_stack_cv as trace_stack_cv, trace_stack_pop as trace_stack_pop, trace_stack_push as trace_stack_push, trace_stack_top as trace_stack_top, trace_update_result as trace_update_result
@@ -13,16 +13,18 @@ from _typeshed import Incomplete
 from collections.abc import AsyncGenerator, Callable as Callable, Mapping, Sequence
 from contextvars import ContextVar
 from dataclasses import dataclass
-from datetime import datetime
+from datetime import datetime, timedelta
 from functools import cached_property as cached_property
 from homeassistant import exceptions as exceptions
 from homeassistant.components import scene as scene
 from homeassistant.components.logger import LOGSEVERITY as LOGSEVERITY
-from homeassistant.const import ATTR_AREA_ID as ATTR_AREA_ID, ATTR_DEVICE_ID as ATTR_DEVICE_ID, ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ALIAS as CONF_ALIAS, CONF_CHOOSE as CONF_CHOOSE, CONF_CONDITION as CONF_CONDITION, CONF_CONDITIONS as CONF_CONDITIONS, CONF_CONTINUE_ON_ERROR as CONF_CONTINUE_ON_ERROR, CONF_CONTINUE_ON_TIMEOUT as CONF_CONTINUE_ON_TIMEOUT, CONF_COUNT as CONF_COUNT, CONF_DEFAULT as CONF_DEFAULT, CONF_DELAY as CONF_DELAY, CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_DOMAIN as CONF_DOMAIN, CONF_ELSE as CONF_ELSE, CONF_ENABLED as CONF_ENABLED, CONF_ERROR as CONF_ERROR, CONF_EVENT as CONF_EVENT, CONF_EVENT_DATA as CONF_EVENT_DATA, CONF_EVENT_DATA_TEMPLATE as CONF_EVENT_DATA_TEMPLATE, CONF_FOR_EACH as CONF_FOR_EACH, CONF_IF as CONF_IF, CONF_MODE as CONF_MODE, CONF_PARALLEL as CONF_PARALLEL, CONF_REPEAT as CONF_REPEAT, CONF_RESPONSE_VARIABLE as CONF_RESPONSE_VARIABLE, CONF_SCENE as CONF_SCENE, CONF_SEQUENCE as CONF_SEQUENCE, CONF_SERVICE as CONF_SERVICE, CONF_SERVICE_DATA as CONF_SERVICE_DATA, CONF_SERVICE_DATA_TEMPLATE as CONF_SERVICE_DATA_TEMPLATE, CONF_SET_CONVERSATION_RESPONSE as CONF_SET_CONVERSATION_RESPONSE, CONF_STOP as CONF_STOP, CONF_TARGET as CONF_TARGET, CONF_THEN as CONF_THEN, CONF_TIMEOUT as CONF_TIMEOUT, CONF_UNTIL as CONF_UNTIL, CONF_VARIABLES as CONF_VARIABLES, CONF_WAIT_FOR_TRIGGER as CONF_WAIT_FOR_TRIGGER, CONF_WAIT_TEMPLATE as CONF_WAIT_TEMPLATE, CONF_WHILE as CONF_WHILE, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, SERVICE_TURN_ON as SERVICE_TURN_ON
+from homeassistant.const import ATTR_AREA_ID as ATTR_AREA_ID, ATTR_DEVICE_ID as ATTR_DEVICE_ID, ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_FLOOR_ID as ATTR_FLOOR_ID, ATTR_LABEL_ID as ATTR_LABEL_ID, CONF_ALIAS as CONF_ALIAS, CONF_CHOOSE as CONF_CHOOSE, CONF_CONDITION as CONF_CONDITION, CONF_CONDITIONS as CONF_CONDITIONS, CONF_CONTINUE_ON_ERROR as CONF_CONTINUE_ON_ERROR, CONF_CONTINUE_ON_TIMEOUT as CONF_CONTINUE_ON_TIMEOUT, CONF_COUNT as CONF_COUNT, CONF_DEFAULT as CONF_DEFAULT, CONF_DELAY as CONF_DELAY, CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_DOMAIN as CONF_DOMAIN, CONF_ELSE as CONF_ELSE, CONF_ENABLED as CONF_ENABLED, CONF_ERROR as CONF_ERROR, CONF_EVENT as CONF_EVENT, CONF_EVENT_DATA as CONF_EVENT_DATA, CONF_EVENT_DATA_TEMPLATE as CONF_EVENT_DATA_TEMPLATE, CONF_FOR_EACH as CONF_FOR_EACH, CONF_IF as CONF_IF, CONF_MODE as CONF_MODE, CONF_PARALLEL as CONF_PARALLEL, CONF_REPEAT as CONF_REPEAT, CONF_RESPONSE_VARIABLE as CONF_RESPONSE_VARIABLE, CONF_SCENE as CONF_SCENE, CONF_SEQUENCE as CONF_SEQUENCE, CONF_SERVICE as CONF_SERVICE, CONF_SERVICE_DATA as CONF_SERVICE_DATA, CONF_SERVICE_DATA_TEMPLATE as CONF_SERVICE_DATA_TEMPLATE, CONF_SET_CONVERSATION_RESPONSE as CONF_SET_CONVERSATION_RESPONSE, CONF_STOP as CONF_STOP, CONF_TARGET as CONF_TARGET, CONF_THEN as CONF_THEN, CONF_TIMEOUT as CONF_TIMEOUT, CONF_UNTIL as CONF_UNTIL, CONF_VARIABLES as CONF_VARIABLES, CONF_WAIT_FOR_TRIGGER as CONF_WAIT_FOR_TRIGGER, CONF_WAIT_TEMPLATE as CONF_WAIT_TEMPLATE, CONF_WHILE as CONF_WHILE, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, SERVICE_TURN_ON as SERVICE_TURN_ON
 from homeassistant.core import Context as Context, Event as Event, HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse, callback as callback
 from homeassistant.util import slugify as slugify
+from homeassistant.util.async_ import create_eager_task as create_eager_task
 from homeassistant.util.dt import utcnow as utcnow
-from typing import Any, TypeVar, TypedDict
+from homeassistant.util.signal_type import SignalType as SignalType, SignalTypeFormat as SignalTypeFormat
+from typing import Any, Literal, TypeVar, TypedDict
 
 _T = TypeVar('_T')
 SCRIPT_MODE_PARALLEL: str
@@ -49,12 +51,15 @@ _TIMEOUT_MSG: str
 _SHUTDOWN_MAX_WAIT: int
 ACTION_TRACE_NODE_MAX_LEN: int
 SCRIPT_BREAKPOINT_HIT: Incomplete
-SCRIPT_DEBUG_CONTINUE_STOP: str
+SCRIPT_DEBUG_CONTINUE_STOP: SignalTypeFormat[Literal['continue', 'stop']]
 SCRIPT_DEBUG_CONTINUE_ALL: str
 script_stack_cv: ContextVar[list[int] | None]
 
+class ScriptStoppedError(Exception): ...
+
+def _set_result_unless_done(future: asyncio.Future[None]) -> None: ...
 def action_trace_append(variables, path): ...
-async def trace_action(hass: HomeAssistant, script_run: _ScriptRun, stop: asyncio.Event, variables: dict[str, Any]) -> AsyncGenerator[TraceElement, None]: ...
+async def trace_action(hass: HomeAssistant, script_run: _ScriptRun, stop: asyncio.Future[None], variables: dict[str, Any]) -> AsyncGenerator[TraceElement, None]: ...
 def make_script_schema(schema: Mapping[Any, Any], default_script_mode: str, extra: int = ...) -> vol.Schema: ...
 
 STATIC_VALIDATION_ACTION_TYPES: Incomplete
@@ -78,6 +83,7 @@ class _ScriptRun:
     _context: Incomplete
     _log_exceptions: Incomplete
     _step: int
+    _started: bool
     _stop: Incomplete
     _stopped: Incomplete
     _conversation_response: Incomplete
@@ -92,9 +98,11 @@ class _ScriptRun:
     async def async_stop(self) -> None: ...
     def _handle_exception(self, exception: Exception, continue_on_error: bool, log_exceptions: bool) -> None: ...
     def _log_exception(self, exception) -> None: ...
-    def _get_pos_time_period_template(self, key): ...
+    def _get_pos_time_period_template(self, key: str) -> timedelta: ...
     async def _async_delay_step(self) -> None: ...
+    def _get_timeout_seconds_from_action(self) -> float | None: ...
     async def _async_wait_template_step(self) -> None: ...
+    def _async_set_remaining_time_var(self, timeout_handle: asyncio.TimerHandle | None) -> None: ...
     async def _async_run_long_action(self, long_task: asyncio.Task[_T]) -> _T | None: ...
     async def _async_call_service_step(self) -> None: ...
     async def _async_device_step(self) -> None: ...
@@ -105,7 +113,9 @@ class _ScriptRun:
     async def _async_repeat_step(self) -> None: ...
     async def _async_choose_step(self) -> None: ...
     async def _async_if_step(self) -> None: ...
+    def _async_futures_with_timeout(self, timeout: float | None) -> tuple[list[asyncio.Future[None]], asyncio.TimerHandle | None, asyncio.Future[None] | None]: ...
     async def _async_wait_for_trigger_step(self) -> None: ...
+    async def _async_wait_with_optional_timeout(self, futures: list[asyncio.Future[None]], timeout_handle: asyncio.TimerHandle | None, timeout_future: asyncio.Future[None] | None, unsub: Callable[[], None]) -> None: ...
     async def _async_variables_step(self) -> None: ...
     async def _async_set_conversation_response_step(self) -> None: ...
     async def _async_stop_step(self) -> None: ...
@@ -183,9 +193,13 @@ class Script:
     @property
     def supports_max(self) -> bool: ...
     @cached_property
+    def referenced_labels(self) -> set[str]: ...
+    @cached_property
+    def referenced_floors(self) -> set[str]: ...
+    @cached_property
     def referenced_areas(self) -> set[str]: ...
     @staticmethod
-    def _find_referenced_areas(referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...
+    def _find_referenced_target(target: Literal['area_id', 'floor_id', 'label_id'], referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...
     @cached_property
     def referenced_devices(self) -> set[str]: ...
     @staticmethod

homeassistant-stubs/helpers/selector.pyi

@@ -241,6 +241,17 @@ class EntitySelector(Selector[EntitySelectorConfig]):
     def __init__(self, config: EntitySelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str | list[str]: ...
 
+class FloorSelectorConfig(TypedDict, total=False):
+    entity: EntityFilterSelectorConfig | list[EntityFilterSelectorConfig]
+    device: DeviceFilterSelectorConfig | list[DeviceFilterSelectorConfig]
+    multiple: bool
+
+class FloorSelector(Selector[AreaSelectorConfig]):
+    selector_type: str
+    CONFIG_SCHEMA: Incomplete
+    def __init__(self, config: FloorSelectorConfig | None = None) -> None: ...
+    def __call__(self, data: Any) -> str | list[str]: ...
+
 class IconSelectorConfig(TypedDict, total=False):
     placeholder: str
 
@@ -250,6 +261,15 @@ class IconSelector(Selector[IconSelectorConfig]):
     def __init__(self, config: IconSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
+class LabelSelectorConfig(TypedDict, total=False):
+    multiple: bool
+
+class LabelSelector(Selector[LabelSelectorConfig]):
+    selector_type: str
+    CONFIG_SCHEMA: Incomplete
+    def __init__(self, config: LabelSelectorConfig | None = None) -> None: ...
+    def __call__(self, data: Any) -> str | list[str]: ...
+
 class LanguageSelectorConfig(TypedDict, total=False):
     languages: list[str]
     native_name: bool

homeassistant-stubs/helpers/service.pyi

@@ -1,6 +1,6 @@
 import dataclasses
 import voluptuous as vol
-from . import area_registry as area_registry, device_registry as device_registry, entity_registry as entity_registry, template as template, translation as translation
+from . import area_registry as area_registry, device_registry as device_registry, entity_registry as entity_registry, floor_registry as floor_registry, label_registry as label_registry, template as template, translation as translation
 from .entity import Entity as Entity
 from .group import expand_entity_ids as expand_entity_ids
 from .selector import TargetSelector as TargetSelector
@@ -8,7 +8,7 @@ from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsTy
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Iterable
 from homeassistant.auth.permissions.const import CAT_ENTITIES as CAT_ENTITIES, POLICY_CONTROL as POLICY_CONTROL
-from homeassistant.const import ATTR_AREA_ID as ATTR_AREA_ID, ATTR_DEVICE_ID as ATTR_DEVICE_ID, ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_SERVICE as CONF_SERVICE, CONF_SERVICE_DATA as CONF_SERVICE_DATA, CONF_SERVICE_DATA_TEMPLATE as CONF_SERVICE_DATA_TEMPLATE, CONF_SERVICE_TEMPLATE as CONF_SERVICE_TEMPLATE, CONF_TARGET as CONF_TARGET, ENTITY_MATCH_ALL as ENTITY_MATCH_ALL, ENTITY_MATCH_NONE as ENTITY_MATCH_NONE
+from homeassistant.const import ATTR_AREA_ID as ATTR_AREA_ID, ATTR_DEVICE_ID as ATTR_DEVICE_ID, ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_FLOOR_ID as ATTR_FLOOR_ID, ATTR_LABEL_ID as ATTR_LABEL_ID, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_SERVICE as CONF_SERVICE, CONF_SERVICE_DATA as CONF_SERVICE_DATA, CONF_SERVICE_DATA_TEMPLATE as CONF_SERVICE_DATA_TEMPLATE, CONF_SERVICE_TEMPLATE as CONF_SERVICE_TEMPLATE, CONF_TARGET as CONF_TARGET, ENTITY_MATCH_ALL as ENTITY_MATCH_ALL, ENTITY_MATCH_NONE as ENTITY_MATCH_NONE
 from homeassistant.core import Context as Context, EntityServiceResponse as EntityServiceResponse, HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, TemplateError as TemplateError, Unauthorized as Unauthorized, UnknownUser as UnknownUser
 from homeassistant.loader import Integration as Integration, async_get_integrations as async_get_integrations, bind_hass as bind_hass
@@ -44,6 +44,8 @@ class ServiceTargetSelector:
     entity_ids: Incomplete
     device_ids: Incomplete
     area_ids: Incomplete
+    floor_ids: Incomplete
+    label_ids: Incomplete
     def __init__(self, service_call: ServiceCall) -> None: ...
     @property
     def has_any_selector(self) -> bool: ...
@@ -54,9 +56,12 @@ class SelectedEntities:
     indirectly_referenced: set[str] = ...
     missing_devices: set[str] = ...
     missing_areas: set[str] = ...
+    missing_floors: set[str] = ...
+    missing_labels: set[str] = ...
     referenced_devices: set[str] = ...
+    referenced_areas: set[str] = ...
     def log_missing(self, missing_entities: set[str]) -> None: ...
-    def __init__(self, referenced, indirectly_referenced, missing_devices, missing_areas, referenced_devices) -> None: ...
+    def __init__(self, referenced, indirectly_referenced, missing_devices, missing_areas, missing_floors, missing_labels, referenced_devices, referenced_areas) -> None: ...
 
 def call_from_config(hass: HomeAssistant, config: ConfigType, blocking: bool = False, variables: TemplateVarsType = None, validate_config: bool = True) -> None: ...
 async def async_call_from_config(hass: HomeAssistant, config: ConfigType, blocking: bool = False, variables: TemplateVarsType = None, validate_config: bool = True, context: Context | None = None) -> None: ...
@@ -75,7 +80,7 @@ def async_set_service_schema(hass: HomeAssistant, domain: str, service: str, sch
 def _get_permissible_entity_candidates(call: ServiceCall, entities: dict[str, Entity], entity_perms: None | Callable[[str, str], bool], target_all_entities: bool, all_referenced: set[str] | None) -> list[Entity]: ...
 async def entity_service_call(hass: HomeAssistant, registered_entities: dict[str, Entity], func: str | HassJob, call: ServiceCall, required_features: Iterable[int] | None = None) -> EntityServiceResponse | None: ...
 async def _handle_entity_call(hass: HomeAssistant, entity: Entity, func: str | HassJob, data: dict | ServiceCall, context: Context) -> ServiceResponse: ...
-async def _async_admin_handler(hass: HomeAssistant, service_job: HassJob[[None], Callable[[ServiceCall], Awaitable[None] | None]], call: ServiceCall) -> None: ...
+async def _async_admin_handler(hass: HomeAssistant, service_job: HassJob[[ServiceCall], Awaitable[None] | None], call: ServiceCall) -> None: ...
 def async_register_admin_service(hass: HomeAssistant, domain: str, service: str, service_func: Callable[[ServiceCall], Awaitable[None] | None], schema: vol.Schema = ...) -> None: ...
 def verify_domain_control(hass: HomeAssistant, domain: str) -> Callable[[Callable[[ServiceCall], Any]], Callable[[ServiceCall], Any]]: ...
 

homeassistant-stubs/helpers/storage.pyi

@@ -1,20 +1,43 @@
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Mapping, Sequence
+from collections.abc import Callable as Callable, Iterable, Mapping, Sequence
 from functools import cached_property as cached_property
-from homeassistant.const import EVENT_HOMEASSISTANT_FINAL_WRITE as EVENT_HOMEASSISTANT_FINAL_WRITE
+from homeassistant.const import EVENT_HOMEASSISTANT_FINAL_WRITE as EVENT_HOMEASSISTANT_FINAL_WRITE, EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, CoreState as CoreState, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
-from homeassistant.loader import MAX_LOAD_CONCURRENTLY as MAX_LOAD_CONCURRENTLY, bind_hass as bind_hass
+from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.util import json as json_util
 from homeassistant.util.file import WriteError as WriteError
 from json import JSONEncoder
 from typing import Any, Generic, TypeVar
 
+MAX_LOAD_CONCURRENTLY: int
 STORAGE_DIR: str
 _LOGGER: Incomplete
 STORAGE_SEMAPHORE: str
+STORAGE_MANAGER: str
+MANAGER_CLEANUP_DELAY: int
 _T = TypeVar('_T', bound=Mapping[str, Any] | Sequence[Any])
 
 async def async_migrator(hass: HomeAssistant, old_path: str, store: Store[_T], *, old_conf_load_func: Callable | None = None, old_conf_migrate_func: Callable | None = None) -> _T | None: ...
+def get_internal_store_manager(hass: HomeAssistant, config_dir: str | None = None) -> _StoreManager: ...
+
+class _StoreManager:
+    _hass: Incomplete
+    _invalidated: Incomplete
+    _files: Incomplete
+    _data_preload: Incomplete
+    _storage_path: Incomplete
+    _cancel_cleanup: Incomplete
+    def __init__(self, hass: HomeAssistant, config_dir: str) -> None: ...
+    async def async_initialize(self) -> None: ...
+    def async_invalidate(self, key: str) -> None: ...
+    def async_fetch(self, key: str) -> tuple[bool, json_util.JsonValueType | None] | None: ...
+    def _async_schedule_cleanup(self, _event: Event) -> None: ...
+    def _async_cancel_and_cleanup(self, _event: Event) -> None: ...
+    def _async_cleanup(self) -> None: ...
+    async def async_preload(self, keys: Iterable[str]) -> None: ...
+    def _preload(self, keys: Iterable[str]) -> None: ...
+    def _initialize_files(self) -> None: ...
 
 class Store(Generic[_T]):
     version: Incomplete
@@ -31,7 +54,8 @@ class Store(Generic[_T]):
     _atomic_writes: Incomplete
     _read_only: Incomplete
     _next_write_time: float
-    def __init__(self, hass: HomeAssistant, version: int, key: str, private: bool = False, *, atomic_writes: bool = False, encoder: type[JSONEncoder] | None = None, minor_version: int = 1, read_only: bool = False) -> None: ...
+    _manager: Incomplete
+    def __init__(self, hass: HomeAssistant, version: int, key: str, private: bool = False, *, atomic_writes: bool = False, encoder: type[JSONEncoder] | None = None, minor_version: int = 1, read_only: bool = False, config_dir: str | None = None) -> None: ...
     @cached_property
     def path(self): ...
     async def async_load(self) -> _T | None: ...

homeassistant-stubs/helpers/system_info.pyi

@@ -1,3 +1,4 @@
+from .importlib import async_import_module as async_import_module
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.loader import bind_hass as bind_hass

homeassistant-stubs/helpers/template.pyi

@@ -1,5 +1,5 @@
 import jinja2
-from . import area_registry as area_registry, device_registry as device_registry, entity_registry as entity_registry
+from . import area_registry as area_registry, device_registry as device_registry, entity_registry as entity_registry, issue_registry as issue_registry, label_registry as label_registry
 from .singleton import singleton as singleton
 from .translation import async_translate_state as async_translate_state
 from .typing import TemplateVarsType as TemplateVarsType
@@ -38,8 +38,8 @@ _COLLECTABLE_STATE_ATTRIBUTES: Incomplete
 _T = TypeVar('_T')
 _R = TypeVar('_R')
 _P = ParamSpec('_P')
-ALL_STATES_RATE_LIMIT: Incomplete
-DOMAIN_STATES_RATE_LIMIT: Incomplete
+ALL_STATES_RATE_LIMIT: int
+DOMAIN_STATES_RATE_LIMIT: int
 _render_info: ContextVar[RenderInfo | None]
 template_cv: ContextVar[tuple[str, str] | None]
 CACHED_TEMPLATE_STATES: int
@@ -189,6 +189,8 @@ class TemplateStateBase(State):
     @property
     def last_changed(self) -> datetime: ...
     @property
+    def last_reported(self) -> datetime: ...
+    @property
     def last_updated(self) -> datetime: ...
     @property
     def context(self) -> Context: ...
@@ -235,12 +237,25 @@ def config_entry_id(hass: HomeAssistant, entity_id: str) -> str | None: ...
 def device_id(hass: HomeAssistant, entity_id_or_device_name: str) -> str | None: ...
 def device_attr(hass: HomeAssistant, device_or_entity_id: str, attr_name: str) -> Any: ...
 def is_device_attr(hass: HomeAssistant, device_or_entity_id: str, attr_name: str, attr_value: Any) -> bool: ...
+def issues(hass: HomeAssistant) -> dict[tuple[str, str], dict[str, Any]]: ...
+def issue(hass: HomeAssistant, domain: str, issue_id: str) -> dict[str, Any] | None: ...
+def floors(hass: HomeAssistant) -> Iterable[str | None]: ...
+def floor_id(hass: HomeAssistant, lookup_value: Any) -> str | None: ...
+def floor_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
+def floor_areas(hass: HomeAssistant, floor_id_or_name: str) -> Iterable[str]: ...
 def areas(hass: HomeAssistant) -> Iterable[str | None]: ...
 def area_id(hass: HomeAssistant, lookup_value: str) -> str | None: ...
 def _get_area_name(area_reg: area_registry.AreaRegistry, valid_area_id: str) -> str: ...
 def area_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
 def area_entities(hass: HomeAssistant, area_id_or_name: str) -> Iterable[str]: ...
 def area_devices(hass: HomeAssistant, area_id_or_name: str) -> Iterable[str]: ...
+def labels(hass: HomeAssistant, lookup_value: Any = None) -> Iterable[str | None]: ...
+def label_id(hass: HomeAssistant, lookup_value: Any) -> str | None: ...
+def label_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
+def _label_id_or_name(hass: HomeAssistant, label_id_or_name: str) -> str | None: ...
+def label_areas(hass: HomeAssistant, label_id_or_name: str) -> Iterable[str]: ...
+def label_devices(hass: HomeAssistant, label_id_or_name: str) -> Iterable[str]: ...
+def label_entities(hass: HomeAssistant, label_id_or_name: str) -> Iterable[str]: ...
 def closest(hass, *args): ...
 def closest_filter(hass, *args): ...
 def distance(hass, *args): ...

homeassistant-stubs/helpers/translation.pyi

@@ -1,7 +1,7 @@
 from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.const import EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
-from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, async_get_hass as async_get_hass, callback as callback
 from homeassistant.loader import Integration as Integration, async_get_config_flows as async_get_config_flows, async_get_integrations as async_get_integrations, bind_hass as bind_hass
 from homeassistant.util.json import load_json as load_json
 from typing import Any
@@ -42,4 +42,5 @@ def _async_get_components(hass: HomeAssistant, category: str) -> set[str]: ...
 def async_setup(hass: HomeAssistant) -> None: ...
 async def async_load_integrations(hass: HomeAssistant, integrations: set[str]) -> None: ...
 def async_translations_loaded(hass: HomeAssistant, components: set[str]) -> bool: ...
+def async_get_exception_message(translation_domain: str, translation_key: str, translation_placeholders: dict[str, str] | None = None) -> str: ...
 def async_translate_state(hass: HomeAssistant, state: str, domain: str, platform: str | None, translation_key: str | None, device_class: str | None) -> str: ...

homeassistant-stubs/helpers/trigger.pyi

@@ -7,6 +7,7 @@ from homeassistant.const import CONF_ALIAS as CONF_ALIAS, CONF_ENABLED as CONF_E
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Context as Context, HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback, is_callback as is_callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.loader import IntegrationNotFound as IntegrationNotFound, async_get_integration as async_get_integration
+from homeassistant.util.async_ import create_eager_task as create_eager_task
 from typing import Any, Protocol, TypedDict
 
 _PLATFORM_ALIASES: Incomplete

homeassistant-stubs/helpers/typing.pyi

@@ -1,8 +1,7 @@
 import homeassistant
-import homeassistant.core
 from _typeshed import Incomplete
 from enum import Enum
-from typing import Any, Generic, TypeVar
+from typing import Any, TypeVar
 
 _DataT = TypeVar('_DataT')
 GPSType = tuple[float, float]
@@ -18,8 +17,6 @@ class UndefinedType(Enum):
     _singleton: int
 
 UNDEFINED: Incomplete
+EventType = homeassistant.core.Event
 HomeAssistantType = homeassistant.core.HomeAssistant
 ServiceCallType = homeassistant.core.ServiceCall
-
-class EventType(homeassistant.core.Event, Generic[_DataT]):
-    data: _DataT

homeassistant-stubs/helpers/update_coordinator.pyi

@@ -8,16 +8,17 @@ from collections.abc import Awaitable, Callable as Callable, Coroutine, Generato
 from datetime import datetime, timedelta
 from homeassistant import config_entries as config_entries
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryError as ConfigEntryError, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.util.dt import utcnow as utcnow
-from typing import Any, Generic, Protocol, TypeVar
+from typing import Any, Generic, Protocol
+from typing_extensions import TypeVar
 
 REQUEST_REFRESH_DEFAULT_COOLDOWN: int
 REQUEST_REFRESH_DEFAULT_IMMEDIATE: bool
-_DataT = TypeVar('_DataT')
+_DataT = TypeVar('_DataT', default=dict[str, Any])
 _BaseDataUpdateCoordinatorT = TypeVar('_BaseDataUpdateCoordinatorT', bound='BaseDataUpdateCoordinatorProtocol')
-_DataUpdateCoordinatorT = TypeVar('_DataUpdateCoordinatorT', bound='DataUpdateCoordinator[Any]')
+_DataUpdateCoordinatorT = TypeVar('_DataUpdateCoordinatorT', bound='DataUpdateCoordinator[Any]', default='DataUpdateCoordinator[dict[str, Any]]')
 
 class UpdateFailed(Exception): ...
 
@@ -36,7 +37,6 @@ class DataUpdateCoordinator(BaseDataUpdateCoordinatorProtocol, Generic[_DataT]):
     data: Incomplete
     _microsecond: Incomplete
     _listeners: Incomplete
-    _job: Incomplete
     _unsub_refresh: Incomplete
     _unsub_shutdown: Incomplete
     _request_refresh_task: Incomplete