homeassistant-stubs 2024.7.4__py3-none-any.whl → 2024.8.0b0__py3-none-any.whl

homeassistant-stubs/helpers/intent.pyi

@@ -5,9 +5,9 @@ from . import area_registry as area_registry, device_registry as device_registry
 from .typing import VolSchemaType as VolSchemaType
 from _typeshed import Incomplete
 from abc import abstractmethod
-from collections.abc import Collection, Iterable
+from collections.abc import Callable, Collection, Iterable
 from dataclasses import dataclass
-from enum import Enum
+from enum import Enum, StrEnum
 from functools import cached_property as cached_property
 from homeassistant.components.homeassistant.exposed_entities import async_should_expose as async_should_expose
 from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_SUPPORTED_FEATURES as ATTR_SUPPORTED_FEATURES
@@ -19,7 +19,7 @@ from typing import Any
 
 _LOGGER: Incomplete
 _SlotsType = dict[str, Any]
-_IntentSlotsType: Incomplete
+_IntentSlotsType = dict[str | tuple[str, str], VolSchemaType | Callable[[Any], Any]]
 INTENT_TURN_OFF: str
 INTENT_TURN_ON: str
 INTENT_TOGGLE: str
@@ -33,6 +33,8 @@ INTENT_DECREASE_TIMER: str
 INTENT_PAUSE_TIMER: str
 INTENT_UNPAUSE_TIMER: str
 INTENT_TIMER_STATUS: str
+INTENT_GET_CURRENT_DATE: str
+INTENT_GET_CURRENT_TIME: str
 SLOT_SCHEMA: Incomplete
 DATA_KEY: HassKey[dict[str, IntentHandler]]
 SPEECH_TYPE_PLAIN: str
@@ -54,17 +56,17 @@ class IntentHandleError(IntentError):
 class IntentUnexpectedError(IntentError): ...
 
 class MatchFailedReason(Enum):
-    NAME: Incomplete
-    AREA: Incomplete
-    FLOOR: Incomplete
-    DOMAIN: Incomplete
-    DEVICE_CLASS: Incomplete
-    FEATURE: Incomplete
-    STATE: Incomplete
-    ASSISTANT: Incomplete
-    INVALID_AREA: Incomplete
-    INVALID_FLOOR: Incomplete
-    DUPLICATE_NAME: Incomplete
+    NAME = ...
+    AREA = ...
+    FLOOR = ...
+    DOMAIN = ...
+    DEVICE_CLASS = ...
+    FEATURE = ...
+    STATE = ...
+    ASSISTANT = ...
+    INVALID_AREA = ...
+    INVALID_FLOOR = ...
+    DUPLICATE_NAME = ...
     def is_no_entities_reason(self) -> bool: ...
 
 @dataclass
@@ -80,13 +82,13 @@ class MatchTargetsConstraints:
     allow_duplicate_names: bool = ...
     @property
     def has_constraints(self) -> bool: ...
-    def __init__(self, name, area_name, floor_name, domains, device_classes, features, states, assistant, allow_duplicate_names) -> None: ...
+    def __init__(self, name=..., area_name=..., floor_name=..., domains=..., device_classes=..., features=..., states=..., assistant=..., allow_duplicate_names=...) -> None: ...
 
 @dataclass
 class MatchTargetsPreferences:
     area_id: str | None = ...
     floor_id: str | None = ...
-    def __init__(self, area_id, floor_id) -> None: ...
+    def __init__(self, area_id=..., floor_id=...) -> None: ...
 
 @dataclass
 class MatchTargetsResult:
@@ -96,7 +98,7 @@ class MatchTargetsResult:
     no_match_name: str | None = ...
     areas: list[area_registry.AreaEntry] = ...
     floors: list[floor_registry.FloorEntry] = ...
-    def __init__(self, is_match, no_match_reason, states, no_match_name, areas, floors) -> None: ...
+    def __init__(self, is_match, no_match_reason=..., states=..., no_match_name=..., areas=..., floors=...) -> None: ...
 
 class MatchFailedError(IntentError):
     result: Incomplete
@@ -116,7 +118,7 @@ class MatchTargetsCandidate:
     floor: floor_registry.FloorEntry | None = ...
     device: device_registry.DeviceEntry | None = ...
     matched_name: str | None = ...
