homeassistant-stubs 2024.4.3__py3-none-any.whl → 2024.5.0__py3-none-any.whl

homeassistant-stubs/helpers/entity_registry.pyi

@@ -1,25 +1,26 @@
 from . import storage as storage
-from .device_registry import EVENT_DEVICE_REGISTRY_UPDATED as EVENT_DEVICE_REGISTRY_UPDATED
+from .device_registry import EVENT_DEVICE_REGISTRY_UPDATED as EVENT_DEVICE_REGISTRY_UPDATED, EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
 from .json import JSON_DUMP as JSON_DUMP, find_paths_unserializable_data as find_paths_unserializable_data, json_bytes as json_bytes, json_fragment as json_fragment
-from .registry import BaseRegistry as BaseRegistry
+from .registry import BaseRegistry as BaseRegistry, BaseRegistryItems as BaseRegistryItems
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
-from collections import UserDict
-from collections.abc import Callable as Callable, Iterable, KeysView, Mapping, ValuesView
+from collections.abc import Callable as Callable, Container, Hashable, KeysView, Mapping
 from enum import StrEnum
-from homeassistant.backports.functools import cached_property as cached_property
+from functools import cached_property as cached_property
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, ATTR_ICON as ATTR_ICON, ATTR_RESTORED as ATTR_RESTORED, ATTR_SUPPORTED_FEATURES as ATTR_SUPPORTED_FEATURES, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, EVENT_HOMEASSISTANT_START as EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, EntityCategory as EntityCategory, MAX_LENGTH_STATE_DOMAIN as MAX_LENGTH_STATE_DOMAIN, MAX_LENGTH_STATE_ENTITY_ID as MAX_LENGTH_STATE_ENTITY_ID, Platform as Platform, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback, split_entity_id as split_entity_id, valid_entity_id as valid_entity_id
 from homeassistant.exceptions import MaxLengthExceeded as MaxLengthExceeded
+from homeassistant.loader import async_suggest_report_issue as async_suggest_report_issue
 from homeassistant.util import slugify as slugify
+from homeassistant.util.event_type import EventType as EventType
 from homeassistant.util.json import format_unserializable_data as format_unserializable_data
 from homeassistant.util.read_only_dict import ReadOnlyDict as ReadOnlyDict
 from typing import Any, Literal, NotRequired, TypeVar, TypedDict
 
 T = TypeVar('T')
 DATA_REGISTRY: str
-EVENT_ENTITY_REGISTRY_UPDATED: str
+EVENT_ENTITY_REGISTRY_UPDATED: EventType[EventEntityRegistryUpdatedData]
 _LOGGER: Incomplete
 STORAGE_VERSION_MAJOR: int
 STORAGE_VERSION_MINOR: int
@@ -93,10 +94,15 @@ class RegistryEntry:
     def hidden(self) -> bool: ...
     @property
     def _as_display_dict(self) -> dict[str, Any] | None: ...
+    @cached_property
     def display_json_repr(self) -> bytes | None: ...
+    @cached_property
     def as_partial_dict(self) -> dict[str, Any]: ...
+    @cached_property
     def extended_dict(self) -> dict[str, Any]: ...
+    @cached_property
     def partial_json_repr(self) -> bytes | None: ...
+    @cached_property
     def as_storage_fragment(self) -> json_fragment: ...
     def write_unavailable_state(self, hass: HomeAssistant) -> None: ...
     def __init__(self, entity_id, unique_id, platform, previous_unique_id, aliases, area_id, categories, capabilities, config_entry_id, device_class, device_id, domain, disabled_by, entity_category, hidden_by, icon, id, has_entity_name, labels, name, options, original_device_class, original_icon, original_name, supported_features, translation_key, unit_of_measurement) -> None: ...
@@ -114,6 +120,7 @@ class DeletedRegistryEntry:
     id: str
     orphaned_timestamp: float | None
     def _domain_default(self) -> str: ...
+    @cached_property
     def as_storage_fragment(self) -> json_fragment: ...
     def __init__(self, entity_id, unique_id, platform, config_entry_id, domain, id, orphaned_timestamp) -> None: ...
     def __lt__(self, other): ...
@@ -124,24 +131,25 @@ class DeletedRegistryEntry:
 class EntityRegistryStore(storage.Store[dict[str, list[dict[str, Any]]]]):
     async def _async_migrate_func(self, old_major_version: int, old_minor_version: int, old_data: dict[str, list[dict[str, Any]]]) -> dict: ...
 
-class EntityRegistryItems(UserDict[str, RegistryEntry]):
+class EntityRegistryItems(BaseRegistryItems[RegistryEntry]):
     _entry_ids: Incomplete
     _index: Incomplete
     _config_entry_id_index: Incomplete
     _device_id_index: Incomplete
     _area_id_index: Incomplete
+    _labels_index: Incomplete
     def __init__(self) -> None: ...
