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

homeassistant-stubs/helpers/condition.pyi

@@ -37,19 +37,19 @@ async def async_from_config(hass: HomeAssistant, config: ConfigType) -> Conditio
 async def async_and_from_config(hass: HomeAssistant, config: ConfigType) -> ConditionCheckerType: ...
 async def async_or_from_config(hass: HomeAssistant, config: ConfigType) -> ConditionCheckerType: ...
 async def async_not_from_config(hass: HomeAssistant, config: ConfigType) -> ConditionCheckerType: ...
-def numeric_state(hass: HomeAssistant, entity: None | str | State, below: float | str | None = None, above: float | str | None = None, value_template: Template | None = None, variables: TemplateVarsType = None) -> bool: ...
-def async_numeric_state(hass: HomeAssistant, entity: None | str | State, below: float | str | None = None, above: float | str | None = None, value_template: Template | None = None, variables: TemplateVarsType = None, attribute: str | None = None) -> bool: ...
+def numeric_state(hass: HomeAssistant, entity: str | State | None, below: float | str | None = None, above: float | str | None = None, value_template: Template | None = None, variables: TemplateVarsType = None) -> bool: ...
+def async_numeric_state(hass: HomeAssistant, entity: str | State | None, below: float | str | None = None, above: float | str | None = None, value_template: Template | None = None, variables: TemplateVarsType = None, attribute: str | None = None) -> bool: ...
 def async_numeric_state_from_config(config: ConfigType) -> ConditionCheckerType: ...
-def state(hass: HomeAssistant, entity: None | str | State, req_state: Any, for_period: timedelta | None = None, attribute: str | None = None, variables: TemplateVarsType = None) -> bool: ...
+def state(hass: HomeAssistant, entity: str | State | None, req_state: Any, for_period: timedelta | None = None, attribute: str | None = None, variables: TemplateVarsType = None) -> bool: ...
 def state_from_config(config: ConfigType) -> ConditionCheckerType: ...
 def sun(hass: HomeAssistant, before: str | None = None, after: str | None = None, before_offset: timedelta | None = None, after_offset: timedelta | None = None) -> bool: ...
 def sun_from_config(config: ConfigType) -> ConditionCheckerType: ...
 def template(hass: HomeAssistant, value_template: Template, variables: TemplateVarsType = None) -> bool: ...
 def async_template(hass: HomeAssistant, value_template: Template, variables: TemplateVarsType = None, trace_result: bool = True) -> bool: ...
 def async_template_from_config(config: ConfigType) -> ConditionCheckerType: ...
-def time(hass: HomeAssistant, before: dt_time | str | None = None, after: dt_time | str | None = None, weekday: None | str | Container[str] = None) -> bool: ...
+def time(hass: HomeAssistant, before: dt_time | str | None = None, after: dt_time | str | None = None, weekday: str | Container[str] | None = None) -> bool: ...
 def time_from_config(config: ConfigType) -> ConditionCheckerType: ...
-def zone(hass: HomeAssistant, zone_ent: None | str | State, entity: None | str | State) -> bool: ...
+def zone(hass: HomeAssistant, zone_ent: str | State | None, entity: str | State | None) -> bool: ...
 def zone_from_config(config: ConfigType) -> ConditionCheckerType: ...
 async def async_trigger_from_config(hass: HomeAssistant, config: ConfigType) -> ConditionCheckerType: ...
 def numeric_state_validate_config(hass: HomeAssistant, config: ConfigType) -> ConfigType: ...

homeassistant-stubs/helpers/config_entry_flow.pyi

@@ -9,13 +9,12 @@ from homeassistant.components.dhcp import DhcpServiceInfo as DhcpServiceInfo
 from homeassistant.components.ssdp import SsdpServiceInfo as SsdpServiceInfo
 from homeassistant.components.zeroconf import ZeroconfServiceInfo as ZeroconfServiceInfo
 from homeassistant.core import HomeAssistant as HomeAssistant