-    def __init__(self, state, entity, area, floor, device, matched_name) -> None: ...
+    def __init__(self, state, entity=..., area=..., floor=..., device=..., matched_name=...) -> None: ...
 
 def find_areas(name: str, areas: area_registry.AreaRegistry) -> Iterable[area_registry.AreaEntry]: ...
 def find_floors(name: str, floors: floor_registry.FloorRegistry) -> Iterable[floor_registry.FloorEntry]: ...
@@ -153,9 +155,10 @@ class DynamicServiceIntentHandler(IntentHandler, metaclass=abc.ABCMeta):
     required_states: Incomplete
     description: Incomplete
     platforms: Incomplete
+    device_classes: Incomplete
     required_slots: Incomplete
     optional_slots: Incomplete
-    def __init__(self, intent_type: str, speech: str | None = None, required_slots: _IntentSlotsType | None = None, optional_slots: _IntentSlotsType | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None) -> None: ...
+    def __init__(self, intent_type: str, speech: str | None = None, required_slots: _IntentSlotsType | None = None, optional_slots: _IntentSlotsType | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None, device_classes: set[type[StrEnum]] | None = None) -> None: ...
     @cached_property
     def slot_schema(self) -> dict: ...
     @abstractmethod
@@ -168,12 +171,12 @@ class DynamicServiceIntentHandler(IntentHandler, metaclass=abc.ABCMeta):
 class ServiceIntentHandler(DynamicServiceIntentHandler):
     domain: Incomplete
     service: Incomplete
-    def __init__(self, intent_type: str, domain: str, service: str, speech: str | None = None, required_slots: _IntentSlotsType | None = None, optional_slots: _IntentSlotsType | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None) -> None: ...
+    def __init__(self, intent_type: str, domain: str, service: str, speech: str | None = None, required_slots: _IntentSlotsType | None = None, optional_slots: _IntentSlotsType | None = None, required_domains: set[str] | None = None, required_features: int | None = None, required_states: set[str] | None = None, description: str | None = None, platforms: set[str] | None = None, device_classes: set[type[StrEnum]] | None = None) -> None: ...
     def get_domain_and_service(self, intent_obj: Intent, state: State) -> tuple[str, str]: ...
 
 class IntentCategory(Enum):
-    ACTION: str
-    QUERY: str
+    ACTION = 'action'
+    QUERY = 'query'
 
 class Intent:
     __slots__: Incomplete
@@ -192,32 +195,32 @@ class Intent:
     def create_response(self) -> IntentResponse: ...
 
 class IntentResponseType(Enum):
-    ACTION_DONE: str
-    PARTIAL_ACTION_DONE: str
-    QUERY_ANSWER: str
-    ERROR: str
+    ACTION_DONE = 'action_done'
+    PARTIAL_ACTION_DONE = 'partial_action_done'
+    QUERY_ANSWER = 'query_answer'
+    ERROR = 'error'
 
 class IntentResponseErrorCode(str, Enum):
-    NO_INTENT_MATCH: str
-    NO_VALID_TARGETS: str
-    FAILED_TO_HANDLE: str
-    UNKNOWN: str
+    NO_INTENT_MATCH = 'no_intent_match'
+    NO_VALID_TARGETS = 'no_valid_targets'
+    FAILED_TO_HANDLE = 'failed_to_handle'
+    UNKNOWN = 'unknown'
 
 class IntentResponseTargetType(str, Enum):
-    AREA: str
-    FLOOR: str
-    DEVICE: str
-    ENTITY: str
-    DOMAIN: str
-    DEVICE_CLASS: str
-    CUSTOM: str
+    AREA = 'area'
+    FLOOR = 'floor'
+    DEVICE = 'device'
+    ENTITY = 'entity'
+    DOMAIN = 'domain'
+    DEVICE_CLASS = 'device_class'
+    CUSTOM = 'custom'
 
 @dataclass(slots=True)
 class IntentResponseTarget:
     name: str
     type: IntentResponseTargetType
     id: str | None = ...
-    def __init__(self, name, type, id) -> None: ...
+    def __init__(self, name, type, id=...) -> None: ...
 
 class IntentResponse:
     language: Incomplete

homeassistant-stubs/helpers/issue_registry.pyi

@@ -23,9 +23,9 @@ class EventIssueRegistryUpdatedData(TypedDict):
     issue_id: str
 
 class IssueSeverity(StrEnum):
-    CRITICAL: str
-    ERROR: str
-    WARNING: str
+    CRITICAL = 'critical'
+    ERROR = 'error'
+    WARNING = 'warning'
 
 @dataclasses.dataclass(slots=True, frozen=True)
 class IssueEntry:

homeassistant-stubs/helpers/json.pyi

@@ -1,12 +1,18 @@
 import json
 import orjson
+from .deprecation import DeprecatedConstant as DeprecatedConstant, all_with_deprecated_constants as all_with_deprecated_constants, check_if_deprecated_constant as check_if_deprecated_constant, deprecated_function as deprecated_function, dir_with_deprecated_constants as dir_with_deprecated_constants
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
-from functools import partial as partial
 from homeassistant.util.file import write_utf8_file as write_utf8_file, write_utf8_file_atomic as write_utf8_file_atomic
-from homeassistant.util.json import JSON_DECODE_EXCEPTIONS as JSON_DECODE_EXCEPTIONS, JSON_ENCODE_EXCEPTIONS as JSON_ENCODE_EXCEPTIONS, SerializationError as SerializationError, format_unserializable_data as format_unserializable_data, json_loads as json_loads
+from homeassistant.util.json import SerializationError as SerializationError, format_unserializable_data as format_unserializable_data
 from typing import Any, Final
 
+_DEPRECATED_JSON_DECODE_EXCEPTIONS: Incomplete
+_DEPRECATED_JSON_ENCODE_EXCEPTIONS: Incomplete
+json_loads: Incomplete
+__getattr__: Incomplete
+__dir__: Incomplete
+__all__: Incomplete
 _LOGGER: Incomplete
 
 class JSONEncoder(json.JSONEncoder):

homeassistant-stubs/helpers/label_registry.pyi

@@ -8,14 +8,16 @@ from collections.abc import Iterable
 from dataclasses import dataclass
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util import slugify as slugify
+from homeassistant.util.dt import utc_from_timestamp as utc_from_timestamp, utcnow as utcnow
 from homeassistant.util.event_type import EventType as EventType
 from homeassistant.util.hass_dict import HassKey as HassKey
-from typing import Literal, TypedDict
+from typing import Any, Literal, TypedDict
 
 DATA_REGISTRY: HassKey[LabelRegistry]
 EVENT_LABEL_REGISTRY_UPDATED: EventType[EventLabelRegistryUpdatedData]
 STORAGE_KEY: str
 STORAGE_VERSION_MAJOR: int
+STORAGE_VERSION_MINOR: int
 
 class _LabelStoreData(TypedDict):
     color: str | None
@@ -23,6 +25,8 @@ class _LabelStoreData(TypedDict):
     icon: str | None
     label_id: str
     name: str
+    created_at: str
+    modified_at: str
 
 class LabelRegistryStoreData(TypedDict):
     labels: list[_LabelStoreData]
@@ -38,7 +42,10 @@ class LabelEntry(NormalizedNameBaseRegistryEntry):
     description: str | None = ...
     color: str | None = ...
     icon: str | None = ...
-    def __init__(self, *, name, normalized_name, label_id, description, color, icon) -> None: ...
+    def __init__(self, *, name, normalized_name, created_at=..., modified_at=..., label_id, description=..., color=..., icon=...) -> None: ...
+
+class LabelRegistryStore(Store[LabelRegistryStoreData]):
+    async def _async_migrate_func(self, old_major_version: int, old_minor_version: int, old_data: dict[str, list[dict[str, Any]]]) -> LabelRegistryStoreData: ...
 
 class LabelRegistry(BaseRegistry[LabelRegistryStoreData]):
     labels: NormalizedNameBaseRegistryItems[LabelEntry]

homeassistant-stubs/helpers/llm.pyi

@@ -64,7 +64,7 @@ class APIInstance:
     tools: list[Tool]
     custom_serializer: Callable[[Any], Any] | None = ...
     async def async_call_tool(self, tool_input: ToolInput) -> JsonObjectType: ...
