homeassistant-stubs 2024.5.4__py3-none-any.whl → 2024.6.0__py3-none-any.whl

homeassistant-stubs/helpers/service.pyi

@@ -6,24 +6,23 @@ from .group import expand_entity_ids as expand_entity_ids
 from .selector import TargetSelector as TargetSelector
 from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
-from collections.abc import Awaitable, Callable as Callable, Iterable
+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.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
 from homeassistant.util.async_ import create_eager_task as create_eager_task
+from homeassistant.util.hass_dict import HassKey as HassKey
 from homeassistant.util.yaml import load_yaml_dict as load_yaml_dict
 from homeassistant.util.yaml.loader import JSON_TYPE as JSON_TYPE
 from types import ModuleType
-from typing import Any, TypeGuard, TypeVar, TypedDict
+from typing import Any, TypeGuard, TypedDict
 
-_EntityT = TypeVar('_EntityT', bound=Entity)
 CONF_SERVICE_ENTITY_ID: str
 _LOGGER: Incomplete
-SERVICE_DESCRIPTION_CACHE: str
-ALL_SERVICE_DESCRIPTIONS_CACHE: str
-_T = TypeVar('_T')
+SERVICE_DESCRIPTION_CACHE: HassKey[dict[tuple[str, str], dict[str, Any] | None]]
+ALL_SERVICE_DESCRIPTIONS_CACHE: HassKey[tuple[set[tuple[str, str]], dict[str, dict[str, Any]]]]
 
 def _base_components() -> dict[str, ModuleType]: ...
 def _validate_option_or_feature(option_or_feature: str, label: str) -> Any: ...
@@ -75,10 +74,11 @@ def async_extract_referenced_entity_ids(hass: HomeAssistant, service_call: Servi
 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]) -> list[JSON_TYPE]: ...
+def async_get_cached_service_description(hass: HomeAssistant, domain: str, service: str) -> dict[str, Any] | None: ...
 async def async_get_all_descriptions(hass: HomeAssistant) -> dict[str, dict[str, Any]]: ...
 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: None | Callable[[str, str], bool], target_all_entities: bool, all_referenced: set[str] | None) -> list[Entity]: ...
+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]: ...
 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 _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], Awaitable[None] | None], call: ServiceCall) -> None: ...
@@ -91,5 +91,5 @@ class ReloadServiceHelper:
     _service_condition: Incomplete
     _pending_reload_targets: Incomplete
     _reload_targets_func: Incomplete
-    def __init__(self, service_func: Callable[[ServiceCall], Awaitable], reload_targets_func: Callable[[ServiceCall], set[_T]]) -> None: ...
+    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: ...

homeassistant-stubs/helpers/service_info/mqtt.pyi

@@ -1,4 +1,3 @@
-import datetime as dt
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.data_entry_flow import BaseServiceInfo as BaseServiceInfo
@@ -12,5 +11,5 @@ class MqttServiceInfo(BaseServiceInfo):
     qos: int
     retain: bool
     subscribed_topic: str
-    timestamp: dt.datetime
+    timestamp: float
     def __init__(self, topic, payload, qos, retain, subscribed_topic, timestamp) -> None: ...

homeassistant-stubs/helpers/signal.pyi

@@ -1,8 +1,11 @@
+import asyncio
 from _typeshed import Incomplete
 from homeassistant.const import RESTART_EXIT_CODE as RESTART_EXIT_CODE
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.util.hass_dict import HassKey as HassKey
 
 _LOGGER: Incomplete
+KEY_HA_STOP: HassKey[asyncio.Task[None]]
 
 def async_register_signal_handling(hass: HomeAssistant) -> None: ...

homeassistant-stubs/helpers/significant_change.pyi

@@ -1,15 +1,19 @@
 from .integration_platform import async_process_integration_platforms as async_process_integration_platforms
 from _typeshed import Incomplete
-from collections.abc import Callable
+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 typing import Any
+from homeassistant.util.hass_dict import HassKey as HassKey
+from typing import Any, Protocol
 
 PLATFORM: str
-DATA_FUNCTIONS: str
+DATA_FUNCTIONS: HassKey[dict[str, CheckTypeFunc]]
 CheckTypeFunc: Incomplete
 ExtraCheckTypeFunc: Incomplete
 
+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: ...
+
 async def create_checker(hass: HomeAssistant, _domain: str, extra_significant_check: ExtraCheckTypeFunc | None = None) -> SignificantlyChangedChecker: ...
 async def _initialize(hass: HomeAssistant) -> None: ...
 def either_one_none(val1: Any | None, val2: Any | None) -> bool: ...

homeassistant-stubs/helpers/singleton.pyi

@@ -1,9 +1,12 @@
 from collections.abc import Callable
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.loader import bind_hass as bind_hass
-from typing import TypeVar
+from homeassistant.util.hass_dict import HassKey as HassKey
+from typing import overload
 
-_T = TypeVar('_T')
 _FuncType = Callable[[HomeAssistant], _T]
 
+@overload
+def singleton(data_key: HassKey[_T]) -> Callable[[_FuncType[_T]], _FuncType[_T]]: ...
+@overload
 def singleton(data_key: str) -> Callable[[_FuncType[_T]], _FuncType[_T]]: ...

homeassistant-stubs/helpers/storage.pyi

@@ -1,5 +1,6 @@
+import asyncio
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Iterable, Mapping, Sequence
+from collections.abc import Callable as Callable, Iterable, Sequence as Sequence
 from functools import cached_property as cached_property
 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
@@ -7,16 +8,15 @@ 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.hass_dict import HassKey as HassKey
 from json import JSONEncoder
-from typing import Any, Generic, TypeVar
 
 MAX_LOAD_CONCURRENTLY: int
 STORAGE_DIR: str
 _LOGGER: Incomplete
-STORAGE_SEMAPHORE: str
-STORAGE_MANAGER: str
+STORAGE_SEMAPHORE: HassKey[asyncio.Semaphore]
+STORAGE_MANAGER: HassKey[_StoreManager]
 MANAGER_CLEANUP_DELAY: int
-_T = TypeVar('_T', bound=Mapping[str, Any] | Sequence[Any])
 
 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: ...
 def get_internal_store_manager(hass: HomeAssistant) -> _StoreManager: ...
@@ -39,7 +39,7 @@ class _StoreManager:
     def _preload(self, keys: Iterable[str]) -> None: ...
     def _initialize_files(self) -> None: ...
 
-class Store(Generic[_T]):
+class Store:
     version: Incomplete
     minor_version: Incomplete
     key: Incomplete
@@ -49,7 +49,7 @@ class Store(Generic[_T]):
     _delay_handle: Incomplete
     _unsub_final_write_listener: Incomplete
     _write_lock: Incomplete
-    _load_task: Incomplete
+    _load_future: Incomplete
     _encoder: Incomplete
     _atomic_writes: Incomplete
     _read_only: Incomplete
@@ -58,6 +58,7 @@ class Store(Generic[_T]):
     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: ...
     async def _async_load(self) -> _T | None: ...
     async def _async_load_data(self): ...

homeassistant-stubs/helpers/sun.pyi

@@ -6,8 +6,9 @@ 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
+from homeassistant.util.hass_dict import HassKey as HassKey
 
-DATA_LOCATION_CACHE: str
+DATA_LOCATION_CACHE: HassKey[dict[tuple[str, str, str, float, float], astral.location.Location]]
 ELEVATION_AGNOSTIC_EVENTS: Incomplete
 _AstralSunEventCallable = Callable[..., datetime.datetime]
 

homeassistant-stubs/helpers/system_info.pyi

@@ -1,4 +1,5 @@
 from .importlib import async_import_module as async_import_module
+from .singleton import singleton as singleton
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.loader import bind_hass as bind_hass
@@ -6,8 +7,10 @@ from homeassistant.util.package import is_docker_env as is_docker_env, is_virtua
 from typing import Any
 
 _LOGGER: Incomplete
+_DATA_MAC_VER: str
 
 def is_official_image() -> bool: ...
+async def async_get_mac_ver(hass: HomeAssistant) -> str: ...
 
 cached_get_user: Incomplete
 

homeassistant-stubs/helpers/template.pyi