-from typing import Any, Generic, TypeVar
+from typing import Any
 
-_R = TypeVar('_R', bound='Awaitable[bool] | bool')
 DiscoveryFunctionType = Callable[[HomeAssistant], _R]
 _LOGGER: Incomplete
 
-class DiscoveryFlowHandler(config_entries.ConfigFlow, Generic[_R]):
+class DiscoveryFlowHandler(config_entries.ConfigFlow):
     VERSION: int
     _domain: Incomplete
     _title: Incomplete

homeassistant-stubs/helpers/config_entry_oauth2_flow.pyi

@@ -10,12 +10,13 @@ from homeassistant import config_entries as config_entries
 from homeassistant.components import http as http
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import async_get_application_credentials as async_get_application_credentials
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any
 
 _LOGGER: Incomplete
 DATA_JWT_SECRET: str
-DATA_IMPLEMENTATIONS: str
-DATA_PROVIDERS: str
+DATA_IMPLEMENTATIONS: HassKey[dict[str, dict[str, AbstractOAuth2Implementation]]]
+DATA_PROVIDERS: HassKey[dict[str, Callable[[HomeAssistant, str], Awaitable[list[AbstractOAuth2Implementation]]]]]
 AUTH_CALLBACK_PATH: str
 HEADER_FRONTEND_BASE: str
 MY_AUTH_CALLBACK_PATH: str
@@ -95,6 +96,7 @@ class OAuth2Session:
     hass: Incomplete
     config_entry: Incomplete
     implementation: Incomplete
+    _token_lock: Incomplete
     def __init__(self, hass: HomeAssistant, config_entry: config_entries.ConfigEntry, implementation: AbstractOAuth2Implementation) -> None: ...
     @property
     def token(self) -> dict: ...

homeassistant-stubs/helpers/config_validation.pyi

@@ -7,14 +7,14 @@ from collections.abc import Callable as Callable, Hashable
 from datetime import date as date_sys, datetime as datetime_sys, time as time_sys, timedelta
 from enum import Enum, StrEnum
 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_ABOVE as CONF_ABOVE, CONF_ALIAS as CONF_ALIAS, CONF_ATTRIBUTE as CONF_ATTRIBUTE, CONF_BELOW as CONF_BELOW, 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_ENTITY_ID as CONF_ENTITY_ID, CONF_ENTITY_NAMESPACE as CONF_ENTITY_NAMESPACE, 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 as CONF_FOR, CONF_FOR_EACH as CONF_FOR_EACH, CONF_ID as CONF_ID, CONF_IF as CONF_IF, CONF_MATCH as CONF_MATCH, CONF_PARALLEL as CONF_PARALLEL, CONF_PLATFORM as CONF_PLATFORM, CONF_REPEAT as CONF_REPEAT, CONF_RESPONSE_VARIABLE as CONF_RESPONSE_VARIABLE, CONF_SCAN_INTERVAL as CONF_SCAN_INTERVAL, 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_SERVICE_TEMPLATE as CONF_SERVICE_TEMPLATE, CONF_SET_CONVERSATION_RESPONSE as CONF_SET_CONVERSATION_RESPONSE, CONF_STATE as CONF_STATE, 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_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE, 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, ENTITY_MATCH_ALL as ENTITY_MATCH_ALL, ENTITY_MATCH_ANY as ENTITY_MATCH_ANY, ENTITY_MATCH_NONE as ENTITY_MATCH_NONE, SUN_EVENT_SUNRISE as SUN_EVENT_SUNRISE, SUN_EVENT_SUNSET as SUN_EVENT_SUNSET, UnitOfTemperature as UnitOfTemperature, WEEKDAYS as WEEKDAYS