-    def __init__(self, api, api_prompt, llm_context, tools, custom_serializer) -> None: ...
+    def __init__(self, api, api_prompt, llm_context, tools, custom_serializer=...) -> None: ...
 
 @dataclass(slots=True, kw_only=True)
 class API(ABC, metaclass=abc.ABCMeta):

homeassistant-stubs/helpers/normalized_name_base_registry.pyi

@@ -1,12 +1,15 @@
 from .registry import BaseRegistryItems as BaseRegistryItems
 from _typeshed import Incomplete
 from dataclasses import dataclass
+from datetime import datetime
 
 @dataclass(slots=True, frozen=True, kw_only=True)
 class NormalizedNameBaseRegistryEntry:
     name: str
     normalized_name: str
-    def __init__(self, *, name, normalized_name) -> None: ...
+    created_at: datetime = ...
+    modified_at: datetime = ...
+    def __init__(self, *, name, normalized_name, created_at=..., modified_at=...) -> None: ...
 
 def normalize_name(name: str) -> str: ...
 

homeassistant-stubs/helpers/recorder.pyi

@@ -1,17 +1,26 @@
 import asyncio
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable, Generator
 from dataclasses import dataclass
+from homeassistant.components.recorder import Recorder as Recorder
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.util.hass_dict import HassKey as HassKey
+from sqlalchemy.orm.session import Session as Session
 from typing import Any
 
+_LOGGER: Incomplete
 DOMAIN: HassKey[RecorderData]
+DATA_INSTANCE: HassKey[Recorder]
 
 @dataclass(slots=True)
 class RecorderData:
     recorder_platforms: dict[str, Any] = ...
     db_connected: asyncio.Future[bool] = ...
-    def __init__(self, recorder_platforms, db_connected) -> None: ...
+    def __init__(self, recorder_platforms=..., db_connected=...) -> None: ...
 
 def async_migration_in_progress(hass: HomeAssistant) -> bool: ...
+def async_migration_is_live(hass: HomeAssistant) -> bool: ...
 def async_initialize_recorder(hass: HomeAssistant) -> None: ...
 async def async_wait_recorder(hass: HomeAssistant) -> bool: ...
+def get_instance(hass: HomeAssistant) -> Recorder: ...
+def session_scope(*, hass: HomeAssistant | None = None, session: Session | None = None, exception_filter: Callable[[Exception], bool] | None = None, read_only: bool = False) -> Generator[Session]: ...

homeassistant-stubs/helpers/schema_config_entry_flow.pyi

@@ -22,7 +22,7 @@ class SchemaFlowFormStep(SchemaFlowStep):
     next_step: Callable[[dict[str, Any]], Coroutine[Any, Any, str | None]] | str | None = ...
     suggested_values: Callable[[SchemaCommonFlowHandler], Coroutine[Any, Any, dict[str, Any]]] | None | UndefinedType = ...
     preview: str | None = ...
-    def __init__(self, schema, validate_user_input, next_step, suggested_values, preview) -> None: ...
+    def __init__(self, schema=..., validate_user_input=..., next_step=..., suggested_values=..., preview=...) -> None: ...
 
 @dataclass(slots=True)
 class SchemaFlowMenuStep(SchemaFlowStep):

homeassistant-stubs/helpers/script.pyi

@@ -9,9 +9,9 @@ from .script_variables import ScriptVariables as ScriptVariables
 from .template import Template as Template
 from .trace import TraceElement as TraceElement, async_trace_path as async_trace_path, script_execution_set as script_execution_set, trace_append_element as trace_append_element, trace_id_get as trace_id_get, trace_path as trace_path, trace_path_get as trace_path_get, trace_path_stack_cv as trace_path_stack_cv, trace_set_result as trace_set_result, trace_stack_cv as trace_stack_cv, trace_stack_pop as trace_stack_pop, trace_stack_push as trace_stack_push, trace_stack_top as trace_stack_top, trace_update_result as trace_update_result
 from .trigger import async_initialize_triggers as async_initialize_triggers, async_validate_trigger_config as async_validate_trigger_config
-from .typing import ConfigType as ConfigType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
+from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Mapping, Sequence
+from collections.abc import AsyncGenerator, Callable as Callable, Mapping, Sequence
 from contextvars import ContextVar
 from dataclasses import dataclass
 from datetime import datetime, timedelta
@@ -20,14 +20,14 @@ from homeassistant import exceptions as exceptions
 from homeassistant.components import scene as scene
 from homeassistant.components.logger import LOGSEVERITY as LOGSEVERITY
 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_ALIAS as CONF_ALIAS, 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_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_EACH as CONF_FOR_EACH, CONF_IF as CONF_IF, CONF_MODE as CONF_MODE, CONF_PARALLEL as CONF_PARALLEL, CONF_REPEAT as CONF_REPEAT, CONF_RESPONSE_VARIABLE as CONF_RESPONSE_VARIABLE, 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_SET_CONVERSATION_RESPONSE as CONF_SET_CONVERSATION_RESPONSE, 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_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, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, SERVICE_TURN_ON as SERVICE_TURN_ON
-from homeassistant.core import Context as Context, Event as Event, HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse, callback as callback
+from homeassistant.core import Context as Context, Event as Event, HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, State as State, SupportsResponse as SupportsResponse, callback as callback
 from homeassistant.util import slugify as slugify
 from homeassistant.util.async_ import create_eager_task as create_eager_task
 from homeassistant.util.dt import utcnow as utcnow
 from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.signal_type import SignalType as SignalType, SignalTypeFormat as SignalTypeFormat
-from typing import Any, Literal, TypedDict
-from typing_extensions import AsyncGenerator
+from types import MappingProxyType
+from typing import Any, Literal, TypedDict, overload
 
 SCRIPT_MODE_PARALLEL: str
 SCRIPT_MODE_QUEUED: str
@@ -64,7 +64,7 @@ class ScriptData(TypedDict):
 class ScriptStoppedError(Exception): ...
 
 def _set_result_unless_done(future: asyncio.Future[None]) -> None: ...
-def action_trace_append(variables, path): ...
+def action_trace_append(variables: dict[str, Any], path: str) -> TraceElement: ...
 async def trace_action(hass: HomeAssistant, script_run: _ScriptRun, stop: asyncio.Future[None], variables: dict[str, Any]) -> AsyncGenerator[TraceElement]: ...
 def make_script_schema(schema: Mapping[Any, Any], default_script_mode: str, extra: int = ...) -> vol.Schema: ...
 
@@ -97,15 +97,15 @@ class _ScriptRun:
     _conversation_response: Incomplete
     def __init__(self, hass: HomeAssistant, script: Script, variables: dict[str, Any], context: Context | None, log_exceptions: bool) -> None: ...
     def _changed(self) -> None: ...
-    async def _async_get_condition(self, config): ...
+    async def _async_get_condition(self, config: ConfigType) -> ConditionCheckerType: ...
     def _log(self, msg: str, *args: Any, level: int = ..., **kwargs: Any) -> None: ...
-    def _step_log(self, default_message, timeout: Incomplete | None = None) -> None: ...
+    def _step_log(self, default_message: str, timeout: float | None = None) -> None: ...
     async def async_run(self) -> ScriptRunResult | None: ...
     async def _async_step(self, log_exceptions: bool) -> None: ...
     def _finish(self) -> None: ...
     async def async_stop(self) -> None: ...
     def _handle_exception(self, exception: Exception, continue_on_error: bool, log_exceptions: bool) -> None: ...
-    def _log_exception(self, exception) -> None: ...
+    def _log_exception(self, exception: Exception) -> None: ...
     def _get_pos_time_period_template(self, key: str) -> timedelta: ...
     async def _async_delay_step(self) -> None: ...
     def _get_timeout_seconds_from_action(self) -> float | None: ...
@@ -117,11 +117,14 @@ class _ScriptRun:
     async def _async_scene_step(self) -> None: ...
     async def _async_event_step(self) -> None: ...
     async def _async_condition_step(self) -> None: ...
-    def _test_conditions(self, conditions, name, condition_path: Incomplete | None = None): ...
+    def _test_conditions(self, conditions: list[ConditionCheckerType], name: str, condition_path: str | None = None) -> bool | None: ...
     async def _async_repeat_step(self) -> None: ...
     async def _async_choose_step(self) -> None: ...
     async def _async_if_step(self) -> None: ...
-    def _async_futures_with_timeout(self, timeout: float | None) -> tuple[list[asyncio.Future[None]], asyncio.TimerHandle | None, asyncio.Future[None] | None]: ...
+    @overload
+    def _async_futures_with_timeout(self, timeout: float) -> tuple[list[asyncio.Future[None]], asyncio.TimerHandle, asyncio.Future[None]]: ...
+    @overload
+    def _async_futures_with_timeout(self, timeout: None) -> tuple[list[asyncio.Future[None]], None, None]: ...
     async def _async_wait_for_trigger_step(self) -> None: ...
     def _async_handle_timeout(self) -> None: ...
     async def _async_wait_with_optional_timeout(self, futures: list[asyncio.Future[None]], timeout_handle: asyncio.TimerHandle | None, timeout_future: asyncio.Future[None] | None, unsub: Callable[[], None]) -> None: ...
@@ -140,8 +143,7 @@ class _QueuedScriptRun(_ScriptRun):
 def _schedule_stop_scripts_after_shutdown(hass: HomeAssistant) -> None: ...
 async def _async_stop_scripts_after_shutdown(hass: HomeAssistant, point_in_time: datetime) -> None: ...
 async def _async_stop_scripts_at_shutdown(hass: HomeAssistant, event: Event) -> None: ...
-
-_VarsType: Incomplete
+_VarsType = dict[str, Any] | MappingProxyType[str, Any]
 
 def _referenced_extract_ids(data: Any, key: str, found: set[str]) -> None: ...
 
@@ -158,7 +160,7 @@ class _IfData(TypedDict):
 class ScriptRunResult:
     conversation_response: str | None | UndefinedType
     service_response: ServiceResponse
-    variables: dict
+    variables: dict[str, Any]
     def __init__(self, conversation_response, service_response, variables) -> None: ...
 
 class Script:
