homeassistant-stubs 2024.10.3__py3-none-any.whl → 2024.11.0__py3-none-any.whl

homeassistant-stubs/helpers/label_registry.pyi

@@ -1,4 +1,4 @@
-from .normalized_name_base_registry import NormalizedNameBaseRegistryEntry as NormalizedNameBaseRegistryEntry, NormalizedNameBaseRegistryItems as NormalizedNameBaseRegistryItems, normalize_name as normalize_name
+from .normalized_name_base_registry import NormalizedNameBaseRegistryEntry as NormalizedNameBaseRegistryEntry, NormalizedNameBaseRegistryItems as NormalizedNameBaseRegistryItems
 from .registry import BaseRegistry as BaseRegistry
 from .singleton import singleton as singleton
 from .storage import Store as Store
@@ -7,7 +7,6 @@ from _typeshed import Incomplete
 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
@@ -34,6 +33,7 @@ class LabelRegistryStoreData(TypedDict):
 class EventLabelRegistryUpdatedData(TypedDict):
     action: Literal['create', 'remove', 'update']
     label_id: str
+type EventLabelRegistryUpdated = Event[EventLabelRegistryUpdatedData]
 
 @dataclass(slots=True, frozen=True, kw_only=True)
 class LabelEntry(NormalizedNameBaseRegistryEntry):
@@ -41,7 +41,7 @@ class LabelEntry(NormalizedNameBaseRegistryEntry):
     description: str | None = ...
     color: str | None = ...
     icon: str | None = ...
-    def __init__(self, *, name, normalized_name, created_at=..., modified_at=..., label_id, description=..., color=..., icon=...) -> None: ...
+    def __init__(self, *, 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: ...

homeassistant-stubs/helpers/llm.pyi

@@ -93,11 +93,10 @@ class AssistAPI(API):
 
 def _get_exposed_entities(hass: HomeAssistant, assistant: str) -> dict[str, dict[str, Any]]: ...
 def _selector_serializer(schema: Any) -> Any: ...
+def _get_cached_script_parameters(hass: HomeAssistant, entity_id: str) -> tuple[str | None, vol.Schema]: ...
 
 class ScriptTool(Tool):
     name: Incomplete
     _entity_id: Incomplete
-    parameters: Incomplete
-    description: Incomplete
     def __init__(self, hass: HomeAssistant, script_entity_id: str) -> None: ...
     async def async_call(self, hass: HomeAssistant, tool_input: ToolInput, llm_context: LLMContext) -> JsonObjectType: ...

homeassistant-stubs/helpers/network.pyi

@@ -1,3 +1,4 @@
+from .hassio import is_hassio as is_hassio
 from collections.abc import Callable as Callable
 from homeassistant.components import http as http
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/helpers/normalized_name_base_registry.pyi

@@ -2,20 +2,23 @@ from .registry import BaseRegistryItems as BaseRegistryItems
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from datetime import datetime
+from homeassistant.util import slugify as slugify
 
 @dataclass(slots=True, frozen=True, kw_only=True)
 class NormalizedNameBaseRegistryEntry:
     name: str
-    normalized_name: str
+    normalized_name: str = ...
     created_at: datetime = ...
     modified_at: datetime = ...
-    def __init__(self, *, name, normalized_name, created_at=..., modified_at=...) -> None: ...
+    def __post_init__(self) -> None: ...
+    def __init__(self, *, name, created_at=..., modified_at=...) -> None: ...
 
 def normalize_name(name: str) -> str: ...
 
-class NormalizedNameBaseRegistryItems(BaseRegistryItems[_VT]):
+class NormalizedNameBaseRegistryItems[_VT: NormalizedNameBaseRegistryEntry](BaseRegistryItems[_VT]):
     _normalized_names: Incomplete
     def __init__(self) -> None: ...
     def _unindex_entry(self, key: str, replacement_entry: _VT | None = None) -> None: ...
     def _index_entry(self, key: str, entry: _VT) -> None: ...
     def get_by_name(self, name: str) -> _VT | None: ...
+    def generate_id_from_name(self, name: str) -> str: ...

homeassistant-stubs/helpers/ratelimit.pyi

@@ -13,4 +13,4 @@ class KeyedRateLimit:
     def async_triggered(self, key: Hashable, now: float | None = None) -> None: ...
     def async_cancel_timer(self, key: Hashable) -> None: ...
     def async_remove(self) -> None: ...
-    def async_schedule_action(self, key: Hashable, rate_limit: float | None, now: float, action: Callable[[Unpack[_Ts]], None], *args: Unpack[_Ts]) -> float | None: ...
+    def async_schedule_action[*_Ts](self, key: Hashable, rate_limit: float | None, now: float, action: Callable[[Unpack[_Ts]], None], *args: Unpack[_Ts]) -> float | None: ...

homeassistant-stubs/helpers/redact.pyi

@@ -6,6 +6,6 @@ REDACTED: str
 
 def partial_redact(x: str | Any, unmasked_prefix: int = 4, unmasked_suffix: int = 4) -> str: ...
 @overload
-def async_redact_data(data: Mapping, to_redact: Iterable[Any] | Mapping[Any, Callable[[_ValueT], _ValueT]]) -> dict: ...
+def async_redact_data[_ValueT](data: Mapping, to_redact: Iterable[Any] | Mapping[Any, Callable[[_ValueT], _ValueT]]) -> dict: ...
 @overload
-def async_redact_data(data: _T, to_redact: Iterable[Any] | Mapping[Any, Callable[[_ValueT], _ValueT]]) -> _T: ...
+def async_redact_data[_T, _ValueT](data: _T, to_redact: Iterable[Any] | Mapping[Any, Callable[[_ValueT], _ValueT]]) -> _T: ...

homeassistant-stubs/helpers/registry.pyi

@@ -1,14 +1,16 @@
 import abc
 from .storage import Store as Store
 from abc import ABC, abstractmethod
-from collections import UserDict
+from collections import UserDict, defaultdict
 from collections.abc import Mapping as Mapping, Sequence as Sequence, ValuesView
 from homeassistant.core import CoreState as CoreState, HomeAssistant as HomeAssistant, callback as callback
+from typing import Any, Literal
 
 SAVE_DELAY: int
 SAVE_DELAY_LONG: int
+type RegistryIndexType = defaultdict[str, dict[str, Literal[True]]]
 
-class BaseRegistryItems(UserDict[str, _DataT], ABC, metaclass=abc.ABCMeta):
+class BaseRegistryItems[_DataT](UserDict[str, _DataT], ABC, metaclass=abc.ABCMeta):
     data: dict[str, _DataT]
     def values(self) -> ValuesView[_DataT]: ...
     @abstractmethod
@@ -19,7 +21,7 @@ class BaseRegistryItems(UserDict[str, _DataT], ABC, metaclass=abc.ABCMeta):
     def _unindex_entry_value(self, key: str, value: str, index: RegistryIndexType) -> None: ...
     def __delitem__(self, key: str) -> None: ...
 
-class BaseRegistry(ABC, metaclass=abc.ABCMeta):
+class BaseRegistry[_StoreDataT: Mapping[str, Any] | Sequence[Any]](ABC, metaclass=abc.ABCMeta):
     hass: HomeAssistant
     _store: Store[_StoreDataT]
     def async_schedule_save(self) -> None: ...

homeassistant-stubs/helpers/script.pyi

@@ -15,7 +15,6 @@ from collections.abc import AsyncGenerator, Callable as Callable, Mapping, Seque
 from contextvars import ContextVar
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-from functools import cached_property as cached_property
 from homeassistant import exceptions as exceptions
 from homeassistant.components import scene as scene
 from homeassistant.components.logger import LOGSEVERITY as LOGSEVERITY
@@ -26,6 +25,7 @@ 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 types import MappingProxyType
 from typing import Any, Literal, TypedDict, overload
 
 SCRIPT_MODE_PARALLEL: str
@@ -110,7 +110,7 @@ class _ScriptRun:
     def _get_timeout_seconds_from_action(self) -> float | None: ...
     async def _async_wait_template_step(self) -> None: ...
     def _async_set_remaining_time_var(self, timeout_handle: asyncio.TimerHandle | None) -> None: ...
-    async def _async_run_long_action(self, long_task: asyncio.Task[_T]) -> _T | None: ...
+    async def _async_run_long_action[_T](self, long_task: asyncio.Task[_T]) -> _T | None: ...
     async def _async_call_service_step(self) -> None: ...
     async def _async_device_step(self) -> None: ...
     async def _async_scene_step(self) -> None: ...
@@ -142,6 +142,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: ...
+type _VarsType = dict[str, Any] | Mapping[str, Any] | MappingProxyType[str, Any]
 def _referenced_extract_ids(data: Any, key: str, found: set[str]) -> None: ...
 
 class _ChooseData(TypedDict):
@@ -203,19 +204,14 @@ class Script:
     def runs(self) -> int: ...
     @property
     def supports_max(self) -> bool: ...
-    @cached_property
     def referenced_labels(self) -> set[str]: ...
-    @cached_property
     def referenced_floors(self) -> set[str]: ...
-    @cached_property
     def referenced_areas(self) -> set[str]: ...
     @staticmethod
     def _find_referenced_target(target: Literal['area_id', 'floor_id', 'label_id'], referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...
-    @cached_property
     def referenced_devices(self) -> set[str]: ...
     @staticmethod
     def _find_referenced_devices(referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...
-    @cached_property
     def referenced_entities(self) -> set[str]: ...
     @staticmethod
     def _find_referenced_entities(referenced: set[str], sequence: Sequence[dict[str, Any]]) -> None: ...

homeassistant-stubs/helpers/selector.pyi

@@ -14,7 +14,7 @@ def _get_selector_class(config: Any) -> type[Selector]: ...
 def selector(config: Any) -> Selector: ...
 def validate_selector(config: Any) -> dict: ...
 
-class Selector:
+class Selector[_T: Mapping[str, Any]]:
     CONFIG_SCHEMA: Callable
     config: _T
     selector_type: str

homeassistant-stubs/helpers/service.pyi

@@ -70,7 +70,7 @@ def call_from_config(hass: HomeAssistant, config: ConfigType, blocking: bool = F
 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: ...
 def async_prepare_call_from_config(hass: HomeAssistant, config: ConfigType, variables: TemplateVarsType = None, validate_config: bool = False) -> ServiceParams: ...
 def extract_entity_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
-async def async_extract_entities(hass: HomeAssistant, entities: Iterable[_EntityT], service_call: ServiceCall, expand_group: bool = True) -> list[_EntityT]: ...
+async def async_extract_entities[_EntityT: Entity](hass: HomeAssistant, entities: Iterable[_EntityT], service_call: ServiceCall, expand_group: bool = True) -> list[_EntityT]: ...
 async def async_extract_entity_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> set[str]: ...
 def _has_match(ids: str | list[str] | None) -> TypeGuard[str | list[str]]: ...
 def async_extract_referenced_entity_ids(hass: HomeAssistant, service_call: ServiceCall, expand_group: bool = True) -> SelectedEntities: ...
@@ -88,7 +88,7 @@ async def _async_admin_handler(hass: HomeAssistant, service_job: HassJob[[Servic
 def async_register_admin_service(hass: HomeAssistant, domain: str, service: str, service_func: Callable[[ServiceCall], Awaitable[None] | None], schema: VolSchemaType = ...) -> None: ...
 def verify_domain_control(hass: HomeAssistant, domain: str) -> Callable[[Callable[[ServiceCall], Any]], Callable[[ServiceCall], Any]]: ...
 
-class ReloadServiceHelper:
+class ReloadServiceHelper[_T]:
     _service_func: Incomplete
     _service_running: bool
     _service_condition: Incomplete

homeassistant-stubs/helpers/service_info/hassio.pyi

@@ -0,0 +1,11 @@
+from dataclasses import dataclass
+from homeassistant.data_entry_flow import BaseServiceInfo as BaseServiceInfo
+from typing import Any
+
+@dataclass(slots=True)
+class HassioServiceInfo(BaseServiceInfo):
+    config: dict[str, Any]
+    name: str
+    slug: str
+    uuid: str
+    def __init__(self, config, name, slug, uuid) -> None: ...

homeassistant-stubs/helpers/service_info/mqtt.pyi

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

homeassistant-stubs/helpers/significant_change.pyi

@@ -4,10 +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]]
+type CheckTypeFunc = Callable[[HomeAssistant, str, dict | MappingProxyType, str, dict | MappingProxyType], bool | None]
+type 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/singleton.pyi

@@ -4,7 +4,8 @@ from homeassistant.loader import bind_hass as bind_hass
 from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import overload
 
+type _FuncType[_T] = Callable[[HomeAssistant], _T]
 @overload
-def singleton(data_key: HassKey[_T]) -> Callable[[_FuncType[_T]], _FuncType[_T]]: ...
+def singleton[_T](data_key: HassKey[_T]) -> Callable[[_FuncType[_T]], _FuncType[_T]]: ...
 @overload
-def singleton(data_key: str) -> Callable[[_FuncType[_T]], _FuncType[_T]]: ...
+def singleton[_T](data_key: str) -> Callable[[_FuncType[_T]], _FuncType[_T]]: ...

homeassistant-stubs/helpers/storage.pyi

@@ -1,7 +1,6 @@
 import asyncio
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Iterable, Sequence as Sequence
-from functools import cached_property as cached_property
+from collections.abc import Callable as Callable, Iterable, Mapping, Sequence as Sequence
 from homeassistant.const import EVENT_HOMEASSISTANT_FINAL_WRITE as EVENT_HOMEASSISTANT_FINAL_WRITE, EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, CoreState as CoreState, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -10,6 +9,7 @@ from homeassistant.util import json as json_util
 from homeassistant.util.file import WriteError as WriteError
 from homeassistant.util.hass_dict import HassKey as HassKey
 from json import JSONEncoder
+from typing import Any
 
 MAX_LOAD_CONCURRENTLY: int
 STORAGE_DIR: str
@@ -18,7 +18,7 @@ STORAGE_SEMAPHORE: HassKey[asyncio.Semaphore]
 STORAGE_MANAGER: HassKey[_StoreManager]
 MANAGER_CLEANUP_DELAY: int
 
-async def async_migrator(hass: HomeAssistant, old_path: str, store: Store[_T], *, old_conf_load_func: Callable | None = None, old_conf_migrate_func: Callable | None = None) -> _T | None: ...
+async def async_migrator[_T: Mapping[str, Any] | Sequence[Any]](hass: HomeAssistant, old_path: str, store: Store[_T], *, old_conf_load_func: Callable | None = None, old_conf_migrate_func: Callable | None = None) -> _T | None: ...
 def get_internal_store_manager(hass: HomeAssistant) -> _StoreManager: ...
 
 class _StoreManager:
@@ -39,7 +39,7 @@ class _StoreManager:
     def _preload(self, keys: Iterable[str]) -> None: ...
     def _initialize_files(self) -> None: ...
 
-class Store:
+class Store[_T: Mapping[str, Any] | Sequence[Any]]:
     version: Incomplete
     minor_version: Incomplete
     key: Incomplete
@@ -56,7 +56,6 @@ class Store:
     _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: ...
-    @cached_property
     def path(self): ...
     def make_read_only(self) -> None: ...
     async def async_load(self) -> _T | None: ...

homeassistant-stubs/helpers/sun.pyi

@@ -2,6 +2,7 @@ import astral
 import astral.location
 import datetime
 from _typeshed import Incomplete
+from collections.abc import Callable
 from homeassistant.const import SUN_EVENT_SUNRISE as SUN_EVENT_SUNRISE, SUN_EVENT_SUNSET as SUN_EVENT_SUNSET
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import bind_hass as bind_hass
@@ -9,6 +10,7 @@ from homeassistant.util.hass_dict import HassKey as HassKey
 
 DATA_LOCATION_CACHE: HassKey[dict[tuple[str, str, str, float, float], astral.location.Location]]
 ELEVATION_AGNOSTIC_EVENTS: Incomplete
+type _AstralSunEventCallable = Callable[..., datetime.datetime]
 
 def get_astral_location(hass: HomeAssistant) -> tuple[astral.location.Location, astral.Elevation]: ...
 def get_astral_event_next(hass: HomeAssistant, event: str, utc_point_in_time: datetime.datetime | None = None, offset: datetime.timedelta | None = None) -> datetime.datetime: ...

homeassistant-stubs/helpers/system_info.pyi

@@ -1,3 +1,4 @@
+from .hassio import is_hassio as is_hassio
 from .importlib import async_import_module as async_import_module
 from .singleton import singleton as singleton
 from _typeshed import Incomplete

homeassistant-stubs/helpers/template.pyi

@@ -9,7 +9,6 @@ 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, 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.exceptions import TemplateError as TemplateError
@@ -170,16 +169,17 @@ class DomainStates:
     def __repr__(self) -> str: ...
 
 class TemplateStateBase(State):
+    __slots__: Incomplete
     _state: State
     __setitem__ = _readonly
     __delitem__ = _readonly
     _hass: Incomplete
     _collect: Incomplete
     _entity_id: Incomplete
+    _cache: Incomplete
     def __init__(self, hass: HomeAssistant, collect: bool, entity_id: str) -> None: ...
     def _collect_state(self) -> None: ...
     def __getitem__(self, item: str) -> Any: ...
-    @cached_property
     def entity_id(self) -> str: ...
     @property
     def state(self) -> str: ...
@@ -211,6 +211,7 @@ class TemplateState(TemplateStateBase):
     def __repr__(self) -> str: ...
 
 class TemplateStateFromEntityId(TemplateStateBase):
+    __slots__: Incomplete
     def __init__(self, hass: HomeAssistant, entity_id: str, collect: bool = True) -> None: ...
     @property
     def _state(self) -> State: ...
@@ -227,7 +228,7 @@ def _resolve_state(hass: HomeAssistant, entity_id_or_state: Any) -> State | Temp
 @overload
 def forgiving_boolean(value: Any) -> bool | object: ...
 @overload
-def forgiving_boolean(value: Any, default: _T) -> bool | _T: ...
+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]: ...

homeassistant-stubs/helpers/trace.pyi

@@ -35,9 +35,9 @@ script_execution_cv: ContextVar[StopReason | None]
 
 def trace_id_set(trace_id: tuple[str, str]) -> None: ...
 def trace_id_get() -> tuple[str, str] | None: ...
-def trace_stack_push(trace_stack_var: ContextVar[list[_T] | None], node: _T) -> None: ...
+def trace_stack_push[_T](trace_stack_var: ContextVar[list[_T] | None], node: _T) -> None: ...
 def trace_stack_pop(trace_stack_var: ContextVar[list[Any] | None]) -> None: ...
-def trace_stack_top(trace_stack_var: ContextVar[list[_T] | None]) -> _T | None: ...
+def trace_stack_top[_T](trace_stack_var: ContextVar[list[_T] | None]) -> _T | None: ...
 def trace_path_push(suffix: str | list[str]) -> int: ...
 def trace_path_pop(count: int) -> None: ...
 def trace_path_get() -> str: ...
@@ -55,4 +55,4 @@ class StopReason:
 def script_execution_set(reason: str, response: ServiceResponse = None) -> None: ...
 def script_execution_get() -> str | None: ...
 def trace_path(suffix: str | list[str]) -> Generator[None]: ...
-def async_trace_path(suffix: str | list[str]) -> Callable[[Callable[[Unpack[_Ts]], Coroutine[Any, Any, None]]], Callable[[Unpack[_Ts]], Coroutine[Any, Any, None]]]: ...
+def async_trace_path[*_Ts](suffix: str | list[str]) -> Callable[[Callable[[Unpack[_Ts]], Coroutine[Any, Any, None]]], Callable[[Unpack[_Ts]], Coroutine[Any, Any, None]]]: ...

homeassistant-stubs/helpers/typing.pyi

@@ -1,7 +1,20 @@
+import voluptuous as vol
 from .deprecation import DeferredDeprecatedAlias as DeferredDeprecatedAlias, 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 _typeshed import Incomplete
+from collections.abc import Mapping
 from enum import Enum
+from typing import Any, Never
 
+type GPSType = tuple[float, float]
+type ConfigType = dict[str, Any]
+type DiscoveryInfoType = dict[str, Any]
+type ServiceDataType = dict[str, Any]
+type StateType = str | int | float | None
+type TemplateVarsType = Mapping[str, Any] | None
+type NoEventData = Mapping[str, Never]
+type VolSchemaType = vol.Schema | vol.All | vol.Any
+type VolDictType = dict[str | vol.Marker, Any]
+type QueryType = Any
 class UndefinedType(Enum):
     _singleton = 0
 

homeassistant-stubs/helpers/update_coordinator.pyi

@@ -2,11 +2,12 @@ import abc
 import logging
 from . import entity as entity, event as event
 from .debounce import Debouncer as Debouncer
+from .frame import report as report
+from .typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from _typeshed import Incomplete
 from abc import abstractmethod
 from collections.abc import Awaitable, Callable as Callable, Coroutine, Generator
 from datetime import datetime, timedelta
-from functools import cached_property as cached_property
 from homeassistant import config_entries as config_entries
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
@@ -44,7 +45,7 @@ class DataUpdateCoordinator(BaseDataUpdateCoordinatorProtocol, Generic[_DataT]):
     last_update_success: bool
     last_exception: Incomplete
     _debounced_refresh: Incomplete
-    def __init__(self, hass: HomeAssistant, logger: logging.Logger, *, name: str, update_interval: timedelta | None = None, update_method: Callable[[], Awaitable[_DataT]] | None = None, setup_method: Callable[[], Awaitable[None]] | None = None, request_refresh_debouncer: Debouncer[Coroutine[Any, Any, None]] | None = None, always_update: bool = True) -> None: ...
+    def __init__(self, hass: HomeAssistant, logger: logging.Logger, *, config_entry: config_entries.ConfigEntry | None | UndefinedType = ..., name: str, update_interval: timedelta | None = None, update_method: Callable[[], Awaitable[_DataT]] | None = None, setup_method: Callable[[], Awaitable[None]] | None = None, request_refresh_debouncer: Debouncer[Coroutine[Any, Any, None]] | None = None, always_update: bool = True) -> None: ...
     async def async_register_shutdown(self) -> None: ...
     def async_add_listener(self, update_callback: CALLBACK_TYPE, context: Any = None) -> Callable[[], None]: ...
     def async_update_listeners(self) -> None: ...
@@ -76,11 +77,10 @@ class TimestampDataUpdateCoordinator(DataUpdateCoordinator[_DataT]):
     last_update_success_time: datetime | None
     def _async_refresh_finished(self) -> None: ...
 
-class BaseCoordinatorEntity(entity.Entity, metaclass=abc.ABCMeta):
+class BaseCoordinatorEntity[_BaseDataUpdateCoordinatorT: BaseDataUpdateCoordinatorProtocol](entity.Entity, metaclass=abc.ABCMeta):
     coordinator: Incomplete
     coordinator_context: Incomplete
     def __init__(self, coordinator: _BaseDataUpdateCoordinatorT, context: Any = None) -> None: ...
-    @cached_property
     def should_poll(self) -> bool: ...
     async def async_added_to_hass(self) -> None: ...
     def _handle_coordinator_update(self) -> None: ...

homeassistant-stubs/loader.pyi

@@ -22,7 +22,6 @@ from _typeshed import Incomplete
 from awesomeversion import AwesomeVersion
 from collections.abc import Callable as Callable, Iterable
 from dataclasses import dataclass
-from functools import cached_property as cached_property
 from types import ModuleType
 from typing import Any, Literal, Protocol, TypedDict
 
@@ -114,6 +113,7 @@ class Manifest(TypedDict, total=False):
     usb: list[dict[str, str]]
     homekit: dict[str, list[str]]
     is_built_in: bool
+    overwrites_built_in: bool
     version: str
     codeowners: list[str]
     loggers: list[str]
@@ -122,7 +122,7 @@ class Manifest(TypedDict, total=False):
 
 def async_setup(hass: HomeAssistant) -> None: ...
 def manifest_from_legacy_module(domain: str, module: ModuleType) -> Manifest: ...
-async def _async_get_custom_components(hass: HomeAssistant) -> dict[str, Integration]: ...
+def _get_custom_components(hass: HomeAssistant) -> dict[str, Integration]: ...
 async def async_get_custom_components(hass: HomeAssistant) -> dict[str, Integration]: ...
 async def async_get_config_flows(hass: HomeAssistant, type_filter: Literal['device', 'helper', 'hub', 'service'] | None = None) -> set[str]: ...
 
@@ -167,39 +167,22 @@ class Integration:
     _missing_platforms_cache: Incomplete
     _top_level_files: Incomplete
     def __init__(self, hass: HomeAssistant, pkg_path: str, file_path: pathlib.Path, manifest: Manifest, top_level_files: set[str] | None = None) -> None: ...
-    @cached_property
     def manifest_json_fragment(self) -> json_fragment: ...
-    @cached_property
     def name(self) -> str: ...
-    @cached_property
     def disabled(self) -> str | None: ...
-    @cached_property
     def domain(self) -> str: ...
-    @cached_property
     def dependencies(self) -> list[str]: ...
-    @cached_property
     def after_dependencies(self) -> list[str]: ...
-    @cached_property
     def requirements(self) -> list[str]: ...
-    @cached_property
     def config_flow(self) -> bool: ...
-    @cached_property
     def documentation(self) -> str | None: ...
-    @cached_property
     def issue_tracker(self) -> str | None: ...
-    @cached_property
     def loggers(self) -> list[str] | None: ...
-    @cached_property
     def quality_scale(self) -> str | None: ...
-    @cached_property
     def iot_class(self) -> str | None: ...
-    @cached_property
     def integration_type(self) -> Literal['entity', 'device', 'hardware', 'helper', 'hub', 'service', 'system', 'virtual']: ...
-    @cached_property
     def import_executor(self) -> bool: ...
-    @cached_property
     def has_translations(self) -> bool: ...
-    @cached_property
     def has_services(self) -> bool: ...
     @property
     def mqtt(self) -> list[str] | None: ...
@@ -218,8 +201,9 @@ class Integration:
     @property
     def is_built_in(self) -> bool: ...
     @property
+    def overwrites_built_in(self) -> bool: ...
+    @property
     def version(self) -> AwesomeVersion | None: ...
-    @cached_property
     def single_config_entry(self) -> bool: ...
     @property
     def all_dependencies(self) -> set[str]: ...
@@ -280,7 +264,7 @@ class Helpers:
     def __init__(self, hass: HomeAssistant) -> None: ...
     def __getattr__(self, helper_name: str) -> ModuleWrapper: ...
 
-def bind_hass(func: _CallableT) -> _CallableT: ...
+def bind_hass[_CallableT: Callable[..., Any]](func: _CallableT) -> _CallableT: ...
 async def _async_component_dependencies(hass: HomeAssistant, integration: Integration) -> set[str]: ...
 def _async_mount_config_dir(hass: HomeAssistant) -> None: ...
 def _lookup_path(hass: HomeAssistant) -> list[str]: ...

homeassistant-stubs/scripts/benchmark/__init__.pyi

@@ -9,7 +9,7 @@ BENCHMARKS: dict[str, Callable]
 
 def run(args) -> None: ...
 async def run_benchmark(bench) -> None: ...
-def benchmark(func: _CallableT) -> _CallableT: ...
+def benchmark[_CallableT: Callable](func: _CallableT) -> _CallableT: ...
 async def fire_events(hass): ...
 async def fire_events_with_filter(hass): ...
 async def state_changed_helper(hass): ...

homeassistant-stubs/util/__init__.pyi

@@ -11,7 +11,7 @@ def raise_if_invalid_filename(filename: str) -> None: ...
 def raise_if_invalid_path(path: str) -> None: ...
 def slugify(text: str | None, *, separator: str = '_') -> str: ...
 def repr_helper(inp: Any) -> str: ...
-def convert(value: _T | None, to_type: Callable[[_T], _U], default: _U | None = None) -> _U | None: ...
+def convert[_T, _U](value: _T | None, to_type: Callable[[_T], _U], default: _U | None = None) -> _U | None: ...
 def ensure_unique_string(preferred_string: str, current_strings: Iterable[str] | KeysView[str]) -> str: ...
 def get_random_string(length: int = 10) -> str: ...
 

homeassistant-stubs/util/async_.pyi

@@ -7,9 +7,9 @@ from typing import Any
 _LOGGER: Incomplete
 _SHUTDOWN_RUN_CALLBACK_THREADSAFE: str
 
-def create_eager_task(coro: Coroutine[Any, Any, _T], *, name: str | None = None, loop: AbstractEventLoop | None = None) -> Task[_T]: ...
+def create_eager_task[_T](coro: Coroutine[Any, Any, _T], *, name: str | None = None, loop: AbstractEventLoop | None = None) -> Task[_T]: ...
 def cancelling(task: Future[Any]) -> bool: ...
-def run_callback_threadsafe(loop: AbstractEventLoop, callback: Callable[[Unpack[_Ts]], _T], *args: Unpack[_Ts]) -> concurrent.futures.Future[_T]: ...
+def run_callback_threadsafe[_T, *_Ts](loop: AbstractEventLoop, callback: Callable[[Unpack[_Ts]], _T], *args: Unpack[_Ts]) -> concurrent.futures.Future[_T]: ...
 async def gather_with_limited_concurrency(limit: int, *tasks: Any, return_exceptions: bool = False) -> Any: ...
 def shutdown_run_callback_threadsafe(loop: AbstractEventLoop) -> None: ...
 def get_scheduled_timer_handles(loop: AbstractEventLoop) -> list[TimerHandle]: ...

homeassistant-stubs/util/decorator.pyi

@@ -1,4 +1,5 @@
 from collections.abc import Callable as Callable, Hashable as Hashable
+from typing import Any
 
-class Registry(dict[_KT, _VT]):
+class Registry[_KT: Hashable, _VT: Callable[..., Any]](dict[_KT, _VT]):
     def register(self, name: _KT) -> Callable[[_VT], _VT]: ...

homeassistant-stubs/util/dt.pyi

@@ -1,4 +1,5 @@
 import datetime as dt
+import zoneinfo
 from _typeshed import Incomplete
 from typing import Any, Literal, overload
 
@@ -13,8 +14,8 @@ POSTGRES_INTERVAL_RE: Incomplete
 
 def get_default_time_zone() -> dt.tzinfo: ...
 def set_default_time_zone(time_zone: dt.tzinfo) -> None: ...
-def get_time_zone(time_zone_str: str) -> dt.tzinfo | None: ...
-async def async_get_time_zone(time_zone_str: str) -> dt.tzinfo | None: ...
+def get_time_zone(time_zone_str: str) -> zoneinfo.ZoneInfo | None: ...
+async def async_get_time_zone(time_zone_str: str) -> zoneinfo.ZoneInfo | None: ...
 
 utcnow: Incomplete
 

homeassistant-stubs/util/enum.pyi

@@ -2,5 +2,5 @@ from collections.abc import Callable as Callable
 from enum import Enum as Enum
 from typing import Any
 
-def lru_cache(func: _T) -> _T: ...
-def try_parse_enum(cls, value: Any) -> _EnumT | None: ...
+def lru_cache[_T: Callable[..., Any]](func: _T) -> _T: ...
+def try_parse_enum[_EnumT: Enum](cls, value: Any) -> _EnumT | None: ...