@@ -14,30 +14,25 @@ from homeassistant.exceptions import TemplateError as TemplateError
 from homeassistant.loader import bind_hass as bind_hass
 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
 from homeassistant.util.json import JSON_DECODE_EXCEPTIONS as JSON_DECODE_EXCEPTIONS, json_loads as json_loads
 from homeassistant.util.read_only_dict import ReadOnlyDict as ReadOnlyDict
 from homeassistant.util.thread import ThreadWithException as ThreadWithException
 from jinja2.sandbox import ImmutableSandboxedEnvironment
 from lru import LRU
 from types import CodeType, TracebackType
-from typing import Any, Literal, NoReturn, ParamSpec, Self, TypeVar, overload
+from typing import Any, Literal, NoReturn, Self, overload
 
 _LOGGER: Incomplete
 _SENTINEL: Incomplete
 DATE_STR_FORMAT: str
-_ENVIRONMENT: str
-_ENVIRONMENT_LIMITED: str
-_ENVIRONMENT_STRICT: str
+_ENVIRONMENT: HassKey[TemplateEnvironment]
+_ENVIRONMENT_LIMITED: HassKey[TemplateEnvironment]
+_ENVIRONMENT_STRICT: HassKey[TemplateEnvironment]
 _HASS_LOADER: str
-_RE_JINJA_DELIMITERS: Incomplete
 _IS_NUMERIC: Incomplete
 _RESERVED_NAMES: Incomplete
-_GROUP_DOMAIN_PREFIX: str
-_ZONE_DOMAIN_PREFIX: str
 _COLLECTABLE_STATE_ATTRIBUTES: Incomplete
-_T = TypeVar('_T')
-_R = TypeVar('_R')
-_P = ParamSpec('_P')
 ALL_STATES_RATE_LIMIT: int
 DOMAIN_STATES_RATE_LIMIT: int
 _render_info: ContextVar[RenderInfo | None]
@@ -75,8 +70,7 @@ RESULT_WRAPPERS: dict[type, type]
 
 def _true(arg: str) -> bool: ...
 def _false(arg: str) -> bool: ...
-
-_cached_literal_eval: Incomplete
+def _cached_parse_result(render_result: str) -> Any: ...
 
 class RenderInfo:
     __slots__: Incomplete
@@ -270,6 +264,7 @@ 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 logarithm(value, base=..., default=...): ...
 def sine(value, default=...): ...
 def cosine(value, default=...): ...

homeassistant-stubs/helpers/trace.pyi

@@ -4,10 +4,7 @@ from collections import deque
 from collections.abc import Callable as Callable, Coroutine, Generator
 from contextvars import ContextVar
 from homeassistant.core import ServiceResponse as ServiceResponse
-from typing import Any, TypeVar, TypeVarTuple
-
-_T = TypeVar('_T')
-_Ts = TypeVarTuple('_Ts')
+from typing import Any
 
 class TraceElement:
     __slots__: Incomplete

homeassistant-stubs/helpers/translation.pyi

@@ -1,6 +1,8 @@
 import pathlib
+from . import singleton as singleton
 from _typeshed import Incomplete
 from collections.abc import Iterable
+from dataclasses import dataclass
 from homeassistant.const import EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, async_get_hass as async_get_hass, callback as callback
 from homeassistant.loader import Integration as Integration, async_get_config_flows as async_get_config_flows, async_get_integrations as async_get_integrations, bind_hass as bind_hass
@@ -12,16 +14,20 @@ TRANSLATION_FLATTEN_CACHE: str
 LOCALE_EN: str
 
 def recursive_flatten(prefix: str, data: dict[str, dict[str, Any] | str]) -> dict[str, str]: ...
-def component_translation_path(language: str, integration: Integration) -> pathlib.Path: ...
 def _load_translations_files_by_language(translation_files: dict[str, dict[str, pathlib.Path]]) -> dict[str, dict[str, Any]]: ...
 def build_resources(translation_strings: dict[str, dict[str, dict[str, Any] | str]], components: set[str], category: str) -> dict[str, dict[str, Any] | str]: ...
 async def _async_get_component_strings(hass: HomeAssistant, languages: Iterable[str], components: set[str], integrations: dict[str, Integration]) -> dict[str, dict[str, Any]]: ...
 
+@dataclass(slots=True)
+class _TranslationsCacheData:
+    loaded: dict[str, set[str]]
+    cache: dict[str, dict[str, dict[str, dict[str, str]]]]
+    def __init__(self, loaded, cache) -> None: ...
+
 class _TranslationCache:
     __slots__: Incomplete
     hass: Incomplete
-    loaded: Incomplete
-    cache: Incomplete
+    cache_data: Incomplete
     lock: Incomplete
     def __init__(self, hass: HomeAssistant) -> None: ...
     def async_is_loaded(self, language: str, components: set[str]) -> bool: ...

homeassistant-stubs/helpers/trigger.pyi

@@ -1,17 +1,20 @@
 import voluptuous as vol
+from .template import Template as Template
 from .typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType
 from _typeshed import Incomplete
+from collections import defaultdict
 from collections.abc import Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.const import CONF_ALIAS as CONF_ALIAS, CONF_ENABLED as CONF_ENABLED, CONF_ID as CONF_ID, CONF_PLATFORM as CONF_PLATFORM, CONF_VARIABLES as CONF_VARIABLES
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Context as Context, HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback, is_callback as is_callback
-from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, TemplateError as TemplateError
 from homeassistant.loader import IntegrationNotFound as IntegrationNotFound, async_get_integration as async_get_integration
 from homeassistant.util.async_ import create_eager_task as create_eager_task
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any, Protocol, TypedDict
 
 _PLATFORM_ALIASES: Incomplete
-DATA_PLUGGABLE_ACTIONS: str
+DATA_PLUGGABLE_ACTIONS: HassKey[defaultdict[tuple, PluggableActionsEntry]]
 
 class TriggerProtocol(Protocol):
     TRIGGER_SCHEMA: vol.Schema

homeassistant-stubs/helpers/update_coordinator.pyi

@@ -17,7 +17,6 @@ from typing_extensions import TypeVar
 REQUEST_REFRESH_DEFAULT_COOLDOWN: int
 REQUEST_REFRESH_DEFAULT_IMMEDIATE: bool
 _DataT = TypeVar('_DataT', default=dict[str, Any])
-_BaseDataUpdateCoordinatorT = TypeVar('_BaseDataUpdateCoordinatorT', bound='BaseDataUpdateCoordinatorProtocol')
 _DataUpdateCoordinatorT = TypeVar('_DataUpdateCoordinatorT', bound='DataUpdateCoordinator[Any]', default='DataUpdateCoordinator[dict[str, Any]]')
 
 class UpdateFailed(Exception): ...
@@ -73,7 +72,7 @@ class TimestampDataUpdateCoordinator(DataUpdateCoordinator[_DataT]):
     last_update_success_time: datetime | None
     def _async_refresh_finished(self) -> None: ...
 
-class BaseCoordinatorEntity(entity.Entity, Generic[_BaseDataUpdateCoordinatorT], metaclass=abc.ABCMeta):
+class BaseCoordinatorEntity(entity.Entity, metaclass=abc.ABCMeta):
     coordinator: Incomplete
     coordinator_context: Incomplete
     def __init__(self, coordinator: _BaseDataUpdateCoordinatorT, context: Any = None) -> None: ...

homeassistant-stubs/loader.pyi

@@ -1,3 +1,4 @@
+import asyncio
 import pathlib
 import voluptuous as vol
 from . import generated as generated
@@ -13,17 +14,18 @@ from .generated.ssdp import SSDP as SSDP
 from .generated.usb import USB as USB
 from .generated.zeroconf import HOMEKIT as HOMEKIT, ZEROCONF as ZEROCONF
 from .helpers import device_registry as dr
+from .helpers.json import json_bytes as json_bytes, json_fragment as json_fragment
 from .helpers.typing import ConfigType as ConfigType
+from .util.hass_dict import HassKey as HassKey
 from .util.json import JSON_DECODE_EXCEPTIONS as JSON_DECODE_EXCEPTIONS, json_loads as json_loads
 from _typeshed import Incomplete
 from awesomeversion import AwesomeVersion
-from collections.abc import Callable, Iterable
+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, TypeVar, TypedDict
+from typing import Any, Literal, Protocol, TypedDict
 
-_CallableT = TypeVar('_CallableT', bound=Callable[..., Any])
 _LOGGER: Incomplete
 BASE_PRELOAD_PLATFORMS: Incomplete
 
@@ -34,11 +36,11 @@ class BlockedIntegration:
     def __init__(self, lowest_good_version, reason) -> None: ...
 
 BLOCKED_CUSTOM_INTEGRATIONS: dict[str, BlockedIntegration]
-DATA_COMPONENTS: str
-DATA_INTEGRATIONS: str
-DATA_MISSING_PLATFORMS: str
-DATA_CUSTOM_COMPONENTS: str
-DATA_PRELOAD_PLATFORMS: str
+DATA_COMPONENTS: HassKey[dict[str, ModuleType | ComponentProtocol]]
+DATA_INTEGRATIONS: HassKey[dict[str, Integration | asyncio.Future[None]]]
+DATA_MISSING_PLATFORMS: HassKey[dict[str, bool]]
+DATA_CUSTOM_COMPONENTS: HassKey[dict[str, Integration] | asyncio.Future[dict[str, Integration]]]
+DATA_PRELOAD_PLATFORMS: HassKey[list[str]]
 PACKAGE_CUSTOM_COMPONENTS: str
 PACKAGE_BUILTIN: str
 CUSTOM_WARNING: str
@@ -167,6 +169,8 @@ class Integration:
     _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: ...

homeassistant-stubs/requirements.pyi

@@ -1,6 +1,7 @@
 import asyncio
 from .core import HomeAssistant as HomeAssistant, callback as callback
 from .exceptions import HomeAssistantError as HomeAssistantError
+from .helpers import singleton as singleton
 from .helpers.typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from .loader import Integration as Integration, IntegrationNotFound as IntegrationNotFound, async_get_integration as async_get_integration
 from _typeshed import Incomplete

homeassistant-stubs/scripts/benchmark/__init__.pyi

@@ -1,12 +1,10 @@
-from collections.abc import Callable
+from collections.abc import Callable as Callable
 from homeassistant import core as core
 from homeassistant.const import EVENT_STATE_CHANGED as EVENT_STATE_CHANGED
 from homeassistant.helpers.entityfilter import convert_include_exclude_filter as convert_include_exclude_filter
 from homeassistant.helpers.event import async_track_state_change as async_track_state_change, async_track_state_change_event as async_track_state_change_event
 from homeassistant.helpers.json import JSONEncoder as JSONEncoder, JSON_DUMP as JSON_DUMP
-from typing import TypeVar
 
-_CallableT = TypeVar('_CallableT', bound=Callable)
 BENCHMARKS: dict[str, Callable]
 
 def run(args) -> None: ...

homeassistant-stubs/setup.pyi

@@ -1,15 +1,17 @@
+import asyncio
 import contextvars
 from . import core as core, loader as loader, requirements as requirements
-from .const import EVENT_COMPONENT_LOADED as EVENT_COMPONENT_LOADED, EVENT_HOMEASSISTANT_START as EVENT_HOMEASSISTANT_START, PLATFORM_FORMAT as PLATFORM_FORMAT, Platform as Platform
+from .const import BASE_PLATFORMS as BASE_PLATFORMS, EVENT_COMPONENT_LOADED as EVENT_COMPONENT_LOADED, EVENT_HOMEASSISTANT_START as EVENT_HOMEASSISTANT_START, PLATFORM_FORMAT as PLATFORM_FORMAT
 from .core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from .exceptions import DependencyError as DependencyError, HomeAssistantError as HomeAssistantError
-from .helpers import translation as translation
+from .helpers import singleton as singleton, translation as translation
 from .helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
 from .helpers.typing import ConfigType as ConfigType
 from .util.async_ import create_eager_task as create_eager_task
+from .util.hass_dict import HassKey as HassKey
 from _typeshed import Incomplete
 from collections import defaultdict
-from collections.abc import Awaitable, Callable as Callable, Generator
+from collections.abc import Awaitable, Callable as Callable, Generator, Mapping
 from enum import StrEnum
 from types import ModuleType
 from typing import Final, TypedDict
@@ -17,13 +19,12 @@ from typing import Final, TypedDict
 current_setup_group: contextvars.ContextVar[tuple[str, str | None] | None]
 _LOGGER: Incomplete
 ATTR_COMPONENT: Final[str]
-BASE_PLATFORMS: Incomplete
-DATA_SETUP: str
-DATA_SETUP_DONE: str
-DATA_SETUP_STARTED: str
-DATA_SETUP_TIME: str
-DATA_DEPS_REQS: str
-DATA_PERSISTENT_ERRORS: str
+DATA_SETUP: HassKey[dict[str, asyncio.Future[bool]]]
+DATA_SETUP_DONE: HassKey[dict[str, asyncio.Future[bool]]]
+DATA_SETUP_STARTED: HassKey[dict[tuple[str, str | None], float]]
+DATA_SETUP_TIME: HassKey[defaultdict[str, defaultdict[str | None, defaultdict[SetupPhases, float]]]]
+DATA_DEPS_REQS: HassKey[set[str]]
+DATA_PERSISTENT_ERRORS: HassKey[dict[str, str | None]]
 NOTIFY_FOR_TRANSLATION_KEYS: Incomplete
 SLOW_SETUP_WARNING: int
 SLOW_SETUP_MAX_WAIT: int
@@ -59,3 +60,4 @@ def async_pause_setup(hass: core.HomeAssistant, phase: SetupPhases) -> Generator
 def _setup_times(hass: core.HomeAssistant) -> defaultdict[str, defaultdict[str | None, defaultdict[SetupPhases, float]]]: ...
 def async_start_setup(hass: core.HomeAssistant, integration: str, phase: SetupPhases, group: str | None = None) -> Generator[None, None, None]: ...
 def async_get_setup_timings(hass: core.HomeAssistant) -> dict[str, float]: ...
+def async_get_domain_setup_times(hass: core.HomeAssistant, domain: str) -> Mapping[str | None, dict[SetupPhases, float]]: ...

homeassistant-stubs/util/__init__.pyi

@@ -2,10 +2,8 @@ from .dt import as_local as as_local, utcnow as utcnow
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable, KeysView
 from datetime import timedelta
-from typing import Any, TypeVar
+from typing import Any
 
-_T = TypeVar('_T')
-_U = TypeVar('_U')
 RE_SANITIZE_FILENAME: Incomplete
 RE_SANITIZE_PATH: Incomplete
 

homeassistant-stubs/util/async_.pyi

@@ -2,12 +2,10 @@ import concurrent.futures
 from _typeshed import Incomplete
 from asyncio import AbstractEventLoop, Future, Task
 from collections.abc import Callable as Callable, Coroutine
-from typing import Any, TypeVar, TypeVarTuple
+from typing import Any
 
 _LOGGER: Incomplete
 _SHUTDOWN_RUN_CALLBACK_THREADSAFE: str
-_T = TypeVar('_T')
-_Ts = TypeVarTuple('_Ts')
 
 def create_eager_task(coro: Coroutine[Any, Any, _T], *, name: str | None = None, loop: AbstractEventLoop | None = None) -> Task[_T]: ...
 def cancelling(task: Future[Any]) -> bool: ...

homeassistant-stubs/util/collection.pyi

@@ -0,0 +1,6 @@
+from collections.abc import Collection, Iterable
+from typing import Any
+
+def take(take_num: int, iterable: Iterable) -> list[Any]: ...
+def chunked(iterable: Iterable, chunked_num: int) -> Iterable[Any]: ...
+def chunked_or_all(iterable: Collection[Any], chunked_num: int) -> Iterable[Any]: ...

homeassistant-stubs/util/decorator.pyi

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

homeassistant-stubs/util/dt.pyi

@@ -11,8 +11,10 @@ STANDARD_DURATION_RE: Incomplete
 ISO8601_DURATION_RE: Incomplete
 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: ...
 
 utcnow: Incomplete
 

homeassistant-stubs/util/enum.pyi

@@ -1,10 +1,6 @@
-from collections.abc import Callable
-from enum import Enum
-from typing import Any, TypeVar
-
-_LruCacheT = TypeVar('_LruCacheT', bound=Callable)
-
-def lru_cache(func: _LruCacheT) -> _LruCacheT: ...
-_EnumT = TypeVar('_EnumT', bound=Enum)
+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: ...

homeassistant-stubs/util/hass_dict.pyi

@@ -0,0 +1,65 @@
+from typing import Any, Generic, TypeVar, overload
+
+__all__ = ['HassDict', 'HassEntryKey', 'HassKey']
+
+_T = TypeVar('_T')
+
+class _Key(Generic[_T]):
+    def __init__(self, value: str) -> None: ...
+    def __len__(self) -> int: ...
+    def __hash__(self) -> int: ...
+    def __eq__(self, other: object) -> bool: ...
+    def __getitem__(self, index: int) -> str: ...
+
+class HassEntryKey(_Key[_T]): ...
+class HassKey(_Key[_T]): ...
+
+class HassDict(dict[_Key[Any] | str, Any]):
+    @overload
+    def __getitem__(self, key: HassEntryKey[_S]) -> dict[str, _S]: ...
+    @overload
+    def __getitem__(self, key: HassKey[_S]) -> _S: ...
+    @overload
+    def __getitem__(self, key: str) -> Any: ...
+    @overload
+    def __setitem__(self, key: HassEntryKey[_S], value: dict[str, _S]) -> None: ...
+    @overload
+    def __setitem__(self, key: HassKey[_S], value: _S) -> None: ...
+    @overload
+    def __setitem__(self, key: str, value: Any) -> None: ...
+    @overload
+    def setdefault(self, key: HassEntryKey[_S], default: dict[str, _S]) -> dict[str, _S]: ...
+    @overload
+    def setdefault(self, key: HassKey[_S], default: _S) -> _S: ...
+    @overload
+    def setdefault(self, key: str, default: None = None) -> Any | None: ...
+    @overload
+    def setdefault(self, key: str, default: Any) -> Any: ...
+    @overload
+    def get(self, key: HassEntryKey[_S]) -> dict[str, _S] | None: ...
+    @overload
+    def get(self, key: HassEntryKey[_S], default: _U) -> dict[str, _S] | _U: ...
+    @overload
+    def get(self, key: HassKey[_S]) -> _S | None: ...
+    @overload
+    def get(self, key: HassKey[_S], default: _U) -> _S | _U: ...
+    @overload
+    def get(self, key: str) -> Any | None: ...
+    @overload
+    def get(self, key: str, default: Any) -> Any: ...
+    @overload
+    def pop(self, key: HassEntryKey[_S]) -> dict[str, _S]: ...
+    @overload
+    def pop(self, key: HassEntryKey[_S], default: dict[str, _S]) -> dict[str, _S]: ...
+    @overload
+    def pop(self, key: HassEntryKey[_S], default: _U) -> dict[str, _S] | _U: ...
+    @overload
+    def pop(self, key: HassKey[_S]) -> _S: ...
+    @overload
+    def pop(self, key: HassKey[_S], default: _S) -> _S: ...
+    @overload
+    def pop(self, key: HassKey[_S], default: _U) -> _S | _U: ...
+    @overload
+    def pop(self, key: str) -> Any: ...
+    @overload
+    def pop(self, key: str, default: _U) -> Any | _U: ...

homeassistant-stubs/util/limited_size_dict.pyi

@@ -1,9 +1,6 @@
 from _typeshed import Incomplete
 from collections import OrderedDict
-from typing import Any, TypeVar
-
-_KT = TypeVar('_KT')
-_VT = TypeVar('_VT')
+from typing import Any
 
 class LimitedSizeDict(OrderedDict[_KT, _VT]):
     size_limit: Incomplete

homeassistant-stubs/util/logging.pyi

@@ -2,10 +2,7 @@ import logging
 import logging.handlers
 from collections.abc import Callable, Coroutine
 from homeassistant.core import HassJobType as HassJobType, HomeAssistant as HomeAssistant, callback as callback, get_hassjob_callable_job_type as get_hassjob_callable_job_type
-from typing import Any, TypeVar, TypeVarTuple, overload
-
-_T = TypeVar('_T')
-_Ts = TypeVarTuple('_Ts')
+from typing import Any, overload
 
 class HomeAssistantQueueHandler(logging.handlers.QueueHandler):
     listener: logging.handlers.QueueListener | None