@@ -222,9 +224,9 @@ class Script:
     def _find_referenced_entities(referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...
     def run(self, variables: _VarsType | None = None, context: Context | None = None) -> None: ...
     async def async_run(self, run_variables: _VarsType | None = None, context: Context | None = None, started_action: Callable[..., Any] | None = None) -> ScriptRunResult | None: ...
-    async def _async_stop(self, aws: list[asyncio.Task], update_state: bool) -> None: ...
+    async def _async_stop(self, aws: list[asyncio.Task[None]], update_state: bool) -> None: ...
     async def async_stop(self, update_state: bool = True, spare: _ScriptRun | None = None) -> None: ...
-    async def _async_get_condition(self, config): ...
+    async def _async_get_condition(self, config: ConfigType) -> ConditionCheckerType: ...
     def _prep_repeat_script(self, step: int) -> Script: ...
     def _get_repeat_script(self, step: int) -> Script: ...
     async def _async_prep_choose_data(self, step: int) -> _ChooseData: ...

homeassistant-stubs/helpers/selector.pyi

@@ -120,8 +120,8 @@ class ColorTempSelectorConfig(TypedDict, total=False):
     min_mireds: int
 
 class ColorTempSelectorUnit(StrEnum):
-    KELVIN: str
-    MIRED: str
+    KELVIN = 'kelvin'
+    MIRED = 'mired'
 
 class ColorTempSelector(Selector[ColorTempSelectorConfig]):
     selector_type: str
@@ -158,10 +158,10 @@ class ConstantSelector(Selector[ConstantSelectorConfig]):
     def __call__(self, data: Any) -> Any: ...
 
 class QrErrorCorrectionLevel(StrEnum):
-    LOW: str
-    MEDIUM: str
-    QUARTILE: str
-    HIGH: str
+    LOW = 'low'
+    MEDIUM = 'medium'
+    QUARTILE = 'quartile'
+    HIGH = 'high'
 
 class QrCodeSelectorConfig(TypedDict, total=False):
     data: str
@@ -222,6 +222,7 @@ class DeviceSelector(Selector[DeviceSelectorConfig]):
 
 class DurationSelectorConfig(TypedDict, total=False):
     enable_day: bool
+    enable_millisecond: bool
     allow_negative: bool
 
 class DurationSelector(Selector[DurationSelectorConfig]):
@@ -310,8 +311,8 @@ class NumberSelectorConfig(TypedDict, total=False):
     mode: NumberSelectorMode
 
 class NumberSelectorMode(StrEnum):
-    BOX: str
-    SLIDER: str
+    BOX = 'box'
+    SLIDER = 'slider'
 
 def validate_slider(data: Any) -> Any: ...
 
@@ -336,8 +337,8 @@ class SelectOptionDict(TypedDict):
     label: str
 
 class SelectSelectorMode(StrEnum):
-    LIST: str
-    DROPDOWN: str
+    LIST = 'list'
+    DROPDOWN = 'dropdown'
 
 class SelectSelectorConfig(TypedDict, total=False):
     options: Required[Sequence[SelectOptionDict] | Sequence[str]]
@@ -390,19 +391,19 @@ class TextSelectorConfig(TypedDict, total=False):
     multiple: bool
 
 class TextSelectorType(StrEnum):
-    COLOR: str
-    DATE: str
-    DATETIME_LOCAL: str
-    EMAIL: str
-    MONTH: str
-    NUMBER: str
-    PASSWORD: str
-    SEARCH: str
-    TEL: str
-    TEXT: str
-    TIME: str
-    URL: str
-    WEEK: str
+    COLOR = 'color'
+    DATE = 'date'
+    DATETIME_LOCAL = 'datetime-local'
+    EMAIL = 'email'
+    MONTH = 'month'
+    NUMBER = 'number'
+    PASSWORD = 'password'
+    SEARCH = 'search'
+    TEL = 'tel'
+    TEXT = 'text'
+    TIME = 'time'
+    URL = 'url'
+    WEEK = 'week'
 
 class TextSelector(Selector[TextSelectorConfig]):
     selector_type: str

homeassistant-stubs/helpers/service.pyi

@@ -7,7 +7,7 @@ from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsTy
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Coroutine, Iterable
 from homeassistant.auth.permissions.const import CAT_ENTITIES as CAT_ENTITIES, POLICY_CONTROL as POLICY_CONTROL
-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_ENTITY_ID as CONF_ENTITY_ID, 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_TARGET as CONF_TARGET, ENTITY_MATCH_ALL as ENTITY_MATCH_ALL, ENTITY_MATCH_NONE as ENTITY_MATCH_NONE
+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_ACTION as CONF_ACTION, CONF_ENTITY_ID as CONF_ENTITY_ID, 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
 from homeassistant.core import Context as Context, EntityServiceResponse as EntityServiceResponse, HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, TemplateError as TemplateError, Unauthorized as Unauthorized, UnknownUser as UnknownUser
 from homeassistant.loader import Integration as Integration, async_get_integrations as async_get_integrations, bind_hass as bind_hass
@@ -64,7 +64,7 @@ class SelectedEntities:
     referenced_devices: set[str] = ...
     referenced_areas: set[str] = ...
     def log_missing(self, missing_entities: set[str]) -> None: ...
-    def __init__(self, referenced, indirectly_referenced, missing_devices, missing_areas, missing_floors, missing_labels, referenced_devices, referenced_areas) -> None: ...
+    def __init__(self, referenced=..., indirectly_referenced=..., missing_devices=..., missing_areas=..., missing_floors=..., missing_labels=..., referenced_devices=..., referenced_areas=...) -> None: ...
 
 def call_from_config(hass: HomeAssistant, config: ConfigType, blocking: bool = False, variables: TemplateVarsType = None, validate_config: bool = True) -> None: ...
 async def async_call_from_config(hass: HomeAssistant, config: ConfigType, blocking: bool = False, variables: TemplateVarsType = None, validate_config: bool = True, context: Context | None = None) -> None: ...

homeassistant-stubs/helpers/service_info/mqtt.pyi

@@ -1,8 +1,7 @@
-from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.data_entry_flow import BaseServiceInfo as BaseServiceInfo
 
-ReceivePayloadType: Incomplete
+ReceivePayloadType = str | bytes
 
 @dataclass(slots=True)
 class MqttServiceInfo(BaseServiceInfo):

homeassistant-stubs/helpers/significant_change.pyi

@@ -4,12 +4,13 @@ from collections.abc import Callable, Mapping
 from homeassistant.const import STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, callback as callback
 from homeassistant.util.hass_dict import HassKey as HassKey
+from types import MappingProxyType
 from typing import Any, Protocol
 
 PLATFORM: str
 DATA_FUNCTIONS: HassKey[dict[str, CheckTypeFunc]]
-CheckTypeFunc: Incomplete
-ExtraCheckTypeFunc: Incomplete
+CheckTypeFunc = Callable[[HomeAssistant, str, dict | MappingProxyType, str, dict | MappingProxyType], bool | None]
+ExtraCheckTypeFunc = Callable[[HomeAssistant, str, dict | MappingProxyType, Any, str, dict | MappingProxyType, Any], bool | None]
 
 class SignificantChangeProtocol(Protocol):
     def async_check_significant_change(self, hass: HomeAssistant, old_state: str, old_attrs: Mapping[str, Any], new_state: str, new_attrs: Mapping[str, Any]) -> bool | None: ...

homeassistant-stubs/helpers/template.pyi

@@ -4,10 +4,11 @@ from .singleton import singleton as singleton
 from .translation import async_translate_state as async_translate_state
 from .typing import TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Iterable
+from collections.abc import Callable as Callable, Generator, Iterable
 from contextlib import AbstractContextManager
 from contextvars import ContextVar
 from datetime import datetime, timedelta
+from functools import cached_property as cached_property
 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, 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.exceptions import TemplateError as TemplateError
@@ -22,7 +23,6 @@ from jinja2.sandbox import ImmutableSandboxedEnvironment
 from lru import LRU
 from types import CodeType, TracebackType
 from typing import Any, Literal, NoReturn, Self, overload
-from typing_extensions import Generator
 
 _LOGGER: Incomplete
 _SENTINEL: Incomplete
@@ -176,7 +176,7 @@ class TemplateStateBase(State):
     def __init__(self, hass: HomeAssistant, collect: bool, entity_id: str) -> None: ...
     def _collect_state(self) -> None: ...
     def __getitem__(self, item: str) -> Any: ...
-    @property
+    @cached_property
     def entity_id(self) -> str: ...
     @property
     def state(self) -> str: ...

homeassistant-stubs/helpers/trace.pyi

@@ -1,11 +1,10 @@
 from .typing import TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
 from collections import deque
-from collections.abc import Callable as Callable, Coroutine
+from collections.abc import Callable as Callable, Coroutine, Generator
 from contextvars import ContextVar
 from homeassistant.core import ServiceResponse as ServiceResponse
 from typing import Any
-from typing_extensions import Generator
 
 class TraceElement:
     __slots__: Incomplete
@@ -20,7 +19,7 @@ class TraceElement:
     def __init__(self, variables: TemplateVarsType, path: str) -> None: ...
     def __repr__(self) -> str: ...
     def set_child_id(self, child_key: str, child_run_id: str) -> None: ...
-    def set_error(self, ex: Exception) -> None: ...
+    def set_error(self, ex: BaseException | None) -> None: ...
     def set_result(self, **kwargs: Any) -> None: ...
     def update_result(self, **kwargs: Any) -> None: ...
     _variables: Incomplete

homeassistant-stubs/helpers/trigger.pyi

@@ -40,7 +40,7 @@ class TriggerInfo(TypedDict):
 class PluggableActionsEntry:
     plugs: set[PluggableAction] = ...
     actions: dict[object, tuple[HassJob[[dict[str, Any], Context | None], Coroutine[Any, Any, None]], dict[str, Any]]] = ...
-    def __init__(self, plugs, actions) -> None: ...
+    def __init__(self, plugs=..., actions=...) -> None: ...
 
 class PluggableAction:
     _entry: PluggableActionsEntry | None

homeassistant-stubs/helpers/typing.pyi

@@ -8,15 +8,15 @@ GPSType = tuple[float, float]
 ConfigType = dict[str, Any]
 DiscoveryInfoType = dict[str, Any]
 ServiceDataType = dict[str, Any]
-StateType: Incomplete
-TemplateVarsType: Incomplete
+StateType = str | int | float | None
+TemplateVarsType = Mapping[str, Any] | None
 NoEventData = Mapping[str, Never]
 VolSchemaType: Incomplete
 VolDictType: Incomplete
 QueryType = Any
 
 class UndefinedType(Enum):
-    _singleton: int
+    _singleton = 0
 
 UNDEFINED: Incomplete