-    def values(self) -> ValuesView[RegistryEntry]: ...
-    def __setitem__(self, key: str, entry: RegistryEntry) -> None: ...
-    def _unindex_entry_value(self, key: str, value: str, index: dict[str, dict[str, Literal[True]]]) -> None: ...
-    def _unindex_entry(self, key: str) -> None: ...
-    def __delitem__(self, key: str) -> None: ...
+    def _index_entry(self, key: str, entry: RegistryEntry) -> None: ...
+    def _unindex_entry(self, key: str, replacement_entry: RegistryEntry | None = None) -> None: ...
     def get_device_ids(self) -> KeysView[str]: ...
     def get_entity_id(self, key: tuple[str, str, str]) -> str | None: ...
     def get_entry(self, key: str) -> RegistryEntry | None: ...
     def get_entries_for_device_id(self, device_id: str, include_disabled_entities: bool = False) -> list[RegistryEntry]: ...
     def get_entries_for_config_entry_id(self, config_entry_id: str) -> list[RegistryEntry]: ...
     def get_entries_for_area_id(self, area_id: str) -> list[RegistryEntry]: ...
+    def get_entries_for_label(self, label: str) -> list[RegistryEntry]: ...
+
+def _validate_item(hass: HomeAssistant, domain: str, platform: str, unique_id: str | Hashable | UndefinedType | Any, *, disabled_by: RegistryEntryDisabler | None | UndefinedType = None, entity_category: EntityCategory | None | UndefinedType = None, hidden_by: RegistryEntryHider | None | UndefinedType = None) -> None: ...
 
 class EntityRegistry(BaseRegistry):
     deleted_entities: dict[tuple[str, str, str], DeletedRegistryEntry]
@@ -154,11 +162,11 @@ class EntityRegistry(BaseRegistry):
     def async_get(self, entity_id_or_uuid: str) -> RegistryEntry | None: ...
     def async_get_entity_id(self, domain: str, platform: str, unique_id: str) -> str | None: ...
     def async_device_ids(self) -> list[str]: ...
-    def _entity_id_available(self, entity_id: str, known_object_ids: Iterable[str] | None) -> bool: ...
-    def async_generate_entity_id(self, domain: str, suggested_object_id: str, known_object_ids: Iterable[str] | None = None) -> str: ...
-    def async_get_or_create(self, domain: str, platform: str, unique_id: str, *, known_object_ids: Iterable[str] | None = None, suggested_object_id: str | None = None, disabled_by: RegistryEntryDisabler | None = None, hidden_by: RegistryEntryHider | None = None, get_initial_options: Callable[[], EntityOptionsType | None] | None = None, capabilities: Mapping[str, Any] | None | UndefinedType = ..., config_entry: ConfigEntry | None | UndefinedType = ..., device_id: str | None | UndefinedType = ..., entity_category: EntityCategory | UndefinedType | None = ..., has_entity_name: bool | UndefinedType = ..., original_device_class: str | None | UndefinedType = ..., original_icon: str | None | UndefinedType = ..., original_name: str | None | UndefinedType = ..., supported_features: int | None | UndefinedType = ..., translation_key: str | None | UndefinedType = ..., unit_of_measurement: str | None | UndefinedType = ...) -> RegistryEntry: ...
+    def _entity_id_available(self, entity_id: str, known_object_ids: Container[str] | None) -> bool: ...
+    def async_generate_entity_id(self, domain: str, suggested_object_id: str, known_object_ids: Container[str] | None = None) -> str: ...
+    def async_get_or_create(self, domain: str, platform: str, unique_id: str, *, known_object_ids: Container[str] | None = None, suggested_object_id: str | None = None, disabled_by: RegistryEntryDisabler | None = None, hidden_by: RegistryEntryHider | None = None, get_initial_options: Callable[[], EntityOptionsType | None] | None = None, capabilities: Mapping[str, Any] | None | UndefinedType = ..., config_entry: ConfigEntry | None | UndefinedType = ..., device_id: str | None | UndefinedType = ..., entity_category: EntityCategory | UndefinedType | None = ..., has_entity_name: bool | UndefinedType = ..., original_device_class: str | None | UndefinedType = ..., original_icon: str | None | UndefinedType = ..., original_name: str | None | UndefinedType = ..., supported_features: int | None | UndefinedType = ..., translation_key: str | None | UndefinedType = ..., unit_of_measurement: str | None | UndefinedType = ...) -> RegistryEntry: ...
     def async_remove(self, entity_id: str) -> None: ...
