homeassistant-stubs 2024.3.2__py3-none-any.whl → 2024.4.0__py3-none-any.whl

homeassistant-stubs/data_entry_flow.pyi

@@ -10,7 +10,8 @@ from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable, Mapping
 from dataclasses import dataclass
 from enum import StrEnum
-from typing import Any, Required, TypedDict
+from typing import Any, Generic, Required, TypedDict
+from typing_extensions import TypeVar
 
 _LOGGER: Incomplete
 
@@ -35,6 +36,8 @@ _DEPRECATED_RESULT_TYPE_MENU: Incomplete
 EVENT_DATA_ENTRY_FLOW_PROGRESSED: str
 FLOW_NOT_COMPLETE_STEPS: Incomplete
 STEP_ID_OPTIONAL_STEPS: Incomplete
+_FlowResultT = TypeVar('_FlowResultT', bound='FlowResult[Any]', default='FlowResult')
+_HandlerT = TypeVar('_HandlerT', default=str)
 
 @dataclass(slots=True)
 class BaseServiceInfo: ...
@@ -52,7 +55,7 @@ class AbortFlow(FlowError):
     description_placeholders: Incomplete
     def __init__(self, reason: str, description_placeholders: Mapping[str, str] | None = None) -> None: ...
 
-class FlowResult(TypedDict, total=False):
+class FlowResult(TypedDict, Generic[_HandlerT], total=False):
     context: dict[str, Any]
     data_schema: vol.Schema | None
     data: Mapping[str, Any]
@@ -61,10 +64,9 @@ class FlowResult(TypedDict, total=False):
     errors: dict[str, str] | None
     extra: str
     flow_id: Required[str]
-    handler: Required[str]
+    handler: Required[_HandlerT]
     last_step: bool | None
     menu_options: list[str] | dict[str, str]
-    minor_version: int
     options: Mapping[str, Any]
     preview: str | None
     progress_action: str
@@ -77,12 +79,11 @@ class FlowResult(TypedDict, total=False):
     translation_domain: str
     type: FlowResultType
     url: str
-    version: int
 
-def _async_flow_handler_to_flow_result(flows: Iterable[FlowHandler], include_uninitialized: bool) -> list[FlowResult]: ...
 def _map_error_to_schema_errors(schema_errors: dict[str, Any], error: vol.Invalid, data_schema: vol.Schema) -> None: ...
 
-class FlowManager(abc.ABC, metaclass=abc.ABCMeta):
+class FlowManager(abc.ABC, Generic[_FlowResultT, _HandlerT], metaclass=abc.ABCMeta):
+    _flow_result: type[_FlowResultT]
     hass: Incomplete
     _preview: Incomplete
     _progress: Incomplete
@@ -90,32 +91,34 @@ class FlowManager(abc.ABC, metaclass=abc.ABCMeta):
     _init_data_process_index: Incomplete
     def __init__(self, hass: HomeAssistant) -> None: ...
     @abc.abstractmethod
-    async def async_create_flow(self, handler_key: str, *, context: dict[str, Any] | None = None, data: dict[str, Any] | None = None) -> FlowHandler: ...
+    async def async_create_flow(self, handler_key: _HandlerT, *, context: dict[str, Any] | None = None, data: dict[str, Any] | None = None) -> FlowHandler[_FlowResultT, _HandlerT]: ...
     @abc.abstractmethod
-    async def async_finish_flow(self, flow: FlowHandler, result: FlowResult) -> FlowResult: ...
-    async def async_post_init(self, flow: FlowHandler, result: FlowResult) -> None: ...
-    def async_has_matching_flow(self, handler: str, match_context: dict[str, Any], data: Any) -> bool: ...
-    def async_get(self, flow_id: str) -> FlowResult: ...
-    def async_progress(self, include_uninitialized: bool = False) -> list[FlowResult]: ...
-    def async_progress_by_handler(self, handler: str, include_uninitialized: bool = False, match_context: dict[str, Any] | None = None) -> list[FlowResult]: ...
-    def async_progress_by_init_data_type(self, init_data_type: type, matcher: Callable[[Any], bool], include_uninitialized: bool = False) -> list[FlowResult]: ...
-    def _async_progress_by_handler(self, handler: str, match_context: dict[str, Any] | None) -> list[FlowHandler]: ...
-    async def async_init(self, handler: str, *, context: dict[str, Any] | None = None, data: Any = None) -> FlowResult: ...
-    async def async_configure(self, flow_id: str, user_input: dict | None = None) -> FlowResult: ...
-    async def _async_configure(self, flow_id: str, user_input: dict | None = None) -> FlowResult: ...
+    async def async_finish_flow(self, flow: FlowHandler[_FlowResultT, _HandlerT], result: _FlowResultT) -> _FlowResultT: ...
+    async def async_post_init(self, flow: FlowHandler[_FlowResultT, _HandlerT], result: _FlowResultT) -> None: ...
+    def async_has_matching_flow(self, handler: _HandlerT, match_context: dict[str, Any], data: Any) -> bool: ...
+    def async_get(self, flow_id: str) -> _FlowResultT: ...
+    def async_progress(self, include_uninitialized: bool = False) -> list[_FlowResultT]: ...
+    def async_progress_by_handler(self, handler: _HandlerT, include_uninitialized: bool = False, match_context: dict[str, Any] | None = None) -> list[_FlowResultT]: ...
+    def async_progress_by_init_data_type(self, init_data_type: type, matcher: Callable[[Any], bool], include_uninitialized: bool = False) -> list[_FlowResultT]: ...
+    def _async_progress_by_handler(self, handler: _HandlerT, match_context: dict[str, Any] | None) -> list[FlowHandler[_FlowResultT, _HandlerT]]: ...
+    async def async_init(self, handler: _HandlerT, *, context: dict[str, Any] | None = None, data: Any = None) -> _FlowResultT: ...
+    async def async_configure(self, flow_id: str, user_input: dict | None = None) -> _FlowResultT: ...
+    async def _async_configure(self, flow_id: str, user_input: dict | None = None) -> _FlowResultT: ...
     def async_abort(self, flow_id: str) -> None: ...
-    def _async_add_flow_progress(self, flow: FlowHandler) -> None: ...
-    def _async_remove_flow_from_index(self, flow: FlowHandler) -> None: ...
+    def _async_add_flow_progress(self, flow: FlowHandler[_FlowResultT, _HandlerT]) -> None: ...
+    def _async_remove_flow_from_index(self, flow: FlowHandler[_FlowResultT, _HandlerT]) -> None: ...
     def _async_remove_flow_progress(self, flow_id: str) -> None: ...
-    async def _async_handle_step(self, flow: FlowHandler, step_id: str, user_input: dict | BaseServiceInfo | None) -> FlowResult: ...
-    def _raise_if_step_does_not_exist(self, flow: FlowHandler, step_id: str) -> None: ...
-    async def _async_setup_preview(self, flow: FlowHandler) -> None: ...
+    async def _async_handle_step(self, flow: FlowHandler[_FlowResultT, _HandlerT], step_id: str, user_input: dict | BaseServiceInfo | None) -> _FlowResultT: ...
+    def _raise_if_step_does_not_exist(self, flow: FlowHandler[_FlowResultT, _HandlerT], step_id: str) -> None: ...
+    async def _async_setup_preview(self, flow: FlowHandler[_FlowResultT, _HandlerT]) -> None: ...
+    def _async_flow_handler_to_flow_result(self, flows: Iterable[FlowHandler[_FlowResultT, _HandlerT]], include_uninitialized: bool) -> list[_FlowResultT]: ...
 
-class FlowHandler:
-    cur_step: FlowResult | None
+class FlowHandler(Generic[_FlowResultT, _HandlerT]):
+    _flow_result: type[_FlowResultT]
+    cur_step: _FlowResultT | None
     flow_id: str
     hass: HomeAssistant
-    handler: str
+    handler: _HandlerT
     context: dict[str, Any]
     init_step: str
     init_data: Any
@@ -129,14 +132,14 @@ class FlowHandler:
     @property
     def show_advanced_options(self) -> bool: ...
     def add_suggested_values_to_schema(self, data_schema: vol.Schema, suggested_values: Mapping[str, Any] | None) -> vol.Schema: ...
-    def async_show_form(self, *, step_id: str | None = None, data_schema: vol.Schema | None = None, errors: dict[str, str] | None = None, description_placeholders: Mapping[str, str | None] | None = None, last_step: bool | None = None, preview: str | None = None) -> FlowResult: ...
-    def async_create_entry(self, *, title: str | None = None, data: Mapping[str, Any], description: str | None = None, description_placeholders: Mapping[str, str] | None = None) -> FlowResult: ...
-    def async_abort(self, *, reason: str, description_placeholders: Mapping[str, str] | None = None) -> FlowResult: ...
-    def async_external_step(self, *, step_id: str | None = None, url: str, description_placeholders: Mapping[str, str] | None = None) -> FlowResult: ...
-    def async_external_step_done(self, *, next_step_id: str) -> FlowResult: ...
-    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) -> FlowResult: ...
-    def async_show_progress_done(self, *, next_step_id: str) -> FlowResult: ...
-    def async_show_menu(self, *, step_id: str | None = None, menu_options: list[str] | dict[str, str], description_placeholders: Mapping[str, str] | None = None) -> FlowResult: ...
+    def async_show_form(self, *, step_id: str | None = None, data_schema: vol.Schema | None = None, errors: dict[str, str] | None = None, description_placeholders: Mapping[str, str | None] | None = None, last_step: bool | None = None, preview: str | None = None) -> _FlowResultT: ...
+    def async_create_entry(self, *, title: str | None = None, data: Mapping[str, Any], description: str | None = None, description_placeholders: Mapping[str, str] | None = None) -> _FlowResultT: ...
+    def async_abort(self, *, reason: str, description_placeholders: Mapping[str, str] | None = None) -> _FlowResultT: ...
+    def async_external_step(self, *, step_id: str | None = None, url: str, description_placeholders: Mapping[str, str] | None = None) -> _FlowResultT: ...
+    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_remove(self) -> None: ...
     @staticmethod
     async def async_setup_preview(hass: HomeAssistant) -> None: ...
@@ -144,8 +147,6 @@ class FlowHandler:
     def async_get_progress_task(self) -> asyncio.Task[Any] | None: ...
     def async_set_progress_task(self, progress_task: asyncio.Task[Any]) -> None: ...
 
-def _create_abort_data(flow_id: str, handler: str, reason: str, description_placeholders: Mapping[str, str] | None = None) -> FlowResult: ...
-
 __getattr__: Incomplete
 __dir__: Incomplete
 __all__: Incomplete

homeassistant-stubs/exceptions.pyi

@@ -1,18 +1,24 @@
 from .core import Context as Context
 from _typeshed import Incomplete
-from collections.abc import Generator, Sequence
+from collections.abc import Callable as Callable, Generator, Sequence
 from dataclasses import dataclass
 
+_function_cache: dict[str, Callable[[str, str, dict[str, str] | None], str]]
+
+def import_async_get_exception_message() -> Callable[[str, str, dict[str, str] | None], str]: ...
+
 class HomeAssistantError(Exception):
+    _message: str | None
+    generate_message: bool
     translation_domain: Incomplete
     translation_key: Incomplete
     translation_placeholders: Incomplete
     def __init__(self, *args: object, translation_domain: str | None = None, translation_key: str | None = None, translation_placeholders: dict[str, str] | None = None) -> None: ...
+    def __str__(self) -> str: ...
 
 class ConfigValidationError(HomeAssistantError, ExceptionGroup[Exception]):
-    _message: Incomplete
-    def __init__(self, message: str, exceptions: list[Exception], translation_domain: str | None = None, translation_key: str | None = None, translation_placeholders: dict[str, str] | None = None) -> None: ...
-    def __str__(self) -> str: ...
+    generate_message: bool
+    def __init__(self, message_translation_key: str, exceptions: list[Exception], translation_domain: str | None = None, translation_placeholders: dict[str, str] | None = None) -> None: ...
 
 class ServiceValidationError(HomeAssistantError): ...
 class InvalidEntityFormatError(HomeAssistantError): ...
@@ -73,13 +79,14 @@ class UnknownUser(Unauthorized): ...
 class ServiceNotFound(HomeAssistantError):
     domain: Incomplete
     service: Incomplete
+    generate_message: bool
     def __init__(self, domain: str, service: str) -> None: ...
-    def __str__(self) -> str: ...
 
 class MaxLengthExceeded(HomeAssistantError):
     value: Incomplete
     property_name: Incomplete
     max_length: Incomplete
+    generate_message: bool
     def __init__(self, value: str, property_name: str, max_length: int) -> None: ...
 
 class DependencyError(HomeAssistantError):

homeassistant-stubs/helpers/area_registry.pyi

@@ -1,9 +1,10 @@
 import dataclasses
+from .normalized_name_base_registry import NormalizedNameBaseRegistryEntry as NormalizedNameBaseRegistryEntry, NormalizedNameBaseRegistryItems as NormalizedNameBaseRegistryItems, normalize_name as normalize_name
+from .registry import BaseRegistry as BaseRegistry
 from .storage import Store as Store
 from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
-from collections import UserDict
-from collections.abc import Iterable, ValuesView
+from collections.abc import Iterable
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util import slugify as slugify
 from typing import Any, Literal, TypedDict
@@ -13,37 +14,26 @@ EVENT_AREA_REGISTRY_UPDATED: str
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 STORAGE_VERSION_MINOR: int
-SAVE_DELAY: int
 
 class EventAreaRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     area_id: str
 
 @dataclasses.dataclass(frozen=True, kw_only=True, slots=True)
-class AreaEntry:
+class AreaEntry(NormalizedNameBaseRegistryEntry):
     aliases: set[str]
     floor_id: str | None
     icon: str | None
     id: str
     labels: set[str] = ...
-    name: str
-    normalized_name: str
     picture: str | None
-    def __init__(self, *, aliases, floor_id, icon, id, labels, name, normalized_name, picture) -> None: ...
-
-class AreaRegistryItems(UserDict[str, AreaEntry]):
-    _normalized_names: Incomplete
-    def __init__(self) -> None: ...
-    def values(self) -> ValuesView[AreaEntry]: ...
-    def __setitem__(self, key: str, entry: AreaEntry) -> None: ...
-    def __delitem__(self, key: str) -> None: ...
-    def get_area_by_name(self, name: str) -> AreaEntry | 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 AreaRegistry:
-    areas: AreaRegistryItems
+class AreaRegistry(BaseRegistry):
+    areas: NormalizedNameBaseRegistryItems[AreaEntry]
     _area_data: dict[str, AreaEntry]
     hass: Incomplete
     _store: Incomplete
@@ -57,7 +47,6 @@ class AreaRegistry:
     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 async_schedule_save(self) -> None: ...
     def _data_to_save(self) -> dict[str, list[dict[str, Any]]]: ...
     def _generate_area_id(self, name: str) -> str: ...
     def _async_setup_cleanup(self) -> None: ...
@@ -66,4 +55,3 @@ def async_get(hass: HomeAssistant) -> AreaRegistry: ...
 async def async_load(hass: HomeAssistant) -> None: ...
 def async_entries_for_floor(registry: AreaRegistry, floor_id: str) -> list[AreaEntry]: ...
 def async_entries_for_label(registry: AreaRegistry, label_id: str) -> list[AreaEntry]: ...
-def normalize_area_name(area_name: str) -> str: ...

homeassistant-stubs/helpers/category_registry.pyi

@@ -0,0 +1,43 @@
+from .registry import BaseRegistry as BaseRegistry
+from .typing import EventType as EventType, 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.util.ulid import ulid_now as ulid_now
+from typing import Literal, TypedDict
+
+DATA_REGISTRY: str
+EVENT_CATEGORY_REGISTRY_UPDATED: str
+STORAGE_KEY: str
+STORAGE_VERSION_MAJOR: int
+
+class EventCategoryRegistryUpdatedData(TypedDict):
+    action: Literal['create', 'remove', 'update']
+    scope: str
+    category_id: str
+EventCategoryRegistryUpdated = EventType[EventCategoryRegistryUpdatedData]
+
+@dataclass(slots=True, kw_only=True, frozen=True)
+class CategoryEntry:
+    category_id: str = ...
+    icon: str | None = ...
+    name: str
+    def __init__(self, *, category_id, icon, name) -> None: ...
+
+class CategoryRegistry(BaseRegistry):
+    hass: Incomplete
+    categories: Incomplete
+    _store: Incomplete
+    def __init__(self, hass: HomeAssistant) -> None: ...
+    def async_get_category(self, *, scope: str, category_id: str) -> CategoryEntry | None: ...
+    def async_list_categories(self, *, scope: str) -> Iterable[CategoryEntry]: ...
+    def async_create(self, *, name: str, scope: str, icon: str | None = None) -> CategoryEntry: ...
+    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 _async_ensure_name_is_available(self, scope: str, name: str, category_id: str | None = None) -> None: ...
+
+def async_get(hass: HomeAssistant) -> CategoryRegistry: ...
+async def async_load(hass: HomeAssistant) -> None: ...

homeassistant-stubs/helpers/check_config.pyi

@@ -2,7 +2,7 @@ from .typing import ConfigType as ConfigType
 from _typeshed import Incomplete
 from collections import OrderedDict
 from homeassistant import loader as loader
-from homeassistant.config import CONF_CORE as CONF_CORE, CONF_PACKAGES as CONF_PACKAGES, CORE_CONFIG_SCHEMA as CORE_CONFIG_SCHEMA, YAML_CONFIG_FILE as YAML_CONFIG_FILE, config_per_platform as config_per_platform, extract_domain_configs as extract_domain_configs, format_homeassistant_error as format_homeassistant_error, format_schema_error as format_schema_error, load_yaml_config_file as load_yaml_config_file, merge_packages_config as merge_packages_config
+from homeassistant.config import CONF_PACKAGES as CONF_PACKAGES, CORE_CONFIG_SCHEMA as CORE_CONFIG_SCHEMA, YAML_CONFIG_FILE as YAML_CONFIG_FILE, config_per_platform as config_per_platform, extract_domain_configs as extract_domain_configs, format_homeassistant_error as format_homeassistant_error, format_schema_error as format_schema_error, load_yaml_config_file as load_yaml_config_file, merge_packages_config as merge_packages_config
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.requirements import RequirementsNotFound as RequirementsNotFound, async_clear_install_history as async_clear_install_history, async_get_integration_with_requirements as async_get_integration_with_requirements

homeassistant-stubs/helpers/config_entry_flow.pyi

@@ -9,7 +9,6 @@ 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 homeassistant.data_entry_flow import FlowResult as FlowResult
 from typing import Any, Generic, TypeVar
 
 _R = TypeVar('_R', bound='Awaitable[bool] | bool')
@@ -22,16 +21,16 @@ class DiscoveryFlowHandler(config_entries.ConfigFlow, Generic[_R]):
     _title: Incomplete
     _discovery_function: Incomplete
     def __init__(self, domain: str, title: str, discovery_function: DiscoveryFunctionType[_R]) -> None: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_confirm(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_discovery(self, discovery_info: DiscoveryInfoType) -> FlowResult: ...
-    async def async_step_bluetooth(self, discovery_info: BluetoothServiceInfoBleak) -> FlowResult: ...
-    async def async_step_dhcp(self, discovery_info: DhcpServiceInfo) -> FlowResult: ...
-    async def async_step_homekit(self, discovery_info: ZeroconfServiceInfo) -> FlowResult: ...
-    async def async_step_mqtt(self, discovery_info: MqttServiceInfo) -> FlowResult: ...
-    async def async_step_zeroconf(self, discovery_info: ZeroconfServiceInfo) -> FlowResult: ...
-    async def async_step_ssdp(self, discovery_info: SsdpServiceInfo) -> FlowResult: ...
-    async def async_step_import(self, _: dict[str, Any] | None) -> FlowResult: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> config_entries.ConfigFlowResult: ...
+    async def async_step_confirm(self, user_input: dict[str, Any] | None = None) -> config_entries.ConfigFlowResult: ...
+    async def async_step_discovery(self, discovery_info: DiscoveryInfoType) -> config_entries.ConfigFlowResult: ...
+    async def async_step_bluetooth(self, discovery_info: BluetoothServiceInfoBleak) -> config_entries.ConfigFlowResult: ...
+    async def async_step_dhcp(self, discovery_info: DhcpServiceInfo) -> config_entries.ConfigFlowResult: ...
+    async def async_step_homekit(self, discovery_info: ZeroconfServiceInfo) -> config_entries.ConfigFlowResult: ...
+    async def async_step_mqtt(self, discovery_info: MqttServiceInfo) -> config_entries.ConfigFlowResult: ...
+    async def async_step_zeroconf(self, discovery_info: ZeroconfServiceInfo) -> config_entries.ConfigFlowResult: ...
+    async def async_step_ssdp(self, discovery_info: SsdpServiceInfo) -> config_entries.ConfigFlowResult: ...
+    async def async_step_import(self, _: dict[str, Any] | None) -> config_entries.ConfigFlowResult: ...
 
 def register_discovery_flow(domain: str, title: str, discovery_function: DiscoveryFunctionType[Awaitable[bool] | bool]) -> None: ...
 
@@ -42,7 +41,7 @@ class WebhookFlowHandler(config_entries.ConfigFlow):
     _description_placeholder: Incomplete
     _allow_multiple: Incomplete
     def __init__(self, domain: str, title: str, description_placeholder: dict, allow_multiple: bool) -> None: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> config_entries.ConfigFlowResult: ...
 
 def register_webhook_flow(domain: str, title: str, description_placeholder: dict, allow_multiple: bool = False) -> None: ...
 async def webhook_async_remove_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry) -> None: ...

homeassistant-stubs/helpers/config_entry_oauth2_flow.pyi

@@ -9,7 +9,6 @@ from collections.abc import Awaitable, Callable as Callable
 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.data_entry_flow import FlowResult as FlowResult
 from homeassistant.loader import async_get_application_credentials as async_get_application_credentials
 from typing import Any
 
@@ -72,12 +71,12 @@ class AbstractOAuth2FlowHandler(config_entries.ConfigFlow, metaclass=ABCMeta):
     @property
     def extra_authorize_data(self) -> dict: ...
     async def async_generate_authorize_url(self) -> str: ...
-    async def async_step_pick_implementation(self, user_input: dict | None = None) -> FlowResult: ...
-    async def async_step_auth(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_creation(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_authorize_rejected(self, data: None = None) -> FlowResult: ...
-    async def async_oauth_create_entry(self, data: dict) -> FlowResult: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def async_step_pick_implementation(self, user_input: dict | None = None) -> config_entries.ConfigFlowResult: ...
+    async def async_step_auth(self, user_input: dict[str, Any] | None = None) -> config_entries.ConfigFlowResult: ...
+    async def async_step_creation(self, user_input: dict[str, Any] | None = None) -> config_entries.ConfigFlowResult: ...
+    async def async_step_authorize_rejected(self, data: None = None) -> config_entries.ConfigFlowResult: ...
+    async def async_oauth_create_entry(self, data: dict) -> config_entries.ConfigFlowResult: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> config_entries.ConfigFlowResult: ...
     @classmethod
     def async_register_implementation(cls, hass: HomeAssistant, local_impl: LocalOAuth2Implementation) -> None: ...
 

homeassistant-stubs/helpers/config_validation.pyi

@@ -6,7 +6,7 @@ from _typeshed import Incomplete
 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, 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.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.exceptions import HomeAssistantError as HomeAssistantError, TemplateError as TemplateError
 from homeassistant.generated import currencies as currencies
@@ -140,6 +140,11 @@ PLATFORM_SCHEMA: Incomplete
 PLATFORM_SCHEMA_BASE: Incomplete
 ENTITY_SERVICE_FIELDS: Incomplete
 TARGET_SERVICE_FIELDS: Incomplete
+_HAS_ENTITY_SERVICE_FIELD: Incomplete
+
+def _make_entity_service_schema(schema: dict, extra: int) -> vol.Schema: ...
+
+BASE_ENTITY_SCHEMA: Incomplete
 
 def make_entity_service_schema(schema: dict, *, extra: int = ...) -> vol.Schema: ...
 
@@ -215,6 +220,8 @@ SCRIPT_ACTION_STOP: str
 SCRIPT_ACTION_VARIABLES: str
 SCRIPT_ACTION_WAIT_FOR_TRIGGER: str
 SCRIPT_ACTION_WAIT_TEMPLATE: str
+ACTIONS_MAP: Incomplete
+ACTIONS_SET: Incomplete
 
 def determine_script_action(action: dict[str, Any]) -> str: ...
 

homeassistant-stubs/helpers/data_entry_flow.pyi

@@ -3,18 +3,22 @@ from aiohttp import web as web
 from homeassistant import data_entry_flow as data_entry_flow
 from homeassistant.components.http import HomeAssistantView as HomeAssistantView
 from homeassistant.components.http.data_validator import RequestDataValidator as RequestDataValidator
-from typing import Any
+from typing import Any, Generic
+from typing_extensions import TypeVar
 
-class _BaseFlowManagerView(HomeAssistantView):
+_FlowManagerT = TypeVar('_FlowManagerT', bound='data_entry_flow.FlowManager[Any]', default=data_entry_flow.FlowManager)
+
+class _BaseFlowManagerView(HomeAssistantView, Generic[_FlowManagerT]):
     _flow_mgr: Incomplete
-    def __init__(self, flow_mgr: data_entry_flow.FlowManager) -> None: ...
+    def __init__(self, flow_mgr: _FlowManagerT) -> None: ...
     def _prepare_result_json(self, result: data_entry_flow.FlowResult) -> data_entry_flow.FlowResult: ...
 
-class FlowManagerIndexView(_BaseFlowManagerView):
+class FlowManagerIndexView(_BaseFlowManagerView[_FlowManagerT]):
     async def post(self, request: web.Request, data: dict[str, Any]) -> web.Response: ...
+    async def _post_impl(self, request: web.Request, data: dict[str, Any]) -> web.Response: ...
     def get_context(self, data: dict[str, Any]) -> dict[str, Any]: ...
 
-class FlowManagerResourceView(_BaseFlowManagerView):
+class FlowManagerResourceView(_BaseFlowManagerView[_FlowManagerT]):
     async def get(self, request: web.Request, flow_id: str) -> web.Response: ...
     async def post(self, request: web.Request, data: dict[str, Any], flow_id: str) -> web.Response: ...
     async def delete(self, request: web.Request, flow_id: str) -> web.Response: ...

homeassistant-stubs/helpers/debounce.pyi

@@ -15,9 +15,10 @@ class Debouncer(Generic[_R_co]):
     _timer_task: Incomplete
     _execute_at_end_of_timer: bool
     _execute_lock: Incomplete
+    _background: Incomplete
     _job: Incomplete
     _shutdown_requested: bool
-    def __init__(self, hass: HomeAssistant, logger: Logger, *, cooldown: float, immediate: bool, function: Callable[[], _R_co] | None = None) -> None: ...
+    def __init__(self, hass: HomeAssistant, logger: Logger, *, cooldown: float, immediate: bool, function: Callable[[], _R_co] | None = None, background: bool = False) -> None: ...
     @property
     def function(self) -> Callable[[], _R_co] | None: ...
     @function.setter

homeassistant-stubs/helpers/device_registry.pyi

@@ -2,7 +2,8 @@ from . import entity_registry as entity_registry, storage as storage, translatio
 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
+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 .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from collections import UserDict
@@ -24,7 +25,6 @@ EVENT_DEVICE_REGISTRY_UPDATED: str
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
 STORAGE_VERSION_MINOR: int
-SAVE_DELAY: int
 CLEANUP_DELAY: int
 CONNECTION_BLUETOOTH: str
 CONNECTION_NETWORK_MAC: str
@@ -85,6 +85,9 @@ class DeviceInfoError(HomeAssistantError):
     def __init__(self, domain: str, device_info: DeviceInfo, message: str) -> None: ...
 
 def _validate_device_info(config_entry: ConfigEntry, device_info: DeviceInfo) -> str: ...
+
+_cached_parse_url: Incomplete
+
 def _validate_configuration_url(value: Any) -> str | None: ...
 
 class DeviceEntry:
@@ -112,6 +115,7 @@ class DeviceEntry:
     @property
     def dict_repr(self) -> dict[str, Any]: ...
     def json_repr(self) -> bytes | None: ...
+    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): ...
     def __le__(self, other): ...
@@ -125,6 +129,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: ...
+    def as_storage_fragment(self) -> json_fragment: ...
     def __init__(self, config_entries, connections, identifiers, id, orphaned_timestamp) -> None: ...
     def __lt__(self, other): ...
     def __le__(self, other): ...
@@ -146,7 +151,7 @@ class DeviceRegistryItems(UserDict[str, _EntryTypeT]):
     def __delitem__(self, key: str) -> None: ...
     def get_entry(self, identifiers: set[tuple[str, str]] | None, connections: set[tuple[str, str]] | None) -> _EntryTypeT | None: ...
 
-class DeviceRegistry:
+class DeviceRegistry(BaseRegistry):
     devices: DeviceRegistryItems[DeviceEntry]
     deleted_devices: DeviceRegistryItems[DeletedDeviceEntry]
     _device_data: dict[str, DeviceEntry]
@@ -161,8 +166,7 @@ class DeviceRegistry:
     def async_update_device(self, device_id: str, *, add_config_entry_id: str | UndefinedType = ..., area_id: str | None | UndefinedType = ..., configuration_url: str | URL | None | UndefinedType = ..., disabled_by: DeviceEntryDisabler | None | UndefinedType = ..., entry_type: DeviceEntryType | None | UndefinedType = ..., hw_version: str | None | UndefinedType = ..., labels: set[str] | UndefinedType = ..., manufacturer: str | None | UndefinedType = ..., merge_connections: set[tuple[str, str]] | UndefinedType = ..., merge_identifiers: set[tuple[str, str]] | UndefinedType = ..., model: str | None | UndefinedType = ..., name_by_user: str | None | UndefinedType = ..., name: str | None | UndefinedType = ..., new_identifiers: set[tuple[str, str]] | UndefinedType = ..., remove_config_entry_id: str | UndefinedType = ..., serial_number: str | None | UndefinedType = ..., suggested_area: str | None | UndefinedType = ..., sw_version: str | None | UndefinedType = ..., via_device_id: str | None | UndefinedType = ...) -> DeviceEntry | None: ...
     def async_remove_device(self, device_id: str) -> None: ...
     async def async_load(self) -> None: ...
-    def async_schedule_save(self) -> None: ...
-    def _data_to_save(self) -> dict[str, list[dict[str, Any]]]: ...
+    def _data_to_save(self) -> dict[str, Any]: ...
     def async_clear_config_entry(self, config_entry_id: str) -> None: ...
     def async_purge_expired_orphaned_devices(self) -> None: ...
     def async_clear_area_id(self, area_id: str) -> None: ...

homeassistant-stubs/helpers/discovery.pyi

@@ -1,3 +1,4 @@
+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 .typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from collections.abc import Callable as Callable, Coroutine
@@ -6,7 +7,7 @@ from homeassistant.const import Platform as Platform
 from homeassistant.loader import bind_hass as bind_hass
 from typing import Any, TypedDict
 
-SIGNAL_PLATFORM_DISCOVERED: str
+SIGNAL_PLATFORM_DISCOVERED: SignalTypeFormat[DiscoveryDict]
 EVENT_LOAD_PLATFORM: str
 ATTR_PLATFORM: str
 ATTR_DISCOVERED: str

homeassistant-stubs/helpers/discovery_flow.pyi

@@ -1,8 +1,8 @@
 from _typeshed import Incomplete
 from collections.abc import Coroutine
+from homeassistant.config_entries import ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED
 from homeassistant.core import CoreState as CoreState, Event as Event, HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.async_ import gather_with_limited_concurrency as gather_with_limited_concurrency
 from typing import Any, NamedTuple
@@ -11,7 +11,7 @@ FLOW_INIT_LIMIT: int
 DISCOVERY_FLOW_DISPATCHER: str
 
 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, FlowResult] | None: ...
+def _async_init_flow(hass: HomeAssistant, domain: str, context: dict[str, Any], data: Any) -> Coroutine[None, None, ConfigFlowResult] | None: ...
 
 class PendingFlowKey(NamedTuple):
     domain: str

homeassistant-stubs/helpers/dispatcher.pyi

@@ -1,23 +1,15 @@
 from _typeshed import Incomplete
 from collections.abc import Callable, Coroutine
-from dataclasses import dataclass
 from homeassistant.core import HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback
 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 typing import Any, Generic, TypeVarTuple, overload
+from homeassistant.util.signal_type import SignalType as SignalType
+from typing import Any, TypeVarTuple, overload
 
 _Ts = TypeVarTuple('_Ts')
 _LOGGER: Incomplete
 DATA_DISPATCHER: str
-
-@dataclass(frozen=True)
-class SignalType(Generic[*_Ts]):
-    name: str
-    def __hash__(self) -> int: ...
-    def __eq__(self, other: Any) -> bool: ...
-    def __init__(self, name) -> None: ...
-
 _DispatcherDataType: Incomplete
 
 @overload