homeassistant-stubs 2025.9.0b0__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/helpers/selector.pyi

@@ -1,3 +1,4 @@
+import voluptuous as vol
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Mapping, Sequence
 from enum import StrEnum
@@ -11,7 +12,7 @@ from typing import Any, Literal, Required, TypedDict
 
 SELECTORS: decorator.Registry[str, type[Selector]]
 
-def _get_selector_class(config: Any) -> type[Selector]: ...
+def _get_selector_type_and_class(config: Any) -> tuple[str, type[Selector]]: ...
 def selector(config: Any) -> Selector: ...
 def validate_selector(config: Any) -> dict: ...
 
@@ -19,6 +20,7 @@ class Selector[_T: Mapping[str, Any]]:
     CONFIG_SCHEMA: Callable
     config: _T
     selector_type: str
+    allowed_context_keys: dict[str, set[str]]
     def __init__(self, config: Mapping[str, Any] | None = None) -> None: ...
     def __eq__(self, other: object) -> bool: ...
     def serialize(self) -> dict[str, dict[str, _T]]: ...
@@ -27,14 +29,13 @@ class Selector[_T: Mapping[str, Any]]:
 def _entity_feature_flag(domain: str, enum_name: str, feature_name: str) -> int: ...
 def _validate_supported_feature(supported_feature: str) -> int: ...
 def _validate_supported_features(supported_features: list[str]) -> int: ...
-
-BASE_SELECTOR_CONFIG_SCHEMA: Incomplete
+def make_selector_config_schema(schema_dict: dict | None = None) -> vol.Schema: ...
 
 class BaseSelectorConfig(TypedDict, total=False):
     read_only: bool
 
 ENTITY_FILTER_SELECTOR_CONFIG_SCHEMA: Incomplete
-LEGACY_ENTITY_SELECTOR_CONFIG_SCHEMA: Incomplete
+_LEGACY_ENTITY_SELECTOR_CONFIG_SCHEMA_DICT: Incomplete
 
 class EntityFilterSelectorConfig(TypedDict, total=False):
     integration: str
@@ -43,7 +44,7 @@ class EntityFilterSelectorConfig(TypedDict, total=False):
     supported_features: list[str]
 
 DEVICE_FILTER_SELECTOR_CONFIG_SCHEMA: Incomplete
-LEGACY_DEVICE_SELECTOR_CONFIG_SCHEMA: Incomplete
+_LEGACY_DEVICE_SELECTOR_CONFIG_SCHEMA_DICT: Incomplete
 
 class DeviceFilterSelectorConfig(TypedDict, total=False):
     integration: str
@@ -55,7 +56,7 @@ class ActionSelectorConfig(BaseSelectorConfig): ...
 
 class ActionSelector(Selector[ActionSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: ActionSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
@@ -84,7 +85,7 @@ class AssistPipelineSelectorConfig(BaseSelectorConfig, total=False): ...
 
 class AssistPipelineSelector(Selector[AssistPipelineSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: AssistPipelineSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
@@ -94,6 +95,7 @@ class AttributeSelectorConfig(BaseSelectorConfig, total=False):
 
 class AttributeSelector(Selector[AttributeSelectorConfig]):
     selector_type: str
+    allowed_context_keys: Incomplete
     CONFIG_SCHEMA: Incomplete
     def __init__(self, config: AttributeSelectorConfig) -> None: ...
     def __call__(self, data: Any) -> str: ...
@@ -102,7 +104,7 @@ class BackupLocationSelectorConfig(BaseSelectorConfig, total=False): ...
 
 class BackupLocationSelector(Selector[BackupLocationSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: BackupLocationSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
@@ -110,15 +112,31 @@ class BooleanSelectorConfig(BaseSelectorConfig): ...
 
 class BooleanSelector(Selector[BooleanSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: BooleanSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> bool: ...
 
+def reject_nested_choose_selector(config: dict[str, Any]) -> dict[str, Any]: ...
+
+class ChooseSelectorChoiceConfig(TypedDict, total=False):
+    selector: Required[Selector | dict[str, Any]]
+
+class ChooseSelectorConfig(BaseSelectorConfig):
+    choices: Required[dict[str, ChooseSelectorChoiceConfig]]
+    translation_key: str
+
+class ChooseSelector(Selector[ChooseSelectorConfig]):
+    selector_type: str
+    CONFIG_SCHEMA: Incomplete
+    def __init__(self, config: ChooseSelectorConfig | None = None) -> None: ...
+    def serialize(self) -> dict[str, dict[str, ChooseSelectorConfig]]: ...
+    def __call__(self, data: Any) -> Any: ...
+
 class ColorRGBSelectorConfig(BaseSelectorConfig): ...
 
 class ColorRGBSelector(Selector[ColorRGBSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: ColorRGBSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> list[int]: ...
 
@@ -143,7 +161,7 @@ class ConditionSelectorConfig(BaseSelectorConfig): ...
 
 class ConditionSelector(Selector[ConditionSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: ConditionSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
@@ -190,7 +208,7 @@ class DateSelectorConfig(BaseSelectorConfig): ...
 
 class DateSelector(Selector[DateSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: DateSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
@@ -198,7 +216,7 @@ class DateTimeSelectorConfig(BaseSelectorConfig): ...
 
 class DateTimeSelector(Selector[DateTimeSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: DateTimeSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> Any: ...
 
@@ -299,13 +317,15 @@ class LocationSelector(Selector[LocationSelectorConfig]):
 
 class MediaSelectorConfig(BaseSelectorConfig, total=False):
     accept: list[str]
+    multiple: bool
 
 class MediaSelector(Selector[MediaSelectorConfig]):
     selector_type: str
+    allowed_context_keys: Incomplete
     CONFIG_SCHEMA: Incomplete
     DATA_SCHEMA: Incomplete
     def __init__(self, config: MediaSelectorConfig | None = None) -> None: ...
-    def __call__(self, data: Any) -> dict[str, str]: ...
+    def __call__(self, data: Any) -> dict[str, str] | list[dict[str, str]]: ...
 
 class NumberSelectorConfig(BaseSelectorConfig, total=False):
     min: float
@@ -327,22 +347,23 @@ class NumberSelector(Selector[NumberSelectorConfig]):
     def __init__(self, config: NumberSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> float: ...
 
-class ObjectSelectorField(TypedDict):
+class ObjectSelectorField(TypedDict, total=False):
     label: str
     required: bool
-    selector: dict[str, Any]
+    selector: Required[Selector | dict[str, Any]]
 
-class ObjectSelectorConfig(BaseSelectorConfig):
+class ObjectSelectorConfig(BaseSelectorConfig, total=False):
     fields: dict[str, ObjectSelectorField]
     multiple: bool
     label_field: str
-    description_field: bool
+    description_field: str
     translation_key: str
 
 class ObjectSelector(Selector[ObjectSelectorConfig]):
     selector_type: str
     CONFIG_SCHEMA: Incomplete
     def __init__(self, config: ObjectSelectorConfig | None = None) -> None: ...
+    def serialize(self) -> dict[str, dict[str, ObjectSelectorConfig]]: ...
     def __call__(self, data: Any) -> Any: ...
 
 class QrErrorCorrectionLevel(StrEnum):
@@ -393,6 +414,7 @@ class StateSelectorConfig(BaseSelectorConfig, total=False):
 
 class StateSelector(Selector[StateSelectorConfig]):
     selector_type: str
+    allowed_context_keys: Incomplete
     CONFIG_SCHEMA: Incomplete
     def __init__(self, config: StateSelectorConfig) -> None: ...
     def __call__(self, data: Any) -> str | list[str]: ...
@@ -421,7 +443,7 @@ class TemplateSelectorConfig(BaseSelectorConfig): ...
 
 class TemplateSelector(Selector[TemplateSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: TemplateSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
@@ -466,7 +488,7 @@ class TimeSelectorConfig(BaseSelectorConfig): ...
 
 class TimeSelector(Selector[TimeSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: TimeSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> str: ...
 
@@ -474,6 +496,6 @@ class TriggerSelectorConfig(BaseSelectorConfig): ...
 
 class TriggerSelector(Selector[TriggerSelectorConfig]):
     selector_type: str
-    CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA
+    CONFIG_SCHEMA: Incomplete
     def __init__(self, config: TriggerSelectorConfig | None = None) -> None: ...
     def __call__(self, data: Any) -> Any: ...

homeassistant-stubs/helpers/service.pyi

@@ -1,11 +1,11 @@
 import logging
-from . import device_registry as device_registry, entity_registry as entity_registry, selector as selector, target as target_helpers, template as template, translation as translation
-from .deprecation import deprecated_class as deprecated_class, deprecated_function as deprecated_function
+from . import device_registry as device_registry, entity_registry as entity_registry, selector as selector, target as target_helpers, template as template
+from .deprecation import deprecated_class as deprecated_class, deprecated_function as deprecated_function, deprecated_hass_argument as deprecated_hass_argument
 from .entity import Entity as Entity
 from .selector import TargetSelector as TargetSelector
 from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType, VolDictType as VolDictType, VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Coroutine, Iterable
+from collections.abc import Callable as Callable, Coroutine, Iterable, Mapping
 from functools import cache
 from homeassistant.auth.permissions.const import CAT_ENTITIES as CAT_ENTITIES, POLICY_CONTROL as POLICY_CONTROL
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ACTION as CONF_ACTION, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_SELECTOR as CONF_SELECTOR, CONF_SERVICE_DATA as CONF_SERVICE_DATA, CONF_SERVICE_DATA_TEMPLATE as CONF_SERVICE_DATA_TEMPLATE, CONF_SERVICE_TEMPLATE as CONF_SERVICE_TEMPLATE, CONF_TARGET as CONF_TARGET, ENTITY_MATCH_ALL as ENTITY_MATCH_ALL, ENTITY_MATCH_NONE as ENTITY_MATCH_NONE
@@ -44,7 +44,7 @@ class ServiceParams(TypedDict):
     service_data: dict[str, Any]
     target: dict | None
 
-class ServiceTargetSelector(target_helpers.TargetSelectorData):
+class ServiceTargetSelector(target_helpers.TargetSelection):
     def __init__(self, service_call: ServiceCall) -> None: ...
 
 class SelectedEntities(target_helpers.SelectedEntities):
@@ -58,18 +58,14 @@ async def async_call_from_config(hass: HomeAssistant, config: ConfigType, blocki
 @callback
 @bind_hass
 def async_prepare_call_from_config(hass: HomeAssistant, config: ConfigType, variables: TemplateVarsType = None, validate_config: bool = False) -> ServiceParams: ...
-@bind_hass
-def extract_entity_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
-@bind_hass
-async def async_extract_entities[_EntityT: Entity](hass: HomeAssistant, entities: Iterable[_EntityT], service_call: ServiceCall, expand_group: bool = True) -> list[_EntityT]: ...
-@bind_hass
-async def async_extract_entity_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
+def extract_entity_ids(service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
+async def async_extract_entities[_EntityT: Entity](entities: Iterable[_EntityT], service_call: ServiceCall, expand_group: bool = True) -> list[_EntityT]: ...
+async def async_extract_entity_ids(service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
 @bind_hass
 def async_extract_referenced_entity_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> SelectedEntities: ...
-@bind_hass
-async def async_extract_config_entry_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
-def _load_services_file(hass: HomeAssistant, integration: Integration) -> JSON_TYPE: ...
-def _load_services_files(hass: HomeAssistant, integrations: Iterable[Integration]) -> dict[str, JSON_TYPE]: ...
+async def async_extract_config_entry_ids(service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
+def _load_services_file(integration: Integration) -> JSON_TYPE: ...
+def _load_services_files(integrations: Iterable[Integration]) -> dict[str, JSON_TYPE]: ...
 @callback
 def async_get_cached_service_description(hass: HomeAssistant, domain: str, service: str) -> dict[str, Any] | None: ...
 @bind_hass
@@ -81,15 +77,14 @@ def remove_entity_service_fields(call: ServiceCall) -> dict[Any, Any]: ...
 def async_set_service_schema(hass: HomeAssistant, domain: str, service: str, schema: dict[str, Any]) -> None: ...
 def _get_permissible_entity_candidates(call: ServiceCall, entities: dict[str, Entity], entity_perms: Callable[[str, str], bool] | None, target_all_entities: bool, all_referenced: set[str] | None) -> list[Entity]: ...
 @bind_hass
-async def entity_service_call(hass: HomeAssistant, registered_entities: dict[str, Entity], func: str | HassJob, call: ServiceCall, required_features: Iterable[int] | None = None) -> EntityServiceResponse | None: ...
+async def entity_service_call(hass: HomeAssistant, registered_entities: dict[str, Entity] | Callable[[], dict[str, Entity]], func: str | HassJob, call: ServiceCall, required_features: Iterable[int] | None = None, *, entity_device_classes: Iterable[str | None] | None = None) -> EntityServiceResponse | None: ...
 async def _handle_entity_call(hass: HomeAssistant, entity: Entity, func: str | HassJob, data: dict | ServiceCall, context: Context) -> ServiceResponse: ...
 async def _async_admin_handler(hass: HomeAssistant, service_job: HassJob[[ServiceCall], Coroutine[Any, Any, ServiceResponse | EntityServiceResponse] | ServiceResponse | EntityServiceResponse | None], call: ServiceCall) -> ServiceResponse | EntityServiceResponse | None: ...
 @bind_hass
 @callback
-def async_register_admin_service(hass: HomeAssistant, domain: str, service: str, service_func: Callable[[ServiceCall], Coroutine[Any, Any, ServiceResponse | EntityServiceResponse] | ServiceResponse | EntityServiceResponse | None], schema: VolSchemaType = ..., supports_response: SupportsResponse = ...) -> None: ...
-@bind_hass
+def async_register_admin_service(hass: HomeAssistant, domain: str, service: str, service_func: Callable[[ServiceCall], Coroutine[Any, Any, ServiceResponse | EntityServiceResponse] | ServiceResponse | EntityServiceResponse | None], schema: VolSchemaType = ..., supports_response: SupportsResponse = ..., *, description_placeholders: Mapping[str, str] | None = None) -> None: ...
 @callback
-def verify_domain_control(hass: HomeAssistant, domain: str) -> Callable[[Callable[[ServiceCall], Any]], Callable[[ServiceCall], Any]]: ...
+def verify_domain_control(domain: str) -> Callable[[Callable[[ServiceCall], Any]], Callable[[ServiceCall], Any]]: ...
 
 class ReloadServiceHelper[_T]:
     _service_func: Incomplete
@@ -100,5 +95,8 @@ class ReloadServiceHelper[_T]:
     def __init__(self, service_func: Callable[[ServiceCall], Coroutine[Any, Any, Any]], reload_targets_func: Callable[[ServiceCall], set[_T]]) -> None: ...
     async def execute_service(self, service_call: ServiceCall) -> None: ...
 
+def _validate_entity_service_schema(schema: VolDictType | VolSchemaType | None, service: str) -> VolSchemaType: ...
+@callback
+def async_register_entity_service(hass: HomeAssistant, domain: str, name: str, *, description_placeholders: Mapping[str, str] | None = None, entity_device_classes: Iterable[str | None] | None = None, entities: dict[str, Entity], func: str | Callable[..., Any], job_type: HassJobType | None, required_features: Iterable[int] | None = None, schema: VolDictType | VolSchemaType | None, supports_response: SupportsResponse = ...) -> None: ...
 @callback
-def async_register_entity_service(hass: HomeAssistant, domain: str, name: str, *, entities: dict[str, Entity], func: str | Callable[..., Any], job_type: HassJobType | None, required_features: Iterable[int] | None = None, schema: VolDictType | VolSchemaType | None, supports_response: SupportsResponse = ...) -> None: ...
+def async_register_platform_entity_service(hass: HomeAssistant, service_domain: str, service_name: str, *, description_placeholders: Mapping[str, str] | None = None, entity_device_classes: Iterable[str | None] | None = None, entity_domain: str, func: str | Callable[..., Any], required_features: Iterable[int] | None = None, schema: VolDictType | VolSchemaType | None, supports_response: SupportsResponse = ...) -> None: ...

homeassistant-stubs/helpers/service_info/esphome.pyi

@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+from homeassistant.data_entry_flow import BaseServiceInfo as BaseServiceInfo
+
+@dataclass(slots=True)
+class ESPHomeServiceInfo(BaseServiceInfo):
+    name: str
+    zwave_home_id: int | None
+    ip_address: str
+    port: int
+    noise_psk: str | None = ...
+    @property
+    def socket_path(self) -> str: ...

homeassistant-stubs/helpers/storage.pyi

@@ -6,7 +6,7 @@ from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, CoreState as Core
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util import json as json_util
-from homeassistant.util.file import WriteError as WriteError
+from homeassistant.util.file import WriteError as WriteError, write_utf8_file as write_utf8_file, write_utf8_file_atomic as write_utf8_file_atomic
 from homeassistant.util.hass_dict import HassKey as HassKey
 from json import JSONEncoder
 from pathlib import Path
@@ -63,7 +63,8 @@ class Store[_T: Mapping[str, Any] | Sequence[Any]]:
     _read_only: Incomplete
     _next_write_time: float
     _manager: Incomplete
-    def __init__(self, hass: HomeAssistant, version: int, key: str, private: bool = False, *, atomic_writes: bool = False, encoder: type[JSONEncoder] | None = None, minor_version: int = 1, read_only: bool = False) -> None: ...
+    _serialize_in_event_loop: Incomplete
+    def __init__(self, hass: HomeAssistant, version: int, key: str, private: bool = False, *, atomic_writes: bool = False, encoder: type[JSONEncoder] | None = None, minor_version: int = 1, read_only: bool = False, serialize_in_event_loop: bool = True) -> None: ...
     @cached_property
     def path(self): ...
     def make_read_only(self) -> None: ...
@@ -86,7 +87,8 @@ class Store[_T: Mapping[str, Any] | Sequence[Any]]:
     async def _async_callback_delayed_write(self) -> None: ...
     async def _async_callback_final_write(self, _event: Event) -> None: ...
     async def _async_handle_write_data(self, *_args) -> None: ...
-    async def _async_write_data(self, path: str, data: dict) -> None: ...
-    def _write_data(self, path: str, data: dict) -> None: ...
+    async def _async_write_data(self, data: dict) -> None: ...
+    def _write_data(self, data: dict) -> None: ...
+    def _write_prepared_data(self, mode: str, json_data: str | bytes) -> None: ...
     async def _async_migrate_func(self, old_major_version, old_minor_version, old_data) -> None: ...
     async def async_remove(self) -> None: ...

homeassistant-stubs/helpers/target.pyi

@@ -1,5 +1,6 @@
 import dataclasses
 from . import group as group
+from .deprecation import deprecated_class as deprecated_class
 from .event import async_track_state_change_event as async_track_state_change_event
 from .typing import ConfigType as ConfigType
 from _typeshed import Incomplete
@@ -19,7 +20,7 @@ class TargetStateChangedData:
 
 def _has_match(ids: str | list[str] | None) -> TypeGuard[str | list[str]]: ...
 
-class TargetSelectorData:
+class TargetSelection:
     __slots__: Incomplete
     entity_ids: Incomplete
     device_ids: Incomplete
@@ -27,6 +28,10 @@ class TargetSelectorData:
     floor_ids: Incomplete
     label_ids: Incomplete
     def __init__(self, config: ConfigType) -> None: ...
+    @property
+    def has_any_target(self) -> bool: ...
+
+class TargetSelectorData(TargetSelection):
     @property
     def has_any_selector(self) -> bool: ...
 
@@ -42,16 +47,16 @@ class SelectedEntities:
     referenced_areas: set[str] = dataclasses.field(default_factory=set)
     def log_missing(self, missing_entities: set[str], logger: Logger) -> None: ...
 
-def async_extract_referenced_entity_ids(hass: HomeAssistant, selector_data: TargetSelectorData, expand_group: bool = True) -> SelectedEntities: ...
+def async_extract_referenced_entity_ids(hass: HomeAssistant, target_selection: TargetSelection, expand_group: bool = True) -> SelectedEntities: ...
 
 class TargetStateChangeTracker:
     _hass: Incomplete
-    _selector_data: Incomplete
+    _target_selection: Incomplete
     _action: Incomplete
     _entity_filter: Incomplete
     _state_change_unsub: CALLBACK_TYPE | None
     _registry_unsubs: list[CALLBACK_TYPE]
-    def __init__(self, hass: HomeAssistant, selector_data: TargetSelectorData, action: Callable[[TargetStateChangedData], Any], entity_filter: Callable[[set[str]], set[str]]) -> None: ...
+    def __init__(self, hass: HomeAssistant, target_selection: TargetSelection, action: Callable[[TargetStateChangedData], Any], entity_filter: Callable[[set[str]], set[str]]) -> None: ...
     def async_setup(self) -> Callable[[], None]: ...
     def _track_entities_state_change(self) -> None: ...
     def _setup_registry_listeners(self) -> None: ...

homeassistant-stubs/helpers/{template.pyi → template/__init__.pyi}

@@ -1,21 +1,18 @@
-import collections.abc
 import jinja2
 import weakref
-from . import area_registry as area_registry, device_registry as device_registry, entity_registry as entity_registry, issue_registry as issue_registry, label_registry as label_registry
-from .deprecation import deprecated_function as deprecated_function
-from .singleton import singleton as singleton
-from .translation import async_translate_state as async_translate_state
-from .typing import TemplateVarsType as TemplateVarsType
+from .context import render_with_context as render_with_context, template_context_manager as template_context_manager, template_cv as template_cv
+from .helpers import raise_no_default as raise_no_default
+from .render_info import RenderInfo as RenderInfo, render_info_cv as render_info_cv
 from _typeshed import Incomplete, OptExcInfo as OptExcInfo
-from collections.abc import Callable as Callable, Generator, Iterable, MutableSequence
-from contextlib import AbstractContextManager
-from contextvars import ContextVar
-from datetime import datetime, timedelta
+from collections.abc import Callable as Callable, Generator, Iterable
+from datetime import datetime
 from functools import cache
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_LATITUDE as ATTR_LATITUDE, ATTR_LONGITUDE as ATTR_LONGITUDE, ATTR_PERSONS as ATTR_PERSONS, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, EVENT_HOMEASSISTANT_START as EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN, UnitOfLength as UnitOfLength
-from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, State as State, callback as callback, split_entity_id as split_entity_id, valid_domain as valid_domain, valid_entity_id as valid_entity_id
+from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, State as State, callback as callback, valid_domain as valid_domain, valid_entity_id as valid_entity_id
 from homeassistant.exceptions import TemplateError as TemplateError
-from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.helpers.singleton import singleton as singleton
+from homeassistant.helpers.translation import async_translate_state as async_translate_state
+from homeassistant.helpers.typing import TemplateVarsType as TemplateVarsType
 from homeassistant.util import convert as convert
 from homeassistant.util.async_ import run_callback_threadsafe as run_callback_threadsafe
 from homeassistant.util.hass_dict import HassKey as HassKey
@@ -26,8 +23,8 @@ from jinja2 import pass_context
 from jinja2.sandbox import ImmutableSandboxedEnvironment
 from lru import LRU
 from propcache.api import under_cached_property
-from types import CodeType, TracebackType
-from typing import Any, Literal, NoReturn, Self, overload
+from types import CodeType
+from typing import Any, Literal, Self, overload
 
 _LOGGER: Incomplete
 _SENTINEL: Incomplete
@@ -39,10 +36,6 @@ _HASS_LOADER: str
 _IS_NUMERIC: Incomplete
 _RESERVED_NAMES: Incomplete
 _COLLECTABLE_STATE_ATTRIBUTES: Incomplete
-ALL_STATES_RATE_LIMIT: int
-DOMAIN_STATES_RATE_LIMIT: int
-_render_info: ContextVar[RenderInfo | None]
-template_cv: ContextVar[tuple[str, str] | None]
 CACHED_TEMPLATE_STATES: int
 EVAL_CACHE_SIZE: int
 MAX_CUSTOM_TEMPLATE_SIZE: Incomplete
@@ -55,9 +48,6 @@ ORJSON_PASSTHROUGH_OPTIONS: Incomplete
 def _template_state_no_collect(hass: HomeAssistant, state: State) -> TemplateState: ...
 def _template_state(hass: HomeAssistant, state: State) -> TemplateState: ...
 def async_setup(hass: HomeAssistant) -> bool: ...
-@bind_hass
-def attach(hass: HomeAssistant, obj: Any) -> None: ...
-def _attach(hass: HomeAssistant, obj: Any) -> None: ...
 def render_complex(value: Any, variables: TemplateVarsType = None, limited: bool = False, parse_result: bool = True) -> Any: ...
 def is_complex(value: Any) -> bool: ...
 def is_template_string(maybe_template: str) -> bool: ...
@@ -77,35 +67,8 @@ class TupleWrapper(tuple, ResultWrapper):
 _types: tuple[type[dict | list | set], ...]
 RESULT_WRAPPERS: dict[type, type]
 
-def _true(arg: str) -> bool: ...
-def _false(arg: str) -> bool: ...
 def _cached_parse_result(render_result: str) -> Any: ...
 
-class RenderInfo:
-    __slots__: Incomplete
-    template: Incomplete
-    filter_lifecycle: Callable[[str], bool]
-    filter: Callable[[str], bool]
-    _result: str | None
-    is_static: bool
-    exception: TemplateError | None
-    all_states: bool
-    all_states_lifecycle: bool
-    domains: collections.abc.Set[str]
-    domains_lifecycle: collections.abc.Set[str]
-    entities: collections.abc.Set[str]
-    rate_limit: float | None
-    has_time: bool
-    def __init__(self, template: Template) -> None: ...
-    def __repr__(self) -> str: ...
-    def _filter_domains_and_entities(self, entity_id: str) -> bool: ...
-    def _filter_entities(self, entity_id: str) -> bool: ...
-    def _filter_lifecycle_domains(self, entity_id: str) -> bool: ...
-    def result(self) -> str: ...
-    def _freeze_static(self) -> None: ...
-    def _freeze_sets(self) -> None: ...
-    def _freeze(self) -> None: ...
-
 class Template:
     __slots__: Incomplete
     template: str
@@ -242,35 +205,9 @@ def forgiving_boolean(value: Any) -> bool | object: ...
 def forgiving_boolean[_T](value: Any, default: _T) -> bool | _T: ...
 def result_as_boolean(template_result: Any | None) -> bool: ...
 def expand(hass: HomeAssistant, *args: Any) -> Iterable[State]: ...
-def device_entities(hass: HomeAssistant, _device_id: str) -> Iterable[str]: ...
 def integration_entities(hass: HomeAssistant, entry_name: str) -> Iterable[str]: ...
 def config_entry_id(hass: HomeAssistant, entity_id: str) -> str | None: ...
-def device_id(hass: HomeAssistant, entity_id_or_device_name: str) -> str | None: ...
-def device_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
-def device_attr(hass: HomeAssistant, device_or_entity_id: str, attr_name: str) -> Any: ...
 def config_entry_attr(hass: HomeAssistant, config_entry_id_: str, attr_name: str) -> Any: ...
-def is_device_attr(hass: HomeAssistant, device_or_entity_id: str, attr_name: str, attr_value: Any) -> bool: ...
-def issues(hass: HomeAssistant) -> dict[tuple[str, str], dict[str, Any]]: ...
-def issue(hass: HomeAssistant, domain: str, issue_id: str) -> dict[str, Any] | None: ...
-def floors(hass: HomeAssistant) -> Iterable[str | None]: ...
-def floor_id(hass: HomeAssistant, lookup_value: Any) -> str | None: ...
-def floor_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
-def floor_areas(hass: HomeAssistant, floor_id_or_name: str) -> Iterable[str]: ...
-def floor_entities(hass: HomeAssistant, floor_id_or_name: str) -> Iterable[str]: ...
-def areas(hass: HomeAssistant) -> Iterable[str | None]: ...
-def area_id(hass: HomeAssistant, lookup_value: str) -> str | None: ...
-def _get_area_name(area_reg: area_registry.AreaRegistry, valid_area_id: str) -> str: ...
-def area_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
-def area_entities(hass: HomeAssistant, area_id_or_name: str) -> Iterable[str]: ...
-def area_devices(hass: HomeAssistant, area_id_or_name: str) -> Iterable[str]: ...
-def labels(hass: HomeAssistant, lookup_value: Any = None) -> Iterable[str | None]: ...
-def label_id(hass: HomeAssistant, lookup_value: Any) -> str | None: ...
-def label_name(hass: HomeAssistant, lookup_value: str) -> str | None: ...
-def label_description(hass: HomeAssistant, lookup_value: str) -> str | None: ...
-def _label_id_or_name(hass: HomeAssistant, label_id_or_name: str) -> str | None: ...
-def label_areas(hass: HomeAssistant, label_id_or_name: str) -> Iterable[str]: ...
-def label_devices(hass: HomeAssistant, label_id_or_name: str) -> Iterable[str]: ...
-def label_entities(hass: HomeAssistant, label_id_or_name: str) -> Iterable[str]: ...
 def closest(hass: HomeAssistant, *args: Any) -> State | None: ...
 def closest_filter(hass: HomeAssistant, *args: Any) -> State | None: ...
 def distance(hass: HomeAssistant, *args: Any) -> float | None: ...
@@ -279,98 +216,31 @@ def is_state(hass: HomeAssistant, entity_id: str, state: str | list[str]) -> boo
 def is_state_attr(hass: HomeAssistant, entity_id: str, name: str, value: Any) -> bool: ...
 def state_attr(hass: HomeAssistant, entity_id: str, name: str) -> Any: ...
 def has_value(hass: HomeAssistant, entity_id: str) -> bool: ...
-def now(hass: HomeAssistant) -> datetime: ...
-def utcnow(hass: HomeAssistant) -> datetime: ...
-def raise_no_default(function: str, value: Any) -> NoReturn: ...
 def forgiving_round(value, precision: int = 0, method: str = 'common', default=...): ...
 def multiply(value, amount, default=...): ...
 def add(value, amount, default=...): ...
 def apply(value, fn, *args, **kwargs): ...
 def as_function(macro: jinja2.runtime.Macro) -> Callable[..., Any]: ...
-def logarithm(value, base=..., default=...): ...
-def sine(value, default=...): ...
-def cosine(value, default=...): ...
-def tangent(value, default=...): ...
-def arc_sine(value, default=...): ...
-def arc_cosine(value, default=...): ...
-def arc_tangent(value, default=...): ...
-def arc_tangent2(*args, default=...): ...
 def version(value): ...
-def square_root(value, default=...): ...
-def timestamp_custom(value, date_format=..., local: bool = True, default=...): ...
-def timestamp_local(value, default=...): ...
-def timestamp_utc(value, default=...): ...
-def forgiving_as_timestamp(value, default=...): ...
-def as_datetime(value: Any, default: Any = ...) -> Any: ...
-def as_timedelta(value: str) -> timedelta | None: ...
 def merge_response(value: ServiceResponse) -> list[Any]: ...
-def strptime(string, fmt, default=...): ...
 def fail_when_undefined(value): ...
-def min_max_from_filter(builtin_filter: Any, name: str) -> Any: ...
-def average(*args: Any, default: Any = ...) -> Any: ...
-def median(*args: Any, default: Any = ...) -> Any: ...
-def statistical_mode(*args: Any, default: Any = ...) -> Any: ...
 def forgiving_float(value, default=...): ...
 def forgiving_float_filter(value, default=...): ...
 def forgiving_int(value, default=..., base: int = 10): ...
 def forgiving_int_filter(value, default=..., base: int = 10): ...
 def is_number(value): ...
-def _is_list(value: Any) -> bool: ...
-def _is_set(value: Any) -> bool: ...
-def _is_tuple(value: Any) -> bool: ...
-def _to_set(value: Any) -> set[Any]: ...
-def _to_tuple(value): ...
-def _is_datetime(value: Any) -> bool: ...
 def _is_string_like(value: Any) -> bool: ...
-def regex_match(value, find: str = '', ignorecase: bool = False): ...
-
-_regex_cache: Incomplete
-
-def regex_replace(value: str = '', find: str = '', replace: str = '', ignorecase: bool = False): ...
-def regex_search(value, find: str = '', ignorecase: bool = False): ...
-def regex_findall_index(value, find: str = '', index: int = 0, ignorecase: bool = False): ...
-def regex_findall(value, find: str = '', ignorecase: bool = False): ...
-def bitwise_and(first_value, second_value): ...
-def bitwise_or(first_value, second_value): ...
-def bitwise_xor(first_value, second_value): ...
 def struct_pack(value: Any | None, format_string: str) -> bytes | None: ...
 def struct_unpack(value: bytes, format_string: str, offset: int = 0) -> Any | None: ...
 def from_hex(value: str) -> bytes: ...
-def base64_encode(value: str | bytes) -> str: ...
-def base64_decode(value: str, encoding: str | None = 'utf-8') -> str | bytes: ...
-def ordinal(value): ...
 def from_json(value, default=...): ...
 def _to_json_default(obj: Any) -> None: ...
 def to_json(value: Any, ensure_ascii: bool = False, pretty_print: bool = False, sort_keys: bool = False) -> str: ...
 @pass_context
 def random_every_time(context, values): ...
-def today_at(hass: HomeAssistant, time_str: str = '') -> datetime: ...
-def relative_time(hass: HomeAssistant, value: Any) -> Any: ...
-def time_since(hass: HomeAssistant, value: Any | datetime, precision: int = 1) -> Any: ...
-def time_until(hass: HomeAssistant, value: Any | datetime, precision: int = 1) -> Any: ...
-def urlencode(value): ...
-def slugify(value, separator: str = '_'): ...
 def iif(value: Any, if_true: Any = True, if_false: Any = False, if_none: Any = ...) -> Any: ...
-def shuffle(*args: Any, seed: Any = None) -> MutableSequence[Any]: ...
 def typeof(value: Any) -> Any: ...
-def flatten(value: Iterable[Any], levels: int | None = None) -> list[Any]: ...
-def intersect(value: Iterable[Any], other: Iterable[Any]) -> list[Any]: ...
-def difference(value: Iterable[Any], other: Iterable[Any]) -> list[Any]: ...
-def union(value: Iterable[Any], other: Iterable[Any]) -> list[Any]: ...
-def symmetric_difference(value: Iterable[Any], other: Iterable[Any]) -> list[Any]: ...
 def combine(*args: Any, recursive: bool = False) -> dict[Any, Any]: ...
-def md5(value: str) -> str: ...
-def sha1(value: str) -> str: ...
-def sha256(value: str) -> str: ...
-def sha512(value: str) -> str: ...
-
-class TemplateContextManager(AbstractContextManager):
-    def set_template(self, template_str: str, action: str) -> None: ...
-    def __exit__(self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None) -> None: ...
-
-_template_context_manager: Incomplete
-
-def _render_with_context(template_str: str, template: jinja2.Template, **kwargs: Any) -> str: ...
 def make_logging_undefined(strict: bool | None, log_fn: Callable[[int, str], None] | None) -> type[jinja2.Undefined]: ...
 async def async_load_custom_templates(hass: HomeAssistant) -> None: ...
 def _load_custom_templates(hass: HomeAssistant) -> dict[str, str]: ...
@@ -388,6 +258,7 @@ class HassLoader(jinja2.BaseLoader):
 
 class TemplateEnvironment(ImmutableSandboxedEnvironment):
     hass: Incomplete
+    limited: Incomplete
     template_cache: weakref.WeakValueDictionary[str | jinja2.nodes.Template, CodeType | None]
     loader: Incomplete
     def __init__(self, hass: HomeAssistant | None, limited: bool | None = False, strict: bool | None = False, log_fn: Callable[[int, str], None] | None = None) -> None: ...

homeassistant-stubs/helpers/template/context.pyi

@@ -0,0 +1,16 @@
+import jinja2
+from _typeshed import Incomplete
+from contextlib import AbstractContextManager
+from contextvars import ContextVar
+from types import TracebackType
+from typing import Any
+
+template_cv: ContextVar[tuple[str, str] | None]
+
+class TemplateContextManager(AbstractContextManager):
+    def set_template(self, template_str: str, action: str) -> None: ...
+    def __exit__(self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None) -> None: ...
+
+template_context_manager: Incomplete
+
+def render_with_context(template_str: str, template: jinja2.Template, **kwargs: Any) -> str: ...