-    def async_device_modified(self, event: Event) -> None: ...
+    def async_device_modified(self, event: Event[EventDeviceRegistryUpdatedData]) -> None: ...
     def _async_update_entity(self, entity_id: str, *, aliases: set[str] | UndefinedType = ..., area_id: str | None | UndefinedType = ..., categories: dict[str, str] | UndefinedType = ..., capabilities: Mapping[str, Any] | None | UndefinedType = ..., config_entry_id: str | None | UndefinedType = ..., device_class: str | None | UndefinedType = ..., device_id: str | None | UndefinedType = ..., disabled_by: RegistryEntryDisabler | None | UndefinedType = ..., entity_category: EntityCategory | None | UndefinedType = ..., hidden_by: RegistryEntryHider | None | UndefinedType = ..., icon: str | None | UndefinedType = ..., has_entity_name: bool | UndefinedType = ..., labels: set[str] | UndefinedType = ..., name: str | None | UndefinedType = ..., new_entity_id: str | UndefinedType = ..., new_unique_id: str | UndefinedType = ..., options: EntityOptionsType | UndefinedType = ..., original_device_class: str | None | UndefinedType = ..., original_icon: str | None | UndefinedType = ..., original_name: str | None | UndefinedType = ..., platform: str | None | UndefinedType = ..., supported_features: int | UndefinedType = ..., translation_key: str | None | UndefinedType = ..., unit_of_measurement: str | None | UndefinedType = ...) -> RegistryEntry: ...
     def async_update_entity(self, entity_id: str, *, aliases: set[str] | UndefinedType = ..., area_id: str | None | UndefinedType = ..., categories: dict[str, str] | UndefinedType = ..., capabilities: Mapping[str, Any] | None | UndefinedType = ..., config_entry_id: str | None | UndefinedType = ..., device_class: str | None | UndefinedType = ..., device_id: str | None | UndefinedType = ..., disabled_by: RegistryEntryDisabler | None | UndefinedType = ..., entity_category: EntityCategory | None | UndefinedType = ..., hidden_by: RegistryEntryHider | None | UndefinedType = ..., icon: str | None | UndefinedType = ..., has_entity_name: bool | UndefinedType = ..., labels: set[str] | UndefinedType = ..., name: str | None | UndefinedType = ..., new_entity_id: str | UndefinedType = ..., new_unique_id: str | UndefinedType = ..., original_device_class: str | None | UndefinedType = ..., original_icon: str | None | UndefinedType = ..., original_name: str | None | UndefinedType = ..., supported_features: int | UndefinedType = ..., translation_key: str | None | UndefinedType = ..., unit_of_measurement: str | None | UndefinedType = ...) -> RegistryEntry: ...
     def async_update_entity_platform(self, entity_id: str, new_platform: str, *, new_config_entry_id: str | UndefinedType = ..., new_unique_id: str | UndefinedType = ..., new_device_id: str | None | UndefinedType = ...) -> RegistryEntry: ...

homeassistant-stubs/helpers/event.pyi

@@ -1,5 +1,6 @@
 import asyncio
 import time
+from . import frame as frame
 from .device_registry import EVENT_DEVICE_REGISTRY_UPDATED as EVENT_DEVICE_REGISTRY_UPDATED, EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
 from .entity_registry import EVENT_ENTITY_REGISTRY_UPDATED as EVENT_ENTITY_REGISTRY_UPDATED, EventEntityRegistryUpdatedData as EventEntityRegistryUpdatedData
 from .ratelimit import KeyedRateLimit as KeyedRateLimit
@@ -11,11 +12,12 @@ from collections.abc import Callable, Coroutine, Iterable, Mapping, Sequence
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from homeassistant.const import EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE, EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, MATCH_ALL as MATCH_ALL, SUN_EVENT_SUNRISE as SUN_EVENT_SUNRISE, SUN_EVENT_SUNSET as SUN_EVENT_SUNSET
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
 from homeassistant.exceptions import TemplateError as TemplateError
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.async_ import run_callback_threadsafe as run_callback_threadsafe
-from typing import Any, Concatenate, Generic, ParamSpec, TypeVar, TypedDict
+from homeassistant.util.event_type import EventType as EventType
+from typing import Any, Concatenate, Generic, ParamSpec, TypeVar
 
 TRACK_STATE_CHANGE_CALLBACKS: str
 TRACK_STATE_CHANGE_LISTENER: str
@@ -40,11 +42,10 @@ _P = ParamSpec('_P')
 class _KeyedEventTracker(Generic[_TypedDictT]):
     listeners_key: str
     callbacks_key: str
-    event_type: str
+    event_type: EventType[_TypedDictT] | str
     dispatcher_callable: Callable[[HomeAssistant, dict[str, list[HassJob[[Event[_TypedDictT]], Any]]], Event[_TypedDictT]], None]
     filter_callable: Callable[[HomeAssistant, dict[str, list[HassJob[[Event[_TypedDictT]], Any]]], _TypedDictT], bool]
-    run_immediately: bool
-    def __init__(self, listeners_key, callbacks_key, event_type, dispatcher_callable, filter_callable, run_immediately) -> None: ...
+    def __init__(self, listeners_key, callbacks_key, event_type, dispatcher_callable, filter_callable) -> None: ...
 
 @dataclass(slots=True)
 class TrackStates:
@@ -67,11 +68,6 @@ class TrackTemplateResult:
     result: Any
     def __init__(self, template, last_result, result) -> None: ...
 
-class EventStateChangedData(TypedDict):
-    entity_id: str
-    old_state: State | None
-    new_state: State | None
-
 def threaded_listener_factory(async_factory: Callable[Concatenate[HomeAssistant, _P], Any]) -> Callable[Concatenate[HomeAssistant, _P], CALLBACK_TYPE]: ...
 def async_track_state_change(hass: HomeAssistant, entity_ids: str | Iterable[str], action: Callable[[str, State | None, State | None], Coroutine[Any, Any, None] | None], from_state: None | str | Iterable[str] = None, to_state: None | str | Iterable[str] = None) -> CALLBACK_TYPE: ...
 
@@ -213,23 +209,20 @@ def async_track_time_interval(hass: HomeAssistant, action: Callable[[datetime],
 
 track_time_interval: Incomplete
 
+@dataclass(slots=True)
 class SunListener:
     hass: HomeAssistant
     job: HassJob[[], Coroutine[Any, Any, None] | None]
     event: str
     offset: timedelta | None
-    _unsub_sun: CALLBACK_TYPE | None
-    _unsub_config: CALLBACK_TYPE | None
+    _unsub_sun: CALLBACK_TYPE | None = ...
+    _unsub_config: CALLBACK_TYPE | None = ...
     def async_attach(self) -> None: ...
     def async_detach(self) -> None: ...
     def _listen_next_sun_event(self) -> None: ...
     def _handle_sun_event(self, _now: Any) -> None: ...
     def _handle_config_event(self, _event: Any) -> None: ...
-    def __init__(self, hass, job, event, offset, unsub_sun, unsub_config) -> None: ...
-    def __lt__(self, other): ...
-    def __le__(self, other): ...
-    def __gt__(self, other): ...
-    def __ge__(self, other): ...
+    def __init__(self, hass, job, event, offset, _unsub_sun, _unsub_config) -> None: ...
 
 def async_track_sunrise(hass: HomeAssistant, action: Callable[[], None], offset: timedelta | None = None) -> CALLBACK_TYPE: ...
 

homeassistant-stubs/helpers/floor_registry.pyi

@@ -7,13 +7,24 @@ from collections.abc import Iterable
 from dataclasses import dataclass
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util import slugify as slugify
+from homeassistant.util.event_type import EventType as EventType
 from typing import Literal, TypedDict
 
 DATA_REGISTRY: str
-EVENT_FLOOR_REGISTRY_UPDATED: str
+EVENT_FLOOR_REGISTRY_UPDATED: EventType[EventFloorRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 
+class _FloorStoreData(TypedDict):
+    aliases: list[str]
+    floor_id: str
+    icon: str | None
+    level: int | None
+    name: str
+
+class FloorRegistryStoreData(TypedDict):
+    floors: list[_FloorStoreData]
+
 class EventFloorRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     floor_id: str
@@ -27,7 +38,7 @@ class FloorEntry(NormalizedNameBaseRegistryEntry):
     level: int | None = ...
     def __init__(self, *, name, normalized_name, aliases, floor_id, icon, level) -> None: ...
 
-class FloorRegistry(BaseRegistry):
+class FloorRegistry(BaseRegistry[FloorRegistryStoreData]):
     floors: NormalizedNameBaseRegistryItems[FloorEntry]
     _floor_data: dict[str, FloorEntry]
     hass: Incomplete
@@ -41,7 +52,7 @@ class FloorRegistry(BaseRegistry):
     def async_delete(self, floor_id: str) -> None: ...
     def async_update(self, floor_id: str, *, aliases: set[str] | UndefinedType = ..., icon: str | None | UndefinedType = ..., level: int | UndefinedType = ..., name: str | UndefinedType = ...) -> FloorEntry: ...
     async def async_load(self) -> None: ...
-    def _data_to_save(self) -> dict[str, list[dict[str, str | int | list[str] | None]]]: ...
+    def _data_to_save(self) -> FloorRegistryStoreData: ...
 
 def async_get(hass: HomeAssistant) -> FloorRegistry: ...
 async def async_load(hass: HomeAssistant) -> None: ...

homeassistant-stubs/helpers/frame.pyi

@@ -34,6 +34,6 @@ def get_integration_frame(exclude_integrations: set | None = None) -> Integratio
 
 class MissingIntegrationFrame(HomeAssistantError): ...
 
-def report(what: str, exclude_integrations: set | None = None, error_if_core: bool = True, level: int = ..., log_custom_component_only: bool = False) -> None: ...
-def _report_integration(what: str, integration_frame: IntegrationFrame, level: int = ...) -> None: ...
+def report(what: str, exclude_integrations: set | None = None, error_if_core: bool = True, level: int = ..., log_custom_component_only: bool = False, error_if_integration: bool = False) -> None: ...
+def _report_integration(what: str, integration_frame: IntegrationFrame, level: int = ..., error: bool = False) -> None: ...
 def warn_use(func: _CallableT, what: str) -> _CallableT: ...

homeassistant-stubs/helpers/httpx_client.pyi

@@ -20,7 +20,7 @@ def get_async_client(hass: HomeAssistant, verify_ssl: bool = True) -> httpx.Asyn
 
 class HassHttpXAsyncClient(httpx.AsyncClient):
     async def __aenter__(self) -> Self: ...
-    async def __aexit__(self, *args: Any) -> None: ...
+    async def __aexit__(self, *args: object) -> None: ...
 
 def create_async_httpx_client(hass: HomeAssistant, verify_ssl: bool = True, auto_cleanup: bool = True, ssl_cipher_list: SSLCipherList = ..., **kwargs: Any) -> httpx.AsyncClient: ...
 def _async_register_async_client_shutdown(hass: HomeAssistant, client: httpx.AsyncClient, original_aclose: Callable[..., Any]) -> None: ...

homeassistant-stubs/helpers/icon.pyi

@@ -1,3 +1,4 @@
+import pathlib
 from .translation import build_resources as build_resources
 from _typeshed import Incomplete
 from collections.abc import Iterable
@@ -9,8 +10,8 @@ from typing import Any
 ICON_CACHE: str
 _LOGGER: Incomplete
 
-def _component_icons_path(component: str, integration: Integration) -> str | None: ...
-def _load_icons_files(icons_files: dict[str, str]) -> dict[str, dict[str, Any]]: ...
+def _component_icons_path(integration: Integration) -> pathlib.Path: ...
+def _load_icons_files(icons_files: dict[str, pathlib.Path]) -> dict[str, dict[str, Any]]: ...
 async def _async_get_component_icons(hass: HomeAssistant, components: set[str], integrations: dict[str, Integration]) -> dict[str, Any]: ...
 
 class _IconsCache:

homeassistant-stubs/helpers/intent.pyi

@@ -13,11 +13,10 @@ from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_ENT
 from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant, State as State, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.loader import bind_hass as bind_hass
-from typing import Any, TypeVar
+from typing import Any
 
 _LOGGER: Incomplete
 _SlotsType = dict[str, Any]
-_T = TypeVar('_T')
 INTENT_TURN_OFF: str
 INTENT_TURN_ON: str
 INTENT_TOGGLE: str

homeassistant-stubs/helpers/label_registry.pyi

@@ -7,13 +7,24 @@ from collections.abc import Iterable
 from dataclasses import dataclass
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util import slugify as slugify
+from homeassistant.util.event_type import EventType as EventType
 from typing import Literal, TypedDict
 
 DATA_REGISTRY: str
-EVENT_LABEL_REGISTRY_UPDATED: str
+EVENT_LABEL_REGISTRY_UPDATED: EventType[EventLabelRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 
+class _LabelStoreData(TypedDict):
+    color: str | None
+    description: str | None
+    icon: str | None
+    label_id: str
+    name: str
+
+class LabelRegistryStoreData(TypedDict):
+    labels: list[_LabelStoreData]
+
 class EventLabelRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     label_id: str
@@ -27,7 +38,7 @@ class LabelEntry(NormalizedNameBaseRegistryEntry):
     icon: str | None = ...
     def __init__(self, *, name, normalized_name, label_id, description, color, icon) -> None: ...
 
-class LabelRegistry(BaseRegistry):
+class LabelRegistry(BaseRegistry[LabelRegistryStoreData]):
     labels: NormalizedNameBaseRegistryItems[LabelEntry]
     _label_data: dict[str, LabelEntry]
     hass: Incomplete
@@ -41,7 +52,7 @@ class LabelRegistry(BaseRegistry):
     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 _data_to_save(self) -> dict[str, list[dict[str, str | None]]]: ...
+    def _data_to_save(self) -> LabelRegistryStoreData: ...
 
 def async_get(hass: HomeAssistant) -> LabelRegistry: ...
 async def async_load(hass: HomeAssistant) -> None: ...

homeassistant-stubs/helpers/network.pyi

@@ -14,9 +14,9 @@ class NoURLAvailableError(HomeAssistantError): ...
 def is_internal_request(hass: HomeAssistant) -> bool: ...
 def get_supervisor_network_url(hass: HomeAssistant, *, allow_ssl: bool = False) -> str | None: ...
 def is_hass_url(hass: HomeAssistant, url: str) -> bool: ...
-def get_url(hass: HomeAssistant, *, require_current_request: bool = False, require_ssl: bool = False, require_standard_port: bool = False, allow_internal: bool = True, allow_external: bool = True, allow_cloud: bool = True, allow_ip: bool | None = None, prefer_external: bool | None = None, prefer_cloud: bool = False) -> str: ...
+def get_url(hass: HomeAssistant, *, require_current_request: bool = False, require_ssl: bool = False, require_standard_port: bool = False, require_cloud: bool = False, allow_internal: bool = True, allow_external: bool = True, allow_cloud: bool = True, allow_ip: bool | None = None, prefer_external: bool | None = None, prefer_cloud: bool = False) -> str: ...
 def _get_request_host() -> str | None: ...
 def _get_internal_url(hass: HomeAssistant, *, allow_ip: bool = True, require_current_request: bool = False, require_ssl: bool = False, require_standard_port: bool = False) -> str: ...
-def _get_external_url(hass: HomeAssistant, *, allow_cloud: bool = True, allow_ip: bool = True, prefer_cloud: bool = False, require_current_request: bool = False, require_ssl: bool = False, require_standard_port: bool = False) -> str: ...
+def _get_external_url(hass: HomeAssistant, *, allow_cloud: bool = True, allow_ip: bool = True, prefer_cloud: bool = False, require_current_request: bool = False, require_ssl: bool = False, require_standard_port: bool = False, require_cloud: bool = False) -> str: ...
 def _get_cloud_url(hass: HomeAssistant, require_current_request: bool = False) -> str: ...
 def is_cloud_connection(hass: HomeAssistant) -> bool: ...

homeassistant-stubs/helpers/normalized_name_base_registry.pyi

@@ -1,6 +1,5 @@
+from .registry import BaseRegistryItems as BaseRegistryItems
 from _typeshed import Incomplete
-from collections import UserDict
-from collections.abc import ValuesView
 from dataclasses import dataclass
 from typing import TypeVar
 
@@ -13,10 +12,9 @@ _VT = TypeVar('_VT', bound=NormalizedNameBaseRegistryEntry)
 
 def normalize_name(name: str) -> str: ...
 
-class NormalizedNameBaseRegistryItems(UserDict[str, _VT]):
+class NormalizedNameBaseRegistryItems(BaseRegistryItems[_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 _unindex_entry(self, key: str, replacement_entry: _VT | None = None) -> None: ...
+    def _index_entry(self, key: str, entry: _VT) -> None: ...
     def get_by_name(self, name: str) -> _VT | None: ...

homeassistant-stubs/helpers/registry.pyi

@@ -1,15 +1,30 @@
 import abc
 from .storage import Store as Store
 from abc import ABC, abstractmethod
+from collections import UserDict
+from collections.abc import Mapping, Sequence, ValuesView
 from homeassistant.core import CoreState as CoreState, HomeAssistant as HomeAssistant, callback as callback
-from typing import Any
+from typing import Any, Generic, Literal, TypeVar
 
 SAVE_DELAY: int
 SAVE_DELAY_LONG: int
+_DataT = TypeVar('_DataT')
+_StoreDataT = TypeVar('_StoreDataT', bound=Mapping[str, Any] | Sequence[Any])
 
-class BaseRegistry(ABC, metaclass=abc.ABCMeta):
+class BaseRegistryItems(UserDict[str, _DataT], ABC, metaclass=abc.ABCMeta):
+    data: dict[str, _DataT]
+    def values(self) -> ValuesView[_DataT]: ...
+    @abstractmethod
+    def _index_entry(self, key: str, entry: _DataT) -> None: ...
+    @abstractmethod
+    def _unindex_entry(self, key: str, replacement_entry: _DataT | None = None) -> None: ...
+    def __setitem__(self, key: str, entry: _DataT) -> None: ...
+    def _unindex_entry_value(self, key: str, value: str, index: dict[str, dict[str, Literal[True]]]) -> None: ...
+    def __delitem__(self, key: str) -> None: ...
+
+class BaseRegistry(ABC, Generic[_StoreDataT], metaclass=abc.ABCMeta):
     hass: HomeAssistant
-    _store: Store
+    _store: Store[_StoreDataT]
     def async_schedule_save(self) -> None: ...
     @abstractmethod
-    def _data_to_save(self) -> dict[str, Any]: ...
+    def _data_to_save(self) -> _StoreDataT: ...

homeassistant-stubs/helpers/reload.pyi

@@ -24,7 +24,7 @@ async def async_integration_yaml_config(hass: HomeAssistant, integration_name: s
 @overload
 async def async_integration_yaml_config(hass: HomeAssistant, integration_name: str, *, raise_on_failure: Literal[True]) -> ConfigType: ...
 @overload
-async def async_integration_yaml_config(hass: HomeAssistant, integration_name: str, *, raise_on_failure: Literal[False] | bool) -> ConfigType | None: ...
+async def async_integration_yaml_config(hass: HomeAssistant, integration_name: str, *, raise_on_failure: Literal[False]) -> ConfigType | None: ...
 def async_get_platform_without_config_entry(hass: HomeAssistant, integration_name: str, integration_platform_name: str) -> EntityPlatform | None: ...
 async def async_setup_reload_service(hass: HomeAssistant, domain: str, platforms: Iterable[str]) -> None: ...
 def setup_reload_service(hass: HomeAssistant, domain: str, platforms: Iterable[str]) -> None: ...

homeassistant-stubs/helpers/schema_config_entry_flow.pyi

@@ -4,7 +4,7 @@ from . import selector as selector
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
-from collections.abc import Callable as Callable, Coroutine, Mapping
+from collections.abc import Callable as Callable, Container, Coroutine, Mapping
 from dataclasses import dataclass
 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
@@ -26,7 +26,7 @@ class SchemaFlowFormStep(SchemaFlowStep):
 
 @dataclass(slots=True)
 class SchemaFlowMenuStep(SchemaFlowStep):
-    options: list[str] | dict[str, str]
+    options: Container[str]
     def __init__(self, options) -> None: ...
 
 class SchemaCommonFlowHandler:

homeassistant-stubs/helpers/script.pyi

@@ -117,6 +117,7 @@ class _ScriptRun:
     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: ...
+    def _async_handle_timeout(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: ...
@@ -212,7 +213,7 @@ class Script:
     def _find_referenced_entities(referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...
     def run(self, variables: _VarsType | None = None, context: Context | None = None) -> None: ...
     async def async_run(self, run_variables: _VarsType | None = None, context: Context | None = None, started_action: Callable[..., Any] | None = None) -> ScriptRunResult | None: ...
-    async def _async_stop(self, aws: list[asyncio.Task], update_state: bool, spare: _ScriptRun | None) -> None: ...
+    async def _async_stop(self, aws: list[asyncio.Task], update_state: bool) -> None: ...
     async def async_stop(self, update_state: bool = True, spare: _ScriptRun | None = None) -> None: ...
     async def _async_get_condition(self, config): ...
     def _prep_repeat_script(self, step: int) -> Script: ...

homeassistant-stubs/helpers/service.pyi

@@ -23,6 +23,7 @@ CONF_SERVICE_ENTITY_ID: str
 _LOGGER: Incomplete
 SERVICE_DESCRIPTION_CACHE: str
 ALL_SERVICE_DESCRIPTIONS_CACHE: str
+_T = TypeVar('_T')
 
 def _base_components() -> dict[str, ModuleType]: ...
 def _validate_option_or_feature(option_or_feature: str, label: str) -> Any: ...
@@ -88,5 +89,7 @@ class ReloadServiceHelper:
     _service_func: Incomplete
     _service_running: bool
     _service_condition: Incomplete
-    def __init__(self, service_func: Callable[[ServiceCall], Awaitable]) -> None: ...
+    _pending_reload_targets: Incomplete
+    _reload_targets_func: Incomplete
+    def __init__(self, service_func: Callable[[ServiceCall], Awaitable], reload_targets_func: Callable[[ServiceCall], set[_T]]) -> None: ...
     async def execute_service(self, service_call: ServiceCall) -> None: ...

homeassistant-stubs/helpers/significant_change.pyi

@@ -13,10 +13,10 @@ ExtraCheckTypeFunc: Incomplete
 async def create_checker(hass: HomeAssistant, _domain: str, extra_significant_check: ExtraCheckTypeFunc | None = None) -> SignificantlyChangedChecker: ...
 async def _initialize(hass: HomeAssistant) -> None: ...
 def either_one_none(val1: Any | None, val2: Any | None) -> bool: ...
-def _check_numeric_change(old_state: int | float | None, new_state: int | float | None, change: int | float, metric: Callable[[int | float, int | float], int | float]) -> bool: ...
-def check_absolute_change(val1: int | float | None, val2: int | float | None, change: int | float) -> bool: ...
-def check_percentage_change(old_state: int | float | None, new_state: int | float | None, change: int | float) -> bool: ...
-def check_valid_float(value: str | int | float) -> bool: ...
+def _check_numeric_change(old_state: float | None, new_state: float | None, change: float, metric: Callable[[int | float, int | float], int | float]) -> bool: ...
+def check_absolute_change(val1: float | None, val2: float | None, change: float) -> bool: ...
+def check_percentage_change(old_state: float | None, new_state: float | None, change: float) -> bool: ...
+def check_valid_float(value: str | float) -> bool: ...
 
 class SignificantlyChangedChecker:
     hass: Incomplete

homeassistant-stubs/helpers/start.pyi

@@ -1,8 +1,10 @@
+from .typing import NoEventData as NoEventData
 from collections.abc import Callable as Callable, Coroutine
 from homeassistant.const import EVENT_HOMEASSISTANT_START as EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, CoreState as CoreState, Event as Event, HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.util.event_type import EventType as EventType
 from typing import Any
 
-def _async_at_core_state(hass: HomeAssistant, at_start_cb: Callable[[HomeAssistant], Coroutine[Any, Any, None] | None], event_type: str, check_state: Callable[[HomeAssistant], bool]) -> CALLBACK_TYPE: ...
+def _async_at_core_state(hass: HomeAssistant, at_start_cb: Callable[[HomeAssistant], Coroutine[Any, Any, None] | None], event_type: EventType[NoEventData], check_state: Callable[[HomeAssistant], bool]) -> CALLBACK_TYPE: ...
 def async_at_start(hass: HomeAssistant, at_start_cb: Callable[[HomeAssistant], Coroutine[Any, Any, None] | None]) -> CALLBACK_TYPE: ...
 def async_at_started(hass: HomeAssistant, at_start_cb: Callable[[HomeAssistant], Coroutine[Any, Any, None] | None]) -> CALLBACK_TYPE: ...

homeassistant-stubs/helpers/template.pyi

@@ -20,7 +20,7 @@ from homeassistant.util.thread import ThreadWithException as ThreadWithException
 from jinja2.sandbox import ImmutableSandboxedEnvironment
 from lru import LRU
 from types import CodeType, TracebackType
-from typing import Any, Literal, NoReturn, ParamSpec, TypeVar, overload
+from typing import Any, Literal, NoReturn, ParamSpec, Self, TypeVar, overload
 
 _LOGGER: Incomplete
 _SENTINEL: Incomplete
@@ -64,7 +64,8 @@ class ResultWrapper:
 def gen_result_wrapper(kls: type[dict | list | set]) -> type: ...
 
 class TupleWrapper(tuple, ResultWrapper):
-    def __new__(cls, value: tuple, *, render_result: str | None = None) -> TupleWrapper: ...
+    __slots__: Incomplete
+    def __new__(cls, value: tuple, *, render_result: str | None = None) -> Self: ...
     render_result: Incomplete
     def __init__(self, value: tuple, *, render_result: str | None = None) -> None: ...
     def __str__(self) -> str: ...
@@ -203,7 +204,7 @@ class TemplateStateBase(State):
     @property
     def state_with_unit(self) -> str: ...
     def format_state(self, rounded: bool, with_unit: bool) -> str: ...
-    def __eq__(self, other: Any) -> bool: ...
+    def __eq__(self, other: object) -> bool: ...
 
 class TemplateState(TemplateStateBase):
     __slots__: Incomplete
@@ -325,6 +326,8 @@ def to_json(value: Any, ensure_ascii: bool = False, pretty_print: bool = False,
 def random_every_time(context, values): ...
 def today_at(hass: HomeAssistant, time_str: str = '') -> datetime: ...
 def relative_time(hass: HomeAssistant, value: Any) -> Any: ...
+def time_since(hass: HomeAssistant, value: Any | datetime, precision: int = 1) -> Any: ...
+def time_until(hass: HomeAssistant, value: Any | datetime, precision: int = 1) -> Any: ...
 def urlencode(value): ...
 def slugify(value, separator: str = '_'): ...
 def iif(value: Any, if_true: Any = True, if_false: Any = False, if_none: Any = ...) -> Any: ...

homeassistant-stubs/helpers/translation.pyi

@@ -1,3 +1,4 @@
+import pathlib
 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
@@ -10,10 +11,9 @@ _LOGGER: Incomplete
 TRANSLATION_FLATTEN_CACHE: str
 LOCALE_EN: str
 
-def recursive_flatten(prefix: Any, data: dict[str, Any]) -> dict[str, Any]: ...
-def component_translation_path(component: str, language: str, integration: Integration) -> str | None: ...
-def _load_translations_files_by_language(translation_files: dict[str, dict[str, str]]) -> dict[str, dict[str, Any]]: ...
-def _merge_resources(translation_strings: dict[str, dict[str, Any]], components: set[str], category: str) -> dict[str, dict[str, Any]]: ...
+def recursive_flatten(prefix: str, data: dict[str, dict[str, Any] | str]) -> dict[str, str]: ...
+def component_translation_path(language: str, integration: Integration) -> pathlib.Path: ...
+def _load_translations_files_by_language(translation_files: dict[str, dict[str, pathlib.Path]]) -> dict[str, dict[str, Any]]: ...
 def build_resources(translation_strings: dict[str, dict[str, dict[str, Any] | str]], components: set[str], category: str) -> dict[str, dict[str, Any] | str]: ...
 async def _async_get_component_strings(hass: HomeAssistant, languages: Iterable[str], components: set[str], integrations: dict[str, Integration]) -> dict[str, dict[str, Any]]: ...
 
@@ -29,16 +29,12 @@ class _TranslationCache:
     async def async_fetch(self, language: str, category: str, components: set[str]) -> dict[str, str]: ...
     def get_cached(self, language: str, category: str, components: set[str]) -> dict[str, str]: ...
     async def _async_load(self, language: str, components: set[str]) -> None: ...
-    def _validate_placeholders(self, language: str, updated_resources: dict[str, Any], cached_resources: dict[str, Any] | None = None) -> dict[str, Any]: ...
+    def _validate_placeholders(self, language: str, updated_resources: dict[str, str], cached_resources: dict[str, str] | None = None) -> dict[str, str]: ...
     def _build_category_cache(self, language: str, components: set[str], translation_strings: dict[str, dict[str, Any]]) -> None: ...
 
 async def async_get_translations(hass: HomeAssistant, language: str, category: str, integrations: Iterable[str] | None = None, config_flow: bool | None = None) -> dict[str, str]: ...
 def async_get_cached_translations(hass: HomeAssistant, language: str, category: str, integration: str | None = None) -> dict[str, str]: ...
 def _async_get_translations_cache(hass: HomeAssistant) -> _TranslationCache: ...
-
-_DIRECT_MAPPED_CATEGORIES: Incomplete
-
-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: ...

homeassistant-stubs/helpers/typing.pyi

@@ -1,22 +1,26 @@
-import homeassistant
+from .deprecation import DeprecatedAlias as DeprecatedAlias, all_with_deprecated_constants as all_with_deprecated_constants, check_if_deprecated_constant as check_if_deprecated_constant, dir_with_deprecated_constants as dir_with_deprecated_constants
 from _typeshed import Incomplete
+from collections.abc import Mapping
 from enum import Enum
-from typing import Any, TypeVar
+from typing import Any, Never
 
-_DataT = TypeVar('_DataT')
 GPSType = tuple[float, float]
 ConfigType = dict[str, Any]
-ContextType = homeassistant.core.Context
 DiscoveryInfoType = dict[str, Any]
 ServiceDataType = dict[str, Any]
 StateType: Incomplete
 TemplateVarsType: Incomplete
+NoEventData = Mapping[str, Never]
 QueryType = Any
 
 class UndefinedType(Enum):
     _singleton: int
 
 UNDEFINED: Incomplete
-EventType = homeassistant.core.Event
-HomeAssistantType = homeassistant.core.HomeAssistant
-ServiceCallType = homeassistant.core.ServiceCall
+_DEPRECATED_ContextType: Incomplete
+_DEPRECATED_EventType: Incomplete
+_DEPRECATED_HomeAssistantType: Incomplete
+_DEPRECATED_ServiceCallType: Incomplete
+__getattr__: Incomplete
+__dir__: Incomplete
+__all__: Incomplete

homeassistant-stubs/helpers/update_coordinator.pyi

@@ -65,12 +65,13 @@ class DataUpdateCoordinator(BaseDataUpdateCoordinatorProtocol, Generic[_DataT]):
     async def async_config_entry_first_refresh(self) -> None: ...
     async def async_refresh(self) -> None: ...
     async def _async_refresh(self, log_failures: bool = True, raise_on_auth_failed: bool = False, scheduled: bool = False, raise_on_entry_error: bool = False) -> None: ...
+    def _async_refresh_finished(self) -> None: ...
     def async_set_update_error(self, err: Exception) -> None: ...
     def async_set_updated_data(self, data: _DataT) -> None: ...
 
 class TimestampDataUpdateCoordinator(DataUpdateCoordinator[_DataT]):
     last_update_success_time: datetime | None
-    async def _async_refresh(self, log_failures: bool = True, raise_on_auth_failed: bool = False, scheduled: bool = False, raise_on_entry_error: bool = False) -> None: ...
+    def _async_refresh_finished(self) -> None: ...
 
 class BaseCoordinatorEntity(entity.Entity, Generic[_BaseDataUpdateCoordinatorT], metaclass=abc.ABCMeta):
     coordinator: Incomplete