homeassistant-stubs 2024.5.5__py3-none-any.whl → 2024.6.0__py3-none-any.whl

homeassistant-stubs/helpers/device_registry.pyi

@@ -3,7 +3,8 @@ from .debounce import Debouncer as Debouncer
 from .deprecation import DeprecatedConstantEnum as DeprecatedConstantEnum, 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 .frame import report as report
 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, BaseRegistryItems as BaseRegistryItems
+from .registry import BaseRegistry as BaseRegistry, BaseRegistryItems as BaseRegistryItems, RegistryIndexType as RegistryIndexType
+from .singleton import singleton as singleton
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from collections.abc import Mapping
@@ -15,12 +16,13 @@ from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, R
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.loader import async_suggest_report_issue as async_suggest_report_issue
 from homeassistant.util.event_type import EventType as EventType
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.json import format_unserializable_data as format_unserializable_data
-from typing import Any, Literal, TypeVar, TypedDict
+from typing import Any, Literal, TypedDict
 from yarl import URL
 
 _LOGGER: Incomplete
-DATA_REGISTRY: str
+DATA_REGISTRY: HassKey[DeviceRegistry]
 EVENT_DEVICE_REGISTRY_UPDATED: EventType[EventDeviceRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
@@ -143,7 +145,6 @@ def format_mac(mac: str) -> str: ...
 
 class DeviceRegistryStore(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[str, Any]: ...
-_EntryTypeT = TypeVar('_EntryTypeT', DeviceEntry, DeletedDeviceEntry)
 
 class DeviceRegistryItems(BaseRegistryItems[_EntryTypeT]):
     _connections: Incomplete
@@ -176,7 +177,7 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
     def _async_get_deleted_device(self, identifiers: set[tuple[str, str]], connections: set[tuple[str, str]]) -> DeletedDeviceEntry | None: ...
     def _substitute_name_placeholders(self, domain: str, name: str, translation_placeholders: Mapping[str, str]) -> str: ...
     def async_get_or_create(self, *, config_entry_id: str, configuration_url: str | URL | None | UndefinedType = ..., connections: set[tuple[str, str]] | None | UndefinedType = ..., default_manufacturer: str | None | UndefinedType = ..., default_model: str | None | UndefinedType = ..., default_name: str | None | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., identifiers: set[tuple[str, str]] | None | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., model: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., translation_key: str | None = None, translation_placeholders: Mapping[str, str] | None = None, via_device: tuple[str, str] | None | UndefinedType = ...) -> DeviceEntry: ...
-    def async_update_device(self, device_id: str, *, add_config_entry_id: str | UndefinedType = ..., area_id: str | None | UndefinedType = ..., configuration_url: str | URL | None | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., merge_connections: set[tuple[str, str]] | UndefinedType = ..., merge_identifiers: set[tuple[str, str]] | UndefinedType = ..., model: str | None | UndefinedType = ..., name_by_user: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., new_identifiers: set[tuple[str, str]] | UndefinedType = ..., remove_config_entry_id: str | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., via_device_id: str | None | UndefinedType = ...) -> DeviceEntry | None: ...
+    def async_update_device(self, device_id: str, *, add_config_entry_id: str | UndefinedType = ..., area_id: str | None | UndefinedType = ..., configuration_url: str | URL | None | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., merge_connections: set[tuple[str, str]] | UndefinedType = ..., merge_identifiers: set[tuple[str, str]] | UndefinedType = ..., model: str | None | UndefinedType = ..., name_by_user: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., new_connections: set[tuple[str, str]] | UndefinedType = ..., new_identifiers: set[tuple[str, str]] | UndefinedType = ..., remove_config_entry_id: str | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., via_device_id: str | None | UndefinedType = ...) -> DeviceEntry | None: ...
     def async_remove_device(self, device_id: str) -> None: ...
     async def async_load(self) -> None: ...
     def _data_to_save(self) -> dict[str, Any]: ...

homeassistant-stubs/helpers/discovery.pyi

@@ -1,5 +1,5 @@
 from ..util.signal_type import SignalTypeFormat as SignalTypeFormat
-from .dispatcher import 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_internal as async_dispatcher_send_internal
 from .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from collections.abc import Callable as Callable, Coroutine
 from homeassistant import core as core, setup as setup

homeassistant-stubs/helpers/discovery_flow.pyi

@@ -5,10 +5,11 @@ from homeassistant.const import EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTA
 from homeassistant.core import CoreState as CoreState, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.async_ import gather_with_limited_concurrency as gather_with_limited_concurrency
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any, NamedTuple
 
 FLOW_INIT_LIMIT: int
-DISCOVERY_FLOW_DISPATCHER: str
+DISCOVERY_FLOW_DISPATCHER: HassKey[FlowDispatcher]
 
 def async_create_flow(hass: HomeAssistant, domain: str, context: dict[str, Any], data: Any) -> None: ...
 def _async_init_flow(hass: HomeAssistant, domain: str, context: dict[str, Any], data: Any) -> Coroutine[None, None, ConfigFlowResult] | None: ...

homeassistant-stubs/helpers/dispatcher.pyi

@@ -1,13 +1,12 @@
 from _typeshed import Incomplete
 from collections.abc import Callable, Coroutine
-from homeassistant.core import HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback, get_hassjob_callable_job_type as get_hassjob_callable_job_type
+from homeassistant.core import HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, callback as callback, get_hassjob_callable_job_type as get_hassjob_callable_job_type
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.async_ import run_callback_threadsafe as run_callback_threadsafe
-from homeassistant.util.logging import catch_log_exception as catch_log_exception
+from homeassistant.util.logging import catch_log_exception as catch_log_exception, log_exception as log_exception
 from homeassistant.util.signal_type import SignalType as SignalType
-from typing import Any, TypeVarTuple, overload
+from typing import Any, overload
 
-_Ts = TypeVarTuple('_Ts')
 _LOGGER: Incomplete
 DATA_DISPATCHER: str
 _DispatcherDataType: Incomplete
@@ -26,8 +25,9 @@ def dispatcher_send(hass: HomeAssistant, signal: SignalType[Unpack[_Ts]], *args:
 @overload
 def dispatcher_send(hass: HomeAssistant, signal: str, *args: Any) -> None: ...
 def _format_err(signal: SignalType[Unpack[_Ts]] | str, target: Callable[[Unpack[_Ts]], Any] | Callable[..., Any], *args: Any) -> str: ...
-def _generate_job(signal: SignalType[Unpack[_Ts]] | str, target: Callable[[Unpack[_Ts]], Any] | Callable[..., Any]) -> HassJob[..., None | Coroutine[Any, Any, None]]: ...
+def _generate_job(signal: SignalType[Unpack[_Ts]] | str, target: Callable[[Unpack[_Ts]], Any] | Callable[..., Any]) -> HassJob[..., Coroutine[Any, Any, None] | None]: ...
 @overload
 def async_dispatcher_send(hass: HomeAssistant, signal: SignalType[Unpack[_Ts]], *args: Unpack[_Ts]) -> None: ...
 @overload
 def async_dispatcher_send(hass: HomeAssistant, signal: str, *args: Any) -> None: ...
+def async_dispatcher_send_internal(hass: HomeAssistant, signal: SignalType[Unpack[_Ts]] | str, *args: Unpack[_Ts]) -> None: ...

homeassistant-stubs/helpers/entity.pyi

@@ -1,6 +1,7 @@
 import asyncio
 import dataclasses
-from . import device_registry as dr, entity_registry as er
+import time
+from . import device_registry as dr, entity_registry as er, singleton as singleton
 from .device_registry import DeviceInfo as DeviceInfo, EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
 from .entity_platform import EntityPlatform as EntityPlatform
 from .event import async_track_device_registry_updated_event as async_track_device_registry_updated_event, async_track_entity_registry_updated_event as async_track_entity_registry_updated_event
@@ -18,9 +19,9 @@ from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, I
 from homeassistant.loader import async_suggest_report_issue as async_suggest_report_issue, bind_hass as bind_hass
 from homeassistant.util import ensure_unique_string as ensure_unique_string, slugify as slugify
 from homeassistant.util.frozen_dataclass_compat import FrozenOrThawed as FrozenOrThawed
-from typing import Any, Final, Literal, NotRequired, TypeVar, TypedDict
+from typing import Any, Final, Literal, NotRequired, TypedDict
 
-_T = TypeVar('_T')
+timer = time.time
 _LOGGER: Incomplete
 SLOW_UPDATE_WARNING: int
 DATA_ENTITY_SOURCE: str
@@ -118,7 +119,7 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     _attr_assumed_state: bool
     _attr_attribution: str | None
     _attr_available: bool
-    _attr_capability_attributes: Mapping[str, Any] | None
+    _attr_capability_attributes: dict[str, Any] | None
     _attr_device_class: str | None
     _attr_device_info: DeviceInfo | None
     _attr_entity_category: EntityCategory | None
@@ -164,7 +165,7 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     @cached_property
     def state(self) -> StateType: ...
     @cached_property
-    def capability_attributes(self) -> Mapping[str, Any] | None: ...
+    def capability_attributes(self) -> dict[str, Any] | None: ...
     def get_initial_entity_options(self) -> er.EntityOptionsType | None: ...
     @cached_property
     def state_attributes(self) -> dict[str, Any] | None: ...

homeassistant-stubs/helpers/entity_platform.pyi

@@ -3,20 +3,21 @@ import voluptuous as vol
 from . import service as service, translation as translation
 from .entity import Entity as Entity
 from .entity_registry import EntityRegistry as EntityRegistry, RegistryEntryDisabler as RegistryEntryDisabler, RegistryEntryHider as RegistryEntryHider
-from .event import async_call_later as async_call_later, async_track_time_interval as async_track_time_interval
+from .event import async_call_later as async_call_later
 from .issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
 from .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, UNDEFINED as UNDEFINED
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Coroutine, Iterable
 from contextvars import ContextVar
-from datetime import datetime, timedelta
+from datetime import timedelta
 from homeassistant import config_entries as config_entries
 from homeassistant.const import ATTR_RESTORED as ATTR_RESTORED, DEVICE_DEFAULT_NAME as DEVICE_DEFAULT_NAME, EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, CoreState as CoreState, HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, SupportsResponse as SupportsResponse, callback as callback, split_entity_id as split_entity_id, valid_entity_id as valid_entity_id
-from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, PlatformNotReady as PlatformNotReady
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryError as ConfigEntryError, ConfigEntryNotReady as ConfigEntryNotReady, HomeAssistantError as HomeAssistantError, PlatformNotReady as PlatformNotReady
 from homeassistant.generated import languages as languages
 from homeassistant.setup import SetupPhases as SetupPhases, async_start_setup as async_start_setup
 from homeassistant.util.async_ import create_eager_task as create_eager_task
+from homeassistant.util.hass_dict import HassKey as HassKey
 from logging import Logger
 from typing import Any, Protocol
 
@@ -25,9 +26,9 @@ SLOW_SETUP_MAX_WAIT: int
 SLOW_ADD_ENTITY_MAX_WAIT: int
 SLOW_ADD_MIN_TIMEOUT: int
 PLATFORM_NOT_READY_RETRIES: int
-DATA_ENTITY_PLATFORM: str
-DATA_DOMAIN_ENTITIES: str
-DATA_DOMAIN_PLATFORM_ENTITIES: str
+DATA_ENTITY_PLATFORM: HassKey[dict[str, list[EntityPlatform]]]
+DATA_DOMAIN_ENTITIES: HassKey[dict[str, dict[str, Entity]]]
+DATA_DOMAIN_PLATFORM_ENTITIES: HassKey[dict[tuple[str, str], dict[str, Entity]]]
 PLATFORM_NOT_READY_BASE_WAIT_TIME: int
 _LOGGER: Incomplete
 
@@ -46,6 +47,7 @@ class EntityPlatform:
     platform_name: Incomplete
     platform: Incomplete
     scan_interval: Incomplete
+    scan_interval_seconds: Incomplete
     entity_namespace: Incomplete
     config_entry: Incomplete
     entities: Incomplete
@@ -55,7 +57,7 @@ class EntityPlatform:
     object_id_platform_translations: Incomplete
     _tasks: Incomplete
     _setup_complete: bool
-    _async_unsub_polling: Incomplete
+    _async_polling_timer: Incomplete
     _async_cancel_retry_setup: Incomplete
     _process_updates: Incomplete
     parallel_updates: Incomplete
@@ -80,7 +82,7 @@ class EntityPlatform:
     async def _async_add_and_update_entities(self, coros: list[Coroutine[Any, Any, None]], entities: list[Entity], timeout: float) -> None: ...
     async def _async_add_entities(self, coros: list[Coroutine[Any, Any, None]], entities: list[Entity], timeout: float) -> None: ...
     async def async_add_entities(self, new_entities: Iterable[Entity], update_before_add: bool = False) -> None: ...
-    def _async_handle_interval_callback(self, now: datetime) -> None: ...
+    def _async_handle_interval_callback(self) -> None: ...
     def _entity_id_already_exists(self, entity_id: str) -> tuple[bool, bool]: ...
     async def _async_add_entity(self, entity: Entity, update_before_add: bool, entity_registry: EntityRegistry) -> None: ...
     async def async_reset(self) -> None: ...
@@ -90,7 +92,7 @@ class EntityPlatform:
     async def async_remove_entity(self, entity_id: str) -> None: ...
     async def async_extract_from_service(self, service_call: ServiceCall, expand_group: bool = True) -> list[Entity]: ...
     def async_register_entity_service(self, name: str, schema: dict[str | vol.Marker, Any] | vol.Schema, func: str | Callable[..., Any], required_features: Iterable[int] | None = None, supports_response: SupportsResponse = ...) -> None: ...
-    async def _update_entity_states(self, now: datetime) -> None: ...
+    async def _async_update_entity_states(self) -> None: ...
 
 current_platform: ContextVar[EntityPlatform | None]
 

homeassistant-stubs/helpers/entity_registry.pyi

@@ -1,7 +1,8 @@
 from . import storage as storage
 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, BaseRegistryItems as BaseRegistryItems
+from .registry import BaseRegistry as BaseRegistry, BaseRegistryItems as BaseRegistryItems, RegistryIndexType as RegistryIndexType
+from .singleton import singleton as singleton
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Container, Hashable, KeysView, Mapping
@@ -14,12 +15,12 @@ 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.hass_dict import HassKey as HassKey
 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
+from typing import Any, Literal, NotRequired, TypedDict
 
-T = TypeVar('T')
-DATA_REGISTRY: str
+DATA_REGISTRY: HassKey[EntityRegistry]
 EVENT_ENTITY_REGISTRY_UPDATED: EventType[EventEntityRegistryUpdatedData]
 _LOGGER: Incomplete
 STORAGE_VERSION_MAJOR: int
@@ -149,7 +150,7 @@ class EntityRegistryItems(BaseRegistryItems[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: ...
+def _validate_item(hass: HomeAssistant, domain: str, platform: str, *, disabled_by: RegistryEntryDisabler | None | UndefinedType = None, entity_category: EntityCategory | None | UndefinedType = None, hidden_by: RegistryEntryHider | None | UndefinedType = None, report_non_string_unique_id: bool = True, unique_id: str | Hashable | UndefinedType | Any) -> None: ...
 
 class EntityRegistry(BaseRegistry):
     deleted_entities: dict[tuple[str, str, str], DeletedRegistryEntry]

homeassistant-stubs/helpers/entity_values.pyi

@@ -1,8 +1,7 @@
 from _typeshed import Incomplete
+from homeassistant.const import MAX_EXPECTED_ENTITY_IDS as MAX_EXPECTED_ENTITY_IDS
 from homeassistant.core import split_entity_id as split_entity_id
 
-_MAX_EXPECTED_ENTITIES: int
-
 class EntityValues:
     _exact: Incomplete
     _domain: Incomplete

homeassistant-stubs/helpers/entityfilter.pyi

@@ -1,7 +1,7 @@
 import re
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
-from homeassistant.const import CONF_DOMAINS as CONF_DOMAINS, CONF_ENTITIES as CONF_ENTITIES, CONF_EXCLUDE as CONF_EXCLUDE, CONF_INCLUDE as CONF_INCLUDE
+from homeassistant.const import CONF_DOMAINS as CONF_DOMAINS, CONF_ENTITIES as CONF_ENTITIES, CONF_EXCLUDE as CONF_EXCLUDE, CONF_INCLUDE as CONF_INCLUDE, MAX_EXPECTED_ENTITY_IDS as MAX_EXPECTED_ENTITY_IDS
 from homeassistant.core import split_entity_id as split_entity_id
 
 CONF_INCLUDE_DOMAINS: str

homeassistant-stubs/helpers/event.pyi

@@ -8,6 +8,7 @@ from .sun import get_astral_event_next as get_astral_event_next
 from .template import RenderInfo as RenderInfo, Template as Template, result_as_boolean as result_as_boolean
 from .typing import TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
+from collections import defaultdict
 from collections.abc import Callable, Coroutine, Iterable, Mapping, Sequence
 from dataclasses import dataclass
 from datetime import datetime, timedelta
@@ -17,18 +18,14 @@ 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 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
-TRACK_STATE_ADDED_DOMAIN_CALLBACKS: str
-TRACK_STATE_ADDED_DOMAIN_LISTENER: str
-TRACK_STATE_REMOVED_DOMAIN_CALLBACKS: str
-TRACK_STATE_REMOVED_DOMAIN_LISTENER: str
-TRACK_ENTITY_REGISTRY_UPDATED_CALLBACKS: str
-TRACK_ENTITY_REGISTRY_UPDATED_LISTENER: str
-TRACK_DEVICE_REGISTRY_UPDATED_CALLBACKS: str
-TRACK_DEVICE_REGISTRY_UPDATED_LISTENER: str
+from homeassistant.util.hass_dict import HassKey as HassKey
+from typing import Any, Concatenate, Generic, TypeVar
+
+_TRACK_STATE_CHANGE_DATA: HassKey[_KeyedEventData[EventStateChangedData]]
+_TRACK_STATE_ADDED_DOMAIN_DATA: HassKey[_KeyedEventData[EventStateChangedData]]
+_TRACK_STATE_REMOVED_DOMAIN_DATA: HassKey[_KeyedEventData[EventStateChangedData]]
+_TRACK_ENTITY_REGISTRY_UPDATED_DATA: HassKey[_KeyedEventData[EventEntityRegistryUpdatedData]]
+_TRACK_DEVICE_REGISTRY_UPDATED_DATA: HassKey[_KeyedEventData[EventDeviceRegistryUpdatedData]]
 _ALL_LISTENER: str
 _DOMAINS_LISTENER: str
 _ENTITIES_LISTENER: str
@@ -36,16 +33,20 @@ _LOGGER: Incomplete
 RANDOM_MICROSECOND_MIN: int
 RANDOM_MICROSECOND_MAX: int
 _TypedDictT = TypeVar('_TypedDictT', bound=Mapping[str, Any])
-_P = ParamSpec('_P')
 
 @dataclass(slots=True, frozen=True)
 class _KeyedEventTracker(Generic[_TypedDictT]):
-    listeners_key: str
-    callbacks_key: str
+    key: HassKey[_KeyedEventData[_TypedDictT]]
     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]
-    def __init__(self, listeners_key, callbacks_key, event_type, dispatcher_callable, filter_callable) -> None: ...
+    def __init__(self, key, event_type, dispatcher_callable, filter_callable) -> None: ...
+
+@dataclass(slots=True, frozen=True)
+class _KeyedEventData(Generic[_TypedDictT]):
+    listener: CALLBACK_TYPE
+    callbacks: defaultdict[str, list[HassJob[[Event[_TypedDictT]], Any]]]
+    def __init__(self, listener, callbacks) -> None: ...
 
 @dataclass(slots=True)
 class TrackStates:
@@ -69,7 +70,7 @@ class TrackTemplateResult:
     def __init__(self, template, last_result, result) -> 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: ...
+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: str | Iterable[str] | None = None, to_state: str | Iterable[str] | None = None) -> CALLBACK_TYPE: ...
 
 track_state_change: Incomplete
 
@@ -81,7 +82,7 @@ _KEYED_TRACK_STATE_CHANGE: Incomplete
 
 def _async_track_state_change_event(hass: HomeAssistant, entity_ids: str | Iterable[str], action: Callable[[Event[EventStateChangedData]], Any], job_type: HassJobType | None) -> CALLBACK_TYPE: ...
 def _remove_empty_listener() -> None: ...
-def _remove_listener(hass: HomeAssistant, listeners_key: str, keys: Iterable[str], job: HassJob[[Event[_TypedDictT]], Any], callbacks: dict[str, list[HassJob[[Event[_TypedDictT]], Any]]]) -> None: ...
+def _remove_listener(hass: HomeAssistant, tracker: _KeyedEventTracker[_TypedDictT], keys: Iterable[str], job: HassJob[[Event[_TypedDictT]], Any], callbacks: dict[str, list[HassJob[[Event[_TypedDictT]], Any]]]) -> None: ...
 def _async_track_event(tracker: _KeyedEventTracker[_TypedDictT], hass: HomeAssistant, keys: str | Iterable[str], action: Callable[[Event[_TypedDictT]], None], job_type: HassJobType | None) -> CALLBACK_TYPE: ...
 def _async_dispatch_old_entity_id_or_entity_id_event(hass: HomeAssistant, callbacks: dict[str, list[HassJob[[Event[EventEntityRegistryUpdatedData]], Any]]], event: Event[EventEntityRegistryUpdatedData]) -> None: ...
 def _async_entity_registry_updated_filter(hass: HomeAssistant, callbacks: dict[str, list[HassJob[[Event[EventEntityRegistryUpdatedData]], Any]]], event_data: EventEntityRegistryUpdatedData) -> bool: ...
@@ -199,11 +200,12 @@ class _TrackTimeInterval:
     cancel_on_shutdown: bool | None
     _track_job: HassJob[[datetime], Coroutine[Any, Any, None] | None] | None = ...
     _run_job: HassJob[[datetime], Coroutine[Any, Any, None] | None] | None = ...
-    _cancel_callback: CALLBACK_TYPE | None = ...
+    _timer_handle: asyncio.TimerHandle | None = ...
     def async_attach(self) -> None: ...
-    def _interval_listener(self, now: datetime) -> None: ...
+    def _schedule_timer(self) -> None: ...
+    def _interval_listener(self, _: Any) -> None: ...
     def async_cancel(self) -> None: ...
-    def __init__(self, hass, seconds, job_name, action, cancel_on_shutdown, _track_job, _run_job, _cancel_callback) -> None: ...
+    def __init__(self, hass, seconds, job_name, action, cancel_on_shutdown, _track_job, _run_job, _timer_handle) -> None: ...
 
 def async_track_time_interval(hass: HomeAssistant, action: Callable[[datetime], Coroutine[Any, Any, None] | None], interval: timedelta, *, name: str | None = None, cancel_on_shutdown: bool | None = None) -> CALLBACK_TYPE: ...
 
@@ -258,7 +260,7 @@ def async_track_time_change(hass: HomeAssistant, action: Callable[[datetime], Co
 
 track_time_change: Incomplete
 
-def process_state_match(parameter: None | str | Iterable[str], invert: bool = False) -> Callable[[str | None], bool]: ...
+def process_state_match(parameter: str | Iterable[str] | None, invert: bool = False) -> Callable[[str | None], bool]: ...
 def _entities_domains_from_render_infos(render_infos: Iterable[RenderInfo]) -> tuple[set[str], set[str]]: ...
 def _render_infos_needs_all_listener(render_infos: Iterable[RenderInfo]) -> bool: ...
 def _render_infos_to_track_states(render_infos: Iterable[RenderInfo]) -> TrackStates: ...

homeassistant-stubs/helpers/floor_registry.pyi

@@ -1,5 +1,6 @@
 from .normalized_name_base_registry import NormalizedNameBaseRegistryEntry as NormalizedNameBaseRegistryEntry, NormalizedNameBaseRegistryItems as NormalizedNameBaseRegistryItems, normalize_name as normalize_name
 from .registry import BaseRegistry as BaseRegistry
+from .singleton import singleton as singleton
 from .storage import Store as Store
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
@@ -8,9 +9,10 @@ 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 homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Literal, TypedDict
 
-DATA_REGISTRY: str
+DATA_REGISTRY: HassKey[FloorRegistry]
 EVENT_FLOOR_REGISTRY_UPDATED: EventType[EventFloorRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int

homeassistant-stubs/helpers/frame.pyi

@@ -1,17 +1,15 @@
 import logging
 from _typeshed import Incomplete
-from collections.abc import Callable
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from functools import cached_property as cached_property
-from homeassistant.core import HomeAssistant as HomeAssistant, async_get_hass as async_get_hass
+from homeassistant.core import async_get_hass_or_none as async_get_hass_or_none
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.loader import async_suggest_report_issue as async_suggest_report_issue
 from types import FrameType
-from typing import TypeVar
 
 _LOGGER: Incomplete
 _REPORTED_INTEGRATIONS: set[str]
-_CallableT = TypeVar('_CallableT', bound=Callable)
 
 @dataclass(kw_only=True)
 class IntegrationFrame:
@@ -19,14 +17,14 @@ class IntegrationFrame:
     integration: str
     module: str | None
     relative_filename: str
-    _frame: FrameType
+    frame: FrameType
     @cached_property
     def line_number(self) -> int: ...
     @cached_property
     def filename(self) -> str: ...
     @cached_property
     def line(self) -> str: ...
-    def __init__(self, *, custom_integration, integration, module, relative_filename, _frame) -> None: ...
+    def __init__(self, *, custom_integration, integration, module, relative_filename, frame) -> None: ...
 
 def get_integration_logger(fallback_name: str) -> logging.Logger: ...
 def get_current_frame(depth: int = 0) -> FrameType: ...

homeassistant-stubs/helpers/httpx_client.pyi

@@ -1,16 +1,17 @@
 import httpx
 from .frame import warn_use as warn_use
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable
+from collections.abc import Callable as Callable, Coroutine
 from homeassistant.const import APPLICATION_NAME as APPLICATION_NAME, EVENT_HOMEASSISTANT_CLOSE as EVENT_HOMEASSISTANT_CLOSE, __version__ as __version__
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.ssl import SSLCipherList as SSLCipherList, client_context as client_context, create_no_verify_ssl_context as create_no_verify_ssl_context
 from typing import Any, Self
 
 KEEP_ALIVE_TIMEOUT: int
-DATA_ASYNC_CLIENT: str
-DATA_ASYNC_CLIENT_NOVERIFY: str
+DATA_ASYNC_CLIENT: HassKey[httpx.AsyncClient]
+DATA_ASYNC_CLIENT_NOVERIFY: HassKey[httpx.AsyncClient]
 DEFAULT_LIMITS: Incomplete
 limits: Incomplete
 SERVER_SOFTWARE: Incomplete
@@ -23,4 +24,4 @@ class HassHttpXAsyncClient(httpx.AsyncClient):
     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: ...
+def _async_register_async_client_shutdown(hass: HomeAssistant, client: httpx.AsyncClient, original_aclose: Callable[[], Coroutine[Any, Any, None]]) -> None: ...

homeassistant-stubs/helpers/icon.pyi

@@ -4,13 +4,13 @@ from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import Integration as Integration, async_get_integrations as async_get_integrations
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.json import load_json_object as load_json_object
 from typing import Any
 
-ICON_CACHE: str
+ICON_CACHE: HassKey[_IconsCache]
 _LOGGER: Incomplete
 
-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]: ...
 

homeassistant-stubs/helpers/importlib.pyi

@@ -1,11 +1,13 @@
+import asyncio
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.util.hass_dict import HassKey as HassKey
 from types import ModuleType
 
 _LOGGER: Incomplete
-DATA_IMPORT_CACHE: str
-DATA_IMPORT_FUTURES: str
-DATA_IMPORT_FAILURES: str
+DATA_IMPORT_CACHE: HassKey[dict[str, ModuleType]]
+DATA_IMPORT_FUTURES: HassKey[dict[str, asyncio.Future[ModuleType]]]
+DATA_IMPORT_FAILURES: HassKey[dict[str, bool]]
 
 def _get_module(cache: dict[str, ModuleType], name: str) -> ModuleType: ...
 async def async_import_module(hass: HomeAssistant, name: str) -> ModuleType: ...

homeassistant-stubs/helpers/integration_platform.pyi

@@ -6,12 +6,13 @@ from homeassistant.const import EVENT_COMPONENT_LOADED as EVENT_COMPONENT_LOADED
 from homeassistant.core import Event as Event, HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import Integration as Integration, async_get_integrations as async_get_integrations, async_get_loaded_integration as async_get_loaded_integration, async_register_preload_platform as async_register_preload_platform, bind_hass as bind_hass
 from homeassistant.setup import ATTR_COMPONENT as ATTR_COMPONENT, EventComponentLoaded as EventComponentLoaded
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.logging import catch_log_exception as catch_log_exception
 from types import ModuleType
 from typing import Any
 
 _LOGGER: Incomplete
-DATA_INTEGRATION_PLATFORMS: str
+DATA_INTEGRATION_PLATFORMS: HassKey[list[IntegrationPlatform]]
 
 @dataclass(slots=True, frozen=True)
 class IntegrationPlatform: