homeassistant-stubs 2024.4.4__py3-none-any.whl → 2024.5.0b0__py3-none-any.whl

homeassistant-stubs/helpers/area_registry.pyi

@@ -7,14 +7,27 @@ from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.core import 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 Any, Literal, TypedDict
 
 DATA_REGISTRY: str
-EVENT_AREA_REGISTRY_UPDATED: str
+EVENT_AREA_REGISTRY_UPDATED: EventType[EventAreaRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 STORAGE_VERSION_MINOR: int
 
+class _AreaStoreData(TypedDict):
+    aliases: list[str]
+    floor_id: str | None
+    icon: str | None
+    id: str
+    labels: list[str]
+    name: str
+    picture: str | None
+
+class AreasRegistryStoreData(TypedDict):
+    areas: list[_AreaStoreData]
+
 class EventAreaRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     area_id: str
@@ -29,11 +42,20 @@ class AreaEntry(NormalizedNameBaseRegistryEntry):
     picture: str | None
     def __init__(self, *, name, normalized_name, aliases, floor_id, icon, id, labels, picture) -> None: ...
 
-class AreaRegistryStore(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]: ...
+class AreaRegistryStore(Store[AreasRegistryStoreData]):
+    async def _async_migrate_func(self, old_major_version: int, old_minor_version: int, old_data: dict[str, list[dict[str, Any]]]) -> AreasRegistryStoreData: ...
+
+class AreaRegistryItems(NormalizedNameBaseRegistryItems[AreaEntry]):
+    _labels_index: Incomplete
+    _floors_index: Incomplete
+    def __init__(self) -> None: ...
+    def _index_entry(self, key: str, entry: AreaEntry) -> None: ...
+    def _unindex_entry(self, key: str, replacement_entry: AreaEntry | None = None) -> None: ...
+    def get_areas_for_label(self, label: str) -> list[AreaEntry]: ...
+    def get_areas_for_floor(self, floor: str) -> list[AreaEntry]: ...
 
-class AreaRegistry(BaseRegistry):
-    areas: NormalizedNameBaseRegistryItems[AreaEntry]
+class AreaRegistry(BaseRegistry[AreasRegistryStoreData]):
+    areas: AreaRegistryItems
     _area_data: dict[str, AreaEntry]
     hass: Incomplete
     _store: Incomplete
@@ -47,7 +69,7 @@ class AreaRegistry(BaseRegistry):
     def async_update(self, area_id: str, *, aliases: set[str] | UndefinedType = ..., floor_id: str | None | UndefinedType = ..., icon: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., name: str | UndefinedType = ..., picture: str | None | UndefinedType = ...) -> AreaEntry: ...
     def _async_update(self, area_id: str, *, aliases: set[str] | UndefinedType = ..., floor_id: str | None | UndefinedType = ..., icon: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., name: str | UndefinedType = ..., picture: str | None | UndefinedType = ...) -> AreaEntry: ...
     async def async_load(self) -> None: ...
-    def _data_to_save(self) -> dict[str, list[dict[str, Any]]]: ...
+    def _data_to_save(self) -> AreasRegistryStoreData: ...
     def _generate_area_id(self, name: str) -> str: ...
     def _async_setup_cleanup(self) -> None: ...
 

homeassistant-stubs/helpers/backports/aiohttp_resolver.pyi

@@ -0,0 +1,21 @@
+from _typeshed import Incomplete
+from aiohttp.abc import AbstractResolver
+from typing import Any, TypedDict
+
+_NUMERIC_SOCKET_FLAGS: Incomplete
+_SUPPORTS_SCOPE_ID: Incomplete
+
+class ResolveResult(TypedDict):
+    hostname: str
+    host: str
+    port: int
+    family: int
+    proto: int
+    flags: int
+
+class AsyncResolver(AbstractResolver):
+    _loop: Incomplete
+    _resolver: Incomplete
+    def __init__(self, *args: Any, **kwargs: Any) -> None: ...
+    async def resolve(self, host: str, port: int = 0, family: int = ...) -> list[ResolveResult]: ...
+    async def close(self) -> None: ...

homeassistant-stubs/helpers/category_registry.pyi

@@ -1,22 +1,32 @@
 from .registry import BaseRegistry as BaseRegistry
-from .typing import EventType as EventType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
+from .storage import Store as Store
+from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from collections.abc import Iterable
 from dataclasses import dataclass
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.util.event_type import EventType as EventType
 from homeassistant.util.ulid import ulid_now as ulid_now
 from typing import Literal, TypedDict
 
 DATA_REGISTRY: str
-EVENT_CATEGORY_REGISTRY_UPDATED: str
+EVENT_CATEGORY_REGISTRY_UPDATED: EventType[EventCategoryRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 
+class _CategoryStoreData(TypedDict):
+    category_id: str
+    icon: str | None
+    name: str
+
+class CategoryRegistryStoreData(TypedDict):
+    categories: dict[str, list[_CategoryStoreData]]
+
 class EventCategoryRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     scope: str
     category_id: str
-EventCategoryRegistryUpdated = EventType[EventCategoryRegistryUpdatedData]
+EventCategoryRegistryUpdated = Event[EventCategoryRegistryUpdatedData]
 
 @dataclass(slots=True, kw_only=True, frozen=True)
 class CategoryEntry:
@@ -25,7 +35,7 @@ class CategoryEntry:
     name: str
     def __init__(self, *, category_id, icon, name) -> None: ...
 
-class CategoryRegistry(BaseRegistry):
+class CategoryRegistry(BaseRegistry[CategoryRegistryStoreData]):
     hass: Incomplete
     categories: Incomplete
     _store: Incomplete
@@ -36,7 +46,7 @@ class CategoryRegistry(BaseRegistry):
     def async_delete(self, *, scope: str, category_id: str) -> None: ...
     def async_update(self, *, scope: str, category_id: str, icon: str | None | UndefinedType = ..., name: str | UndefinedType = ...) -> CategoryEntry: ...
     async def async_load(self) -> None: ...
-    def _data_to_save(self) -> dict[str, dict[str, list[dict[str, str | None]]]]: ...
+    def _data_to_save(self) -> CategoryRegistryStoreData: ...
     def _async_ensure_name_is_available(self, scope: str, name: str, category_id: str | None = None) -> None: ...
 
 def async_get(hass: HomeAssistant) -> CategoryRegistry: ...

homeassistant-stubs/helpers/data_entry_flow.pyi

@@ -6,7 +6,7 @@ from homeassistant.components.http.data_validator import RequestDataValidator as
 from typing import Any, Generic
 from typing_extensions import TypeVar
 
-_FlowManagerT = TypeVar('_FlowManagerT', bound='data_entry_flow.FlowManager[Any]', default=data_entry_flow.FlowManager)
+_FlowManagerT = TypeVar('_FlowManagerT', bound=data_entry_flow.FlowManager[Any], default=data_entry_flow.FlowManager)
 
 class _BaseFlowManagerView(HomeAssistantView, Generic[_FlowManagerT]):
     _flow_mgr: Incomplete

homeassistant-stubs/helpers/deprecation.pyi

@@ -22,6 +22,11 @@ class DeprecatedConstantEnum(NamedTuple):
     enum: Enum
     breaks_in_ha_version: str | None
 
+class DeprecatedAlias(NamedTuple):
+    value: Any
+    replacement: str
+    breaks_in_ha_version: str | None
+
 _PREFIX_DEPRECATED: str
 
 def check_if_deprecated_constant(name: str, module_globals: dict[str, Any]) -> Any: ...

homeassistant-stubs/helpers/device_registry.pyi

@@ -3,25 +3,25 @@ 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
+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 Mapping, ValuesView
+from collections.abc import 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 EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
-from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback, get_release_channel as get_release_channel
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, ReleaseChannel as ReleaseChannel, callback as callback, get_release_channel as get_release_channel
 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.json import format_unserializable_data as format_unserializable_data
 from typing import Any, Literal, TypeVar, TypedDict
 from yarl import URL
 
 _LOGGER: Incomplete
 DATA_REGISTRY: str
-EVENT_DEVICE_REGISTRY_UPDATED: str
+EVENT_DEVICE_REGISTRY_UPDATED: EventType[EventDeviceRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 STORAGE_VERSION_MINOR: int
@@ -114,7 +114,9 @@ class DeviceEntry:
     def disabled(self) -> bool: ...
     @property
     def dict_repr(self) -> dict[str, Any]: ...
+    @cached_property
     def json_repr(self) -> bytes | None: ...
+    @cached_property
     def as_storage_fragment(self) -> json_fragment: ...
     def __init__(self, area_id, config_entries, configuration_url, connections, disabled_by, entry_type, hw_version, id, identifiers, labels, manufacturer, model, name_by_user, name, serial_number, suggested_area, sw_version, via_device_id, is_new) -> None: ...
     def __lt__(self, other): ...
@@ -129,6 +131,7 @@ class DeletedDeviceEntry:
     id: str
     orphaned_timestamp: float | None
     def to_device_entry(self, config_entry_id: str, connections: set[tuple[str, str]], identifiers: set[tuple[str, str]]) -> DeviceEntry: ...
+    @cached_property
     def as_storage_fragment(self) -> json_fragment: ...
     def __init__(self, config_entries, connections, identifiers, id, orphaned_timestamp) -> None: ...
     def __lt__(self, other): ...
@@ -142,17 +145,27 @@ 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(UserDict[str, _EntryTypeT]):
+class DeviceRegistryItems(BaseRegistryItems[_EntryTypeT]):
     _connections: Incomplete
     _identifiers: Incomplete
     def __init__(self) -> None: ...
-    def values(self) -> ValuesView[_EntryTypeT]: ...
-    def __setitem__(self, key: str, entry: _EntryTypeT) -> None: ...
-    def __delitem__(self, key: str) -> None: ...
+    def _index_entry(self, key: str, entry: _EntryTypeT) -> None: ...
+    def _unindex_entry(self, key: str, replacement_entry: _EntryTypeT | None = None) -> None: ...
     def get_entry(self, identifiers: set[tuple[str, str]] | None, connections: set[tuple[str, str]] | None) -> _EntryTypeT | None: ...
 
-class DeviceRegistry(BaseRegistry):
-    devices: DeviceRegistryItems[DeviceEntry]
+class ActiveDeviceRegistryItems(DeviceRegistryItems[DeviceEntry]):
+    _area_id_index: Incomplete
+    _config_entry_id_index: Incomplete
+    _labels_index: Incomplete
+    def __init__(self) -> None: ...
+    def _index_entry(self, key: str, entry: DeviceEntry) -> None: ...
+    def _unindex_entry(self, key: str, replacement_entry: DeviceEntry | None = None) -> None: ...
+    def get_devices_for_area_id(self, area_id: str) -> list[DeviceEntry]: ...
+    def get_devices_for_label(self, label: str) -> list[DeviceEntry]: ...
+    def get_devices_for_config_entry_id(self, config_entry_id: str) -> list[DeviceEntry]: ...
+
+class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
+    devices: ActiveDeviceRegistryItems
     deleted_devices: DeviceRegistryItems[DeletedDeviceEntry]
     _device_data: dict[str, DeviceEntry]
     hass: Incomplete

homeassistant-stubs/helpers/dispatcher.pyi

@@ -1,6 +1,6 @@
 from _typeshed import Incomplete
 from collections.abc import Callable, Coroutine
-from homeassistant.core import HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback
+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.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

homeassistant-stubs/helpers/entity.pyi

@@ -8,12 +8,12 @@ from .typing import StateType as StateType, UNDEFINED as UNDEFINED, UndefinedTyp
 from _typeshed import Incomplete
 from abc import ABCMeta
 from collections import deque
-from collections.abc import Callable as Callable, Coroutine, Iterable, Mapping, MutableMapping
+from collections.abc import Callable as Callable, Coroutine, Iterable, Mapping
 from enum import Enum, IntFlag
 from functools import cached_property as cached_property
 from homeassistant.config import DATA_CUSTOMIZE as DATA_CUSTOMIZE
 from homeassistant.const import ATTR_ASSUMED_STATE as ATTR_ASSUMED_STATE, ATTR_ATTRIBUTION as ATTR_ATTRIBUTION, ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_ENTITY_PICTURE as ATTR_ENTITY_PICTURE, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, ATTR_ICON as ATTR_ICON, ATTR_SUPPORTED_FEATURES as ATTR_SUPPORTED_FEATURES, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, DEVICE_DEFAULT_NAME as DEVICE_DEFAULT_NAME, EntityCategory as EntityCategory, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Context as Context, Event as Event, HassJobType as HassJobType, HomeAssistant as HomeAssistant, callback as callback, get_hassjob_callable_job_type as get_hassjob_callable_job_type, get_release_channel as get_release_channel
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Context as Context, Event as Event, HassJobType as HassJobType, HomeAssistant as HomeAssistant, ReleaseChannel as ReleaseChannel, callback as callback, get_hassjob_callable_job_type as get_hassjob_callable_job_type, get_release_channel as get_release_channel
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, InvalidStateError as InvalidStateError, NoEntitySpecifiedError as NoEntitySpecifiedError
 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
@@ -111,6 +111,7 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     __combined_unrecorded_attributes: frozenset[str]
     _job_types: dict[str, HassJobType] | None
     _state_info: StateInfo
+    _is_custom_component: bool
     __capabilities_updated_at: deque[float]
     __capabilities_updated_at_reported: bool
     __remove_future: asyncio.Future[None] | None
@@ -125,7 +126,7 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     _attr_entity_picture: str | None
     _attr_entity_registry_enabled_default: bool
     _attr_entity_registry_visible_default: bool
-    _attr_extra_state_attributes: MutableMapping[str, Any]
+    _attr_extra_state_attributes: dict[str, Any]
     _attr_force_update: bool
     _attr_icon: str | None
     _attr_name: str | None
@@ -203,6 +204,8 @@ class Entity(cached_properties=CACHED_PROPERTIES_WITH_ATTR_, metaclass=ABCCached
     def enabled(self) -> bool: ...
     def async_set_context(self, context: Context) -> None: ...
     async def async_update_ha_state(self, force_refresh: bool = False) -> None: ...
+    def _async_verify_state_writable(self) -> None: ...
+    def _async_write_ha_state_from_call_soon_threadsafe(self) -> None: ...
     def async_write_ha_state(self) -> None: ...
     def _stringify_state(self, available: bool) -> str: ...
     def _friendly_name_internal(self) -> str | None: ...

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