-from homeassistant.core import HomeAssistant as HomeAssistant, async_get_hass as async_get_hass, split_entity_id as split_entity_id, valid_entity_id as valid_entity_id
+from homeassistant.core import async_get_hass as async_get_hass, async_get_hass_or_none as async_get_hass_or_none, split_entity_id as split_entity_id, valid_entity_id as valid_entity_id
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, TemplateError as TemplateError
 from homeassistant.generated import currencies as currencies
 from homeassistant.generated.countries import COUNTRIES as COUNTRIES
 from homeassistant.generated.languages import LANGUAGES as LANGUAGES
 from homeassistant.util import raise_if_invalid_path as raise_if_invalid_path
 from homeassistant.util.yaml.objects import NodeStrClass as NodeStrClass
-from typing import Any, TypeVar, overload
+from typing import Any, overload
 
 TIME_PERIOD_ERROR: str
 
@@ -34,7 +34,6 @@ longitude: Incomplete
 gps: Incomplete
 sun_event: Incomplete
 port: Incomplete
-_T = TypeVar('_T')
 
 def path(value: Any) -> str: ...
 def has_at_least_one_key(*keys: Any) -> Callable[[dict], dict]: ...
@@ -90,7 +89,7 @@ positive_time_period: Incomplete
 def remove_falsy(value: list[_T]) -> list[_T]: ...
 def service(value: Any) -> str: ...
 def slug(value: Any) -> str: ...
-def schema_with_slug_keys(value_schema: _T | Callable, *, slug_validator: Callable[[Any], str] = ...) -> Callable: ...
+def schema_with_slug_keys(value_schema: dict | Callable, *, slug_validator: Callable[[Any], str] = ...) -> Callable: ...
 def slugify(value: Any) -> str: ...
 def string(value: Any) -> str: ...
 def string_with_no_html(value: Any) -> str: ...
@@ -202,7 +201,7 @@ _SCRIPT_IF_SCHEMA: Incomplete
 _SCRIPT_SET_SCHEMA: Incomplete
 _SCRIPT_SET_CONVERSATION_RESPONSE_SCHEMA: Incomplete
 _SCRIPT_STOP_SCHEMA: Incomplete
-_SCRIPT_PARALLEL_SEQUENCE: Incomplete
+_SCRIPT_SEQUENCE_SCHEMA: Incomplete
 _parallel_sequence_action: Incomplete
 _SCRIPT_PARALLEL_SCHEMA: Incomplete
 SCRIPT_ACTION_ACTIVATE_SCENE: str
@@ -215,6 +214,7 @@ SCRIPT_ACTION_FIRE_EVENT: str
 SCRIPT_ACTION_IF: str
 SCRIPT_ACTION_PARALLEL: str
 SCRIPT_ACTION_REPEAT: str
+SCRIPT_ACTION_SEQUENCE: str
 SCRIPT_ACTION_SET_CONVERSATION_RESPONSE: str
 SCRIPT_ACTION_STOP: str
 SCRIPT_ACTION_VARIABLES: str

homeassistant-stubs/helpers/debounce.pyi

@@ -2,11 +2,8 @@ from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from homeassistant.core import HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback
 from logging import Logger
-from typing import Generic, TypeVar
 
-_R_co = TypeVar('_R_co', covariant=True)
-
-class Debouncer(Generic[_R_co]):
+class Debouncer:
     hass: Incomplete
     logger: Incomplete
     _function: Incomplete

homeassistant-stubs/helpers/deprecation.pyi

@@ -1,10 +1,6 @@
 from collections.abc import Callable as Callable
 from enum import Enum
-from typing import Any, NamedTuple, ParamSpec, TypeVar
-
-_ObjectT = TypeVar('_ObjectT', bound=object)
-_R = TypeVar('_R')
-_P = ParamSpec('_P')
+from typing import Any, NamedTuple
 
 def deprecated_substitute(substitute_name: str) -> Callable[[Callable[[_ObjectT], Any]], Callable[[_ObjectT], Any]]: ...
 def get_deprecated(config: dict[str, Any], new_name: str, old_name: str, default: Any | None = None) -> Any | None: ...

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

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:

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: