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

homeassistant-stubs/core.pyi

@@ -3,18 +3,19 @@ import datetime
 import enum
 import threading
 import voluptuous as vol
-from . import block_async_io as block_async_io, util as util
+from . import util as util
 from .auth import AuthManager as AuthManager
 from .components.http import ApiConfig as ApiConfig, HomeAssistantHTTP as HomeAssistantHTTP
 from .config_entries import ConfigEntries as ConfigEntries
 from .const import ATTR_DOMAIN as ATTR_DOMAIN, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, ATTR_SERVICE as ATTR_SERVICE, ATTR_SERVICE_DATA as ATTR_SERVICE_DATA, COMPRESSED_STATE_ATTRIBUTES as COMPRESSED_STATE_ATTRIBUTES, COMPRESSED_STATE_CONTEXT as COMPRESSED_STATE_CONTEXT, COMPRESSED_STATE_LAST_CHANGED as COMPRESSED_STATE_LAST_CHANGED, COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE, EVENT_CALL_SERVICE as EVENT_CALL_SERVICE, EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE, EVENT_HOMEASSISTANT_CLOSE as EVENT_HOMEASSISTANT_CLOSE, EVENT_HOMEASSISTANT_FINAL_WRITE as EVENT_HOMEASSISTANT_FINAL_WRITE, EVENT_HOMEASSISTANT_START as EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, EVENT_LOGGING_CHANGED as EVENT_LOGGING_CHANGED, EVENT_SERVICE_REGISTERED as EVENT_SERVICE_REGISTERED, EVENT_SERVICE_REMOVED as EVENT_SERVICE_REMOVED, EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, EVENT_STATE_REPORTED as EVENT_STATE_REPORTED, MATCH_ALL as MATCH_ALL, MAX_LENGTH_EVENT_EVENT_TYPE as MAX_LENGTH_EVENT_EVENT_TYPE, MAX_LENGTH_STATE_STATE as MAX_LENGTH_STATE_STATE, UnitOfLength as UnitOfLength, __version__ as __version__
-from .exceptions import HomeAssistantError as HomeAssistantError, InvalidEntityFormatError as InvalidEntityFormatError, InvalidStateError as InvalidStateError, MaxLengthExceeded as MaxLengthExceeded, ServiceNotFound as ServiceNotFound, Unauthorized as Unauthorized
+from .exceptions import HomeAssistantError as HomeAssistantError, InvalidEntityFormatError as InvalidEntityFormatError, InvalidStateError as InvalidStateError, MaxLengthExceeded as MaxLengthExceeded, ServiceNotFound as ServiceNotFound, ServiceValidationError as ServiceValidationError, Unauthorized as Unauthorized
 from .helpers.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 .helpers.entity import StateInfo as StateInfo
 from .helpers.json import json_bytes as json_bytes, json_fragment as json_fragment
 from .helpers.storage import Store as Store
 from .util import location as location
 from .util.async_ import cancelling as cancelling, create_eager_task as create_eager_task, run_callback_threadsafe as run_callback_threadsafe, shutdown_run_callback_threadsafe as shutdown_run_callback_threadsafe
+from .util.event_type import EventType as EventType
 from .util.executor import InterruptibleThreadPoolExecutor as InterruptibleThreadPoolExecutor
 from .util.json import JsonObjectType as JsonObjectType
 from .util.read_only_dict import ReadOnlyDict as ReadOnlyDict
@@ -26,7 +27,7 @@ from collections import UserDict
 from collections.abc import Callable, Collection, Coroutine, Iterable, KeysView, Mapping, ValuesView
 from dataclasses import dataclass
 from functools import cached_property as cached_property
-from typing import Any, Generic, Literal, NotRequired, ParamSpec, Self, TypeVarTuple, TypedDict, overload
+from typing import Any, Generic, NotRequired, ParamSpec, Self, TypeVarTuple, TypedDict, overload
 from typing_extensions import TypeVar
 
 STOPPING_STAGE_SHUTDOWN_TIMEOUT: int
@@ -39,6 +40,7 @@ _R_co = TypeVar('_R_co', covariant=True)
 _P = ParamSpec('_P')
 _Ts = TypeVarTuple('_Ts')
 _UNDEF: dict[Any, Any]
+_SENTINEL: Incomplete
 _CallableT = TypeVar('_CallableT', bound=Callable[..., Any])
 _DataT = TypeVar('_DataT', bound=Mapping[str, Any], default=Mapping[str, Any])
 CALLBACK_TYPE = Callable[[], None]
@@ -56,6 +58,11 @@ class ConfigSource(enum.StrEnum):
     STORAGE: str
     YAML: str
 
+class EventStateChangedData(TypedDict):
+    entity_id: str
+    old_state: State | None
+    new_state: State | None
+
 _DEPRECATED_SOURCE_DISCOVERED: Incomplete
 _DEPRECATED_SOURCE_STORAGE: Incomplete
 _DEPRECATED_SOURCE_YAML: Incomplete
@@ -84,7 +91,14 @@ class _Hass(threading.local):
 _hass: Incomplete
 
 def async_get_hass() -> HomeAssistant: ...
-def get_release_channel() -> Literal['beta', 'dev', 'nightly', 'stable']: ...
+
+class ReleaseChannel(enum.StrEnum):
+    BETA: str
+    DEV: str
+    NIGHTLY: str
+    STABLE: str
+
+def get_release_channel() -> ReleaseChannel: ...
 
 class HassJobType(enum.Enum):
     Coroutinefunction: int
@@ -124,7 +138,7 @@ class HomeAssistant:
     auth: AuthManager
     http: HomeAssistantHTTP
     config_entries: ConfigEntries
-    def __new__(cls, config_dir: str) -> HomeAssistant: ...
+    def __new__(cls, config_dir: str) -> Self: ...
     def __repr__(self) -> str: ...
     data: Incomplete
     loop: Incomplete
@@ -144,6 +158,7 @@ class HomeAssistant:
     _shutdown_jobs: Incomplete
     import_executor: Incomplete
     def __init__(self, config_dir: str) -> None: ...
+    def verify_event_loop_thread(self, what: str) -> None: ...
     @property
     def _active_tasks(self) -> set[asyncio.Future[Any]]: ...
     @cached_property
@@ -165,9 +180,14 @@ class HomeAssistant:
     def async_add_hass_job(self, hassjob: HassJob[..., Coroutine[Any, Any, _R]], *args: Any, eager_start: bool = False, background: bool = False) -> asyncio.Future[_R] | None: ...
     @overload
     def async_add_hass_job(self, hassjob: HassJob[..., Coroutine[Any, Any, _R] | _R], *args: Any, eager_start: bool = False, background: bool = False) -> asyncio.Future[_R] | None: ...
+    @overload
+    def _async_add_hass_job(self, hassjob: HassJob[..., Coroutine[Any, Any, _R]], *args: Any, eager_start: bool = False, background: bool = False) -> asyncio.Future[_R] | None: ...
+    @overload
+    def _async_add_hass_job(self, hassjob: HassJob[..., Coroutine[Any, Any, _R] | _R], *args: Any, eager_start: bool = False, background: bool = False) -> asyncio.Future[_R] | None: ...
     def create_task(self, target: Coroutine[Any, Any, Any], name: str | None = None) -> None: ...
-    def async_create_task(self, target: Coroutine[Any, Any, _R], name: str | None = None, eager_start: bool = False) -> asyncio.Task[_R]: ...
-    def async_create_background_task(self, target: Coroutine[Any, Any, _R], name: str, eager_start: bool = False) -> asyncio.Task[_R]: ...
+    def async_create_task(self, target: Coroutine[Any, Any, _R], name: str | None = None, eager_start: bool = True) -> asyncio.Task[_R]: ...
+    def async_create_task_internal(self, target: Coroutine[Any, Any, _R], name: str | None = None, eager_start: bool = True) -> asyncio.Task[_R]: ...
+    def async_create_background_task(self, target: Coroutine[Any, Any, _R], name: str, eager_start: bool = True) -> asyncio.Task[_R]: ...
     def async_add_executor_job(self, target: Callable[[Unpack[_Ts]], _T], *args: Unpack[_Ts]) -> asyncio.Future[_T]: ...
     def async_add_import_executor_job(self, target: Callable[[Unpack[_Ts]], _T], *args: Unpack[_Ts]) -> asyncio.Future[_T]: ...
     @overload
@@ -198,7 +218,7 @@ class Context:
     parent_id: Incomplete
     origin_event: Incomplete
     def __init__(self, user_id: str | None = None, parent_id: str | None = None, id: str | None = None) -> None: ...
-    def __eq__(self, other: Any) -> bool: ...
+    def __eq__(self, other: object) -> bool: ...
     @cached_property
     def _as_dict(self) -> dict[str, str | None]: ...
     def as_dict(self) -> ReadOnlyDict[str, str | None]: ...
@@ -218,7 +238,7 @@ class Event(Generic[_DataT]):
     origin: Incomplete
     time_fired_timestamp: Incomplete
     context: Incomplete
-    def __init__(self, event_type: str, data: _DataT | None = None, origin: EventOrigin = ..., time_fired_timestamp: float | None = None, context: Context | None = None) -> None: ...
+    def __init__(self, event_type: EventType[_DataT] | str, data: _DataT | None = None, origin: EventOrigin = ..., time_fired_timestamp: float | None = None, context: Context | None = None) -> None: ...
     @cached_property
     def time_fired(self) -> datetime.datetime: ...
     @cached_property
@@ -230,21 +250,23 @@ class Event(Generic[_DataT]):
     def json_fragment(self) -> json_fragment: ...
     def __repr__(self) -> str: ...
 
-def _event_repr(event_type: str, origin: EventOrigin, data: Mapping[str, Any] | None) -> str: ...
+def _event_repr(event_type: EventType[_DataT] | str, origin: EventOrigin, data: _DataT | None) -> str: ...
 
 _FilterableJobType: Incomplete
 
 @dataclass(slots=True)
-class _OneTimeListener:
+class _OneTimeListener(Generic[_DataT]):
     hass: HomeAssistant
-    listener_job: HassJob[[Event], Coroutine[Any, Any, None] | None]
+    listener_job: HassJob[[Event[_DataT]], Coroutine[Any, Any, None] | None]
     remove: CALLBACK_TYPE | None = ...
-    def __call__(self, event: Event) -> None: ...
+    def __call__(self, event: Event[_DataT]) -> None: ...
     def __repr__(self) -> str: ...
     def __init__(self, hass, listener_job, remove) -> None: ...
 
 EMPTY_LIST: list[Any]
 
+def _verify_event_type_length_or_raise(event_type: EventType[_DataT] | str) -> None: ...
+
 class EventBus:
     __slots__: Incomplete
     _listeners: Incomplete
@@ -253,18 +275,18 @@ class EventBus:
     def __init__(self, hass: HomeAssistant) -> None: ...
     _debug: Incomplete
     def _async_logging_changed(self, event: Event | None = None) -> None: ...
-    def async_listeners(self) -> dict[str, int]: ...
+    def async_listeners(self) -> dict[EventType[Any] | str, int]: ...
     @property
-    def listeners(self) -> dict[str, int]: ...
-    def fire(self, event_type: str, event_data: Mapping[str, Any] | None = None, origin: EventOrigin = ..., context: Context | None = None) -> None: ...
-    def async_fire(self, event_type: str, event_data: Mapping[str, Any] | None = None, origin: EventOrigin = ..., context: Context | None = None, time_fired: float | None = None) -> None: ...
-    def _async_fire(self, event_type: str, event_data: Mapping[str, Any] | None = None, origin: EventOrigin = ..., context: Context | None = None, time_fired: float | None = None) -> None: ...
-    def listen(self, event_type: str, listener: Callable[[Event[Any]], Coroutine[Any, Any, None] | None]) -> CALLBACK_TYPE: ...
-    def async_listen(self, event_type: str, listener: Callable[[Event[_DataT]], Coroutine[Any, Any, None] | None], event_filter: Callable[[_DataT], bool] | None = None, run_immediately: bool = False) -> CALLBACK_TYPE: ...
-    def _async_listen_filterable_job(self, event_type: str, filterable_job: _FilterableJobType[Any]) -> CALLBACK_TYPE: ...
-    def listen_once(self, event_type: str, listener: Callable[[Event[Any]], Coroutine[Any, Any, None] | None]) -> CALLBACK_TYPE: ...
-    def async_listen_once(self, event_type: str, listener: Callable[[Event[Any]], Coroutine[Any, Any, None] | None], run_immediately: bool = False) -> CALLBACK_TYPE: ...
-    def _async_remove_listener(self, event_type: str, filterable_job: _FilterableJobType) -> None: ...
+    def listeners(self) -> dict[EventType[Any] | str, int]: ...
+    def fire(self, event_type: EventType[_DataT] | str, event_data: _DataT | None = None, origin: EventOrigin = ..., context: Context | None = None) -> None: ...
+    def async_fire(self, event_type: EventType[_DataT] | str, event_data: _DataT | None = None, origin: EventOrigin = ..., context: Context | None = None, time_fired: float | None = None) -> None: ...
+    def async_fire_internal(self, event_type: EventType[_DataT] | str, event_data: _DataT | None = None, origin: EventOrigin = ..., context: Context | None = None, time_fired: float | None = None) -> None: ...
+    def listen(self, event_type: EventType[_DataT] | str, listener: Callable[[Event[_DataT]], Coroutine[Any, Any, None] | None]) -> CALLBACK_TYPE: ...
+    def async_listen(self, event_type: EventType[_DataT] | str, listener: Callable[[Event[_DataT]], Coroutine[Any, Any, None] | None], event_filter: Callable[[_DataT], bool] | None = None, run_immediately: bool | object = ...) -> CALLBACK_TYPE: ...
+    def _async_listen_filterable_job(self, event_type: EventType[_DataT] | str, filterable_job: _FilterableJobType[_DataT]) -> CALLBACK_TYPE: ...
+    def listen_once(self, event_type: EventType[_DataT] | str, listener: Callable[[Event[_DataT]], Coroutine[Any, Any, None] | None]) -> CALLBACK_TYPE: ...
+    def async_listen_once(self, event_type: EventType[_DataT] | str, listener: Callable[[Event[_DataT]], Coroutine[Any, Any, None] | None], run_immediately: bool | object = ...) -> CALLBACK_TYPE: ...
+    def _async_remove_listener(self, event_type: EventType[_DataT] | str, filterable_job: _FilterableJobType[_DataT]) -> None: ...
 
 class CompressedState(TypedDict):
     s: str
@@ -376,19 +398,29 @@ class ServiceRegistry:
     def supports_response(self, domain: str, service: str) -> SupportsResponse: ...
     def register(self, domain: str, service: str, service_func: Callable[[ServiceCall], Coroutine[Any, Any, ServiceResponse] | ServiceResponse | None], schema: vol.Schema | None = None, supports_response: SupportsResponse = ...) -> None: ...
     def async_register(self, domain: str, service: str, service_func: Callable[[ServiceCall], Coroutine[Any, Any, ServiceResponse | EntityServiceResponse] | ServiceResponse | EntityServiceResponse | None], schema: vol.Schema | None = None, supports_response: SupportsResponse = ..., job_type: HassJobType | None = None) -> None: ...
+    def _async_register(self, domain: str, service: str, service_func: Callable[[ServiceCall], Coroutine[Any, Any, ServiceResponse | EntityServiceResponse] | ServiceResponse | EntityServiceResponse | None], schema: vol.Schema | None = None, supports_response: SupportsResponse = ..., job_type: HassJobType | None = None) -> None: ...
     def remove(self, domain: str, service: str) -> None: ...
     def async_remove(self, domain: str, service: str) -> None: ...
+    def _async_remove(self, domain: str, service: str) -> None: ...
     def call(self, domain: str, service: str, service_data: dict[str, Any] | None = None, blocking: bool = False, context: Context | None = None, target: dict[str, Any] | None = None, return_response: bool = False) -> ServiceResponse: ...
     async def async_call(self, domain: str, service: str, service_data: dict[str, Any] | None = None, blocking: bool = False, context: Context | None = None, target: dict[str, Any] | None = None, return_response: bool = False) -> ServiceResponse: ...
     async def _run_service_call_catch_exceptions(self, coro_or_task: Coroutine[Any, Any, Any] | asyncio.Task[Any], service_call: ServiceCall) -> None: ...
     async def _execute_service(self, handler: Service, service_call: ServiceCall) -> ServiceResponse: ...
 
+class _ComponentSet(set[str]):
+    _top_level_components: Incomplete
+    def __init__(self, top_level_components: set[str]) -> None: ...
+    def add(self, component: str) -> None: ...
+    def remove(self, component: str) -> None: ...
+    def discard(self, component: str) -> None: ...
+
 class Config:
     _store: Config._ConfigStore
     hass: Incomplete
     latitude: int
     longitude: int
     elevation: int
+    debug: bool
     location_name: str
     time_zone: str
     units: Incomplete
@@ -400,6 +432,7 @@ class Config:
     config_source: Incomplete
     skip_pip: bool
     skip_pip_packages: Incomplete
+    top_level_components: Incomplete
     components: Incomplete
     api: Incomplete
     config_dir: Incomplete

homeassistant-stubs/data_entry_flow.pyi

@@ -7,7 +7,7 @@ from .helpers.deprecation import DeprecatedConstantEnum as DeprecatedConstantEnu
 from .helpers.frame import report as report
 from .loader import async_suggest_report_issue as async_suggest_report_issue
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Iterable, Mapping
+from collections.abc import Callable as Callable, Container, Iterable, Mapping
 from dataclasses import dataclass
 from enum import StrEnum
 from typing import Any, Generic, Required, TypedDict
@@ -66,7 +66,7 @@ class FlowResult(TypedDict, Generic[_HandlerT], total=False):
     flow_id: Required[str]
     handler: Required[_HandlerT]
     last_step: bool | None
-    menu_options: list[str] | dict[str, str]
+    menu_options: Container[str]
     options: Mapping[str, Any]
     preview: str | None
     progress_action: str
@@ -139,7 +139,7 @@ class FlowHandler(Generic[_FlowResultT, _HandlerT]):
     def async_external_step_done(self, *, next_step_id: str) -> _FlowResultT: ...
     def async_show_progress(self, *, step_id: str | None = None, progress_action: str, description_placeholders: Mapping[str, str] | None = None, progress_task: asyncio.Task[Any] | None = None) -> _FlowResultT: ...
     def async_show_progress_done(self, *, next_step_id: str) -> _FlowResultT: ...
-    def async_show_menu(self, *, step_id: str | None = None, menu_options: list[str] | dict[str, str], description_placeholders: Mapping[str, str] | None = None) -> _FlowResultT: ...
+    def async_show_menu(self, *, step_id: str | None = None, menu_options: Container[str], description_placeholders: Mapping[str, str] | None = None) -> _FlowResultT: ...
     def async_remove(self) -> None: ...
     @staticmethod
     async def async_setup_preview(hass: HomeAssistant) -> None: ...

homeassistant-stubs/exceptions.pyi

@@ -1,7 +1,9 @@
 from .core import Context as Context
+from .util.event_type import EventType as EventType
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Generator, Sequence
 from dataclasses import dataclass
+from typing import Any
 
 _function_cache: dict[str, Callable[[str, str, dict[str, str] | None], str]]
 
@@ -76,7 +78,7 @@ class Unauthorized(HomeAssistantError):
 
 class UnknownUser(Unauthorized): ...
 
-class ServiceNotFound(HomeAssistantError):
+class ServiceNotFound(ServiceValidationError):
     domain: Incomplete
     service: Incomplete
     generate_message: bool
@@ -87,7 +89,7 @@ class MaxLengthExceeded(HomeAssistantError):
     property_name: Incomplete
     max_length: Incomplete
     generate_message: bool
-    def __init__(self, value: str, property_name: str, max_length: int) -> None: ...
+    def __init__(self, value: EventType[Any] | str, property_name: str, max_length: int) -> None: ...
 
 class DependencyError(HomeAssistantError):
     failed_dependencies: Incomplete

homeassistant-stubs/helpers/aiohttp_client.pyi

@@ -1,4 +1,5 @@
 import aiohttp
+from .backports.aiohttp_resolver import AsyncResolver as AsyncResolver
 from .frame import warn_use as warn_use
 from .json import json_dumps as json_dumps
 from _typeshed import Incomplete

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