homeassistant-stubs 2025.4.0b9__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/components/tts/__init__.pyi

@@ -1,21 +1,23 @@
 import asyncio
 from .const import DEFAULT_CACHE_DIR as DEFAULT_CACHE_DIR, TtsAudioType as TtsAudioType
-from .entity import TextToSpeechEntity as TextToSpeechEntity
+from .entity import TTSAudioResponse as TTSAudioResponse, TextToSpeechEntity as TextToSpeechEntity
 from .legacy import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE, Provider as Provider
 from .media_source import generate_media_source_id as generate_media_source_id
 from .models import Voice as Voice
 from _typeshed import Incomplete
 from aiohttp import web
-from collections.abc import AsyncGenerator
-from dataclasses import dataclass
+from collections.abc import AsyncGenerator, MutableMapping
+from dataclasses import dataclass, field
 from datetime import datetime
 from homeassistant.components.http import HomeAssistantView
 from homeassistant.core import CALLBACK_TYPE, Event, HomeAssistant, callback
 from homeassistant.helpers.typing import ConfigType
+from pathlib import Path
 from propcache.api import cached_property
-from typing import Any
+from time import monotonic
+from typing import Any, Protocol
 
-__all__ = ['ATTR_AUDIO_OUTPUT', 'ATTR_PREFERRED_FORMAT', 'ATTR_PREFERRED_SAMPLE_BYTES', 'ATTR_PREFERRED_SAMPLE_CHANNELS', 'ATTR_PREFERRED_SAMPLE_RATE', 'CONF_LANG', 'DEFAULT_CACHE_DIR', 'PLATFORM_SCHEMA', 'PLATFORM_SCHEMA_BASE', 'Provider', 'ResultStream', 'SampleFormat', 'TextToSpeechEntity', 'TtsAudioType', 'Voice', 'async_default_engine', 'async_get_media_source_audio', 'generate_media_source_id']
+__all__ = ['ATTR_AUDIO_OUTPUT', 'ATTR_PREFERRED_FORMAT', 'ATTR_PREFERRED_SAMPLE_BYTES', 'ATTR_PREFERRED_SAMPLE_CHANNELS', 'ATTR_PREFERRED_SAMPLE_RATE', 'CONF_LANG', 'DEFAULT_CACHE_DIR', 'PLATFORM_SCHEMA', 'PLATFORM_SCHEMA_BASE', 'Provider', 'ResultStream', 'SampleFormat', 'TTSAudioResponse', 'TextToSpeechEntity', 'TtsAudioType', 'Voice', 'async_default_engine', 'async_get_media_source_audio', 'generate_media_source_id']
 
 ATTR_AUDIO_OUTPUT: str
 ATTR_PREFERRED_FORMAT: str
@@ -43,6 +45,8 @@ async def async_get_media_source_audio(hass: HomeAssistant, media_source_id: str
 
 @dataclass
 class ResultStream:
+    last_used: float = field(default_factory=monotonic, init=False)
+    hass: HomeAssistant
     token: str
     extension: str
     content_type: str
@@ -50,24 +54,33 @@ class ResultStream:
     use_file_cache: bool
     language: str
     options: dict
+    supports_streaming_input: bool
     _manager: SpeechManager
+    _override_media_path: Path | None = ...
     @cached_property
     def url(self) -> str: ...
     @cached_property
+    def media_source_id(self) -> str: ...
+    @cached_property
     def _result_cache(self) -> asyncio.Future[TTSCache]: ...
     @callback
-    def async_set_message_cache(self, cache: TTSCache) -> None: ...
-    @callback
     def async_set_message(self, message: str) -> None: ...
+    @callback
+    def async_set_message_stream(self, message_stream: AsyncGenerator[str]) -> None: ...
     async def async_stream_result(self) -> AsyncGenerator[bytes]: ...
+    def async_override_result(self, media_path: str | Path) -> None: ...
+    async def _async_stream_override_result(self) -> AsyncGenerator[bytes]: ...
 
-class MemcacheCleanup:
+class HasLastUsed(Protocol):
+    last_used: float
+
+class DictCleaning[T: HasLastUsed]:
     unsub: CALLBACK_TYPE | None
     hass: Incomplete
     maxage: Incomplete
     memcache: Incomplete
     cleanup_job: Incomplete
-    def __init__(self, hass: HomeAssistant, maxage: float, memcache: dict[str, TTSCache]) -> None: ...
+    def __init__(self, hass: HomeAssistant, maxage: float, memcache: MutableMapping[str, T]) -> None: ...
     @callback
     def schedule(self) -> None: ...
     @callback
@@ -85,6 +98,7 @@ class SpeechManager:
     mem_cache: dict[str, TTSCache]
     token_to_stream: dict[str, ResultStream]
     memcache_cleanup: Incomplete
+    token_to_stream_cleanup: Incomplete
     def __init__(self, hass: HomeAssistant, use_file_cache: bool, cache_dir: str, memory_cache_maxage: int) -> None: ...
     def _init_cache(self) -> dict[str, str]: ...
     async def async_init_cache(self) -> None: ...
@@ -94,13 +108,15 @@ class SpeechManager:
     @callback
     def process_options(self, engine_instance: TextToSpeechEntity | Provider, language: str | None, options: dict | None) -> tuple[str, dict[str, Any]]: ...
     @callback
-    def async_create_result_stream(self, engine: str, message: str | None = None, use_file_cache: bool | None = None, language: str | None = None, options: dict | None = None) -> ResultStream: ...
+    def async_get_result_stream(self, token: str) -> ResultStream | None: ...
+    @callback
+    def async_create_result_stream(self, engine: str, use_file_cache: bool | None = None, language: str | None = None, options: dict | None = None) -> ResultStream: ...
     @callback
-    def async_cache_message_in_memory(self, engine: str, message: str, use_file_cache: bool | None = None, language: str | None = None, options: dict | None = None) -> TTSCache: ...
+    def async_cache_message_stream_in_memory(self, engine: str, message_stream: AsyncGenerator[str], language: str, options: dict) -> TTSCache: ...
     @callback
-    def _async_ensure_cached_in_memory(self, engine: str, engine_instance: TextToSpeechEntity | Provider, message: str, use_file_cache: bool, language: str, options: dict) -> TTSCache: ...
+    def async_cache_message_in_memory(self, engine: str, message: str, use_file_cache: bool, language: str, options: dict) -> TTSCache: ...
     async def _load_data_into_cache(self, cache: TTSCache, engine_instance: TextToSpeechEntity | Provider, message: str, store_to_disk: bool, language: str, options: dict) -> None: ...
-    async def _async_generate_tts_audio(self, engine_instance: TextToSpeechEntity | Provider, message: str, language: str, options: dict[str, Any]) -> AsyncGenerator[bytes]: ...
+    async def _async_generate_tts_audio(self, engine_instance: TextToSpeechEntity | Provider, message_or_stream: str | AsyncGenerator[str], language: str, options: dict[str, Any]) -> AsyncGenerator[bytes]: ...
     async def _async_load_file(self, cache_key: str) -> AsyncGenerator[bytes]: ...
     @staticmethod
     def write_tags(filename: str, data: bytes, engine_name: str, message: str, language: str, options: dict | None) -> bytes: ...
@@ -119,6 +135,7 @@ class TextToSpeechView(HomeAssistantView):
     name: str
     manager: Incomplete
     def __init__(self, manager: SpeechManager) -> None: ...
+    async def head(self, request: web.Request, token: str) -> web.StreamResponse: ...
     async def get(self, request: web.Request, token: str) -> web.StreamResponse: ...
 
 # Names in __all__ with no definition:

homeassistant-stubs/components/tts/const.pyi

@@ -16,4 +16,5 @@ DEFAULT_TIME_MEMORY: int
 DOMAIN: str
 DATA_COMPONENT: HassKey[EntityComponent[TextToSpeechEntity]]
 DATA_TTS_MANAGER: HassKey[SpeechManager]
+MEDIA_SOURCE_STREAM_PATH: str
 type TtsAudioType = tuple[str | None, bytes | None]

homeassistant-stubs/components/tts/entity.pyi

@@ -43,6 +43,7 @@ class TextToSpeechEntity(RestoreEntity, cached_properties=CACHED_PROPERTIES_WITH
     def supported_options(self) -> list[str] | None: ...
     @cached_property
     def default_options(self) -> Mapping[str, Any] | None: ...
+    def async_supports_streaming_input(self) -> bool: ...
     @callback
     def async_get_supported_voices(self, language: str) -> list[Voice] | None: ...
     async def async_internal_added_to_hass(self) -> None: ...
@@ -51,4 +52,6 @@ class TextToSpeechEntity(RestoreEntity, cached_properties=CACHED_PROPERTIES_WITH
     async def internal_async_stream_tts_audio(self, request: TTSAudioRequest) -> TTSAudioResponse: ...
     async def async_stream_tts_audio(self, request: TTSAudioRequest) -> TTSAudioResponse: ...
     def get_tts_audio(self, message: str, language: str, options: dict[str, Any]) -> TtsAudioType: ...
+    @final
+    async def async_internal_get_tts_audio(self, message: str, language: str, options: dict[str, Any]) -> TtsAudioType: ...
     async def async_get_tts_audio(self, message: str, language: str, options: dict[str, Any]) -> TtsAudioType: ...

homeassistant-stubs/components/tts/legacy.pyi

@@ -14,7 +14,7 @@ from homeassistant.helpers.service import async_set_service_schema as async_set_
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from homeassistant.setup import SetupPhases as SetupPhases, async_prepare_setup_platform as async_prepare_setup_platform, async_start_setup as async_start_setup
 from homeassistant.util.yaml import load_yaml_dict as load_yaml_dict
-from typing import Any
+from typing import Any, final
 
 _LOGGER: Incomplete
 CONF_SERVICE_NAME: str
@@ -31,6 +31,7 @@ async def async_setup_legacy(hass: HomeAssistant, config: ConfigType) -> list[Co
 class Provider(metaclass=abc.ABCMeta):
     hass: HomeAssistant | None
     name: str | None
+    has_entity: bool
     @property
     def default_language(self) -> str | None: ...
     @property
@@ -44,3 +45,5 @@ class Provider(metaclass=abc.ABCMeta):
     def default_options(self) -> Mapping[str, Any] | None: ...
     def get_tts_audio(self, message: str, language: str, options: dict[str, Any]) -> TtsAudioType: ...
     async def async_get_tts_audio(self, message: str, language: str, options: dict[str, Any]) -> TtsAudioType: ...
+    @final
+    async def async_internal_get_tts_audio(self, message: str, language: str, options: dict[str, Any]) -> TtsAudioType: ...

homeassistant-stubs/components/tts/media_source.pyi

@@ -1,4 +1,4 @@
-from .const import DATA_COMPONENT as DATA_COMPONENT, DATA_TTS_MANAGER as DATA_TTS_MANAGER, DOMAIN as DOMAIN
+from .const import DATA_COMPONENT as DATA_COMPONENT, DATA_TTS_MANAGER as DATA_TTS_MANAGER, DOMAIN as DOMAIN, MEDIA_SOURCE_STREAM_PATH as MEDIA_SOURCE_STREAM_PATH
 from .helper import get_engine_instance as get_engine_instance
 from _typeshed import Incomplete
 from homeassistant.components.media_player import BrowseError as BrowseError, MediaClass as MediaClass
@@ -15,13 +15,19 @@ def generate_media_source_id(hass: HomeAssistant, message: str, engine: str | No
 
 class MediaSourceOptions(TypedDict):
     engine: str
-    message: str
     language: str | None
     options: dict | None
     use_file_cache: bool | None
 
+class ParsedMediaSourceId(TypedDict):
+    options: MediaSourceOptions
+    message: str
+
+class ParsedMediaSourceStreamId(TypedDict):
+    stream: str
+
 @callback
-def media_source_id_to_kwargs(media_source_id: str) -> MediaSourceOptions: ...
+def parse_media_source_id(media_source_id: str) -> ParsedMediaSourceId | ParsedMediaSourceStreamId: ...
 
 class TTSMediaSource(MediaSource):
     name: str

homeassistant-stubs/components/unifi/__init__.pyi

@@ -1,4 +1,4 @@
-from .const import PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
+from .const import DOMAIN as DOMAIN, PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
 from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
 from .services import async_setup_services as async_setup_services

homeassistant-stubs/components/unifi/button.pyi

@@ -1,9 +1,9 @@
 import aiounifi
 from . import UnifiConfigEntry as UnifiConfigEntry
-from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from .hub import UnifiHub as UnifiHub
-from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
-from aiounifi.models.api import ApiItemT
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, ItemEvent as ItemEvent
+from aiounifi.models.api import ApiItem as ApiItem
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
@@ -19,14 +19,14 @@ async def async_power_cycle_port_control_fn(api: aiounifi.Controller, obj_id: st
 async def async_regenerate_password_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiButtonEntityDescription(ButtonEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
+class UnifiButtonEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](ButtonEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     control_fn: Callable[[aiounifi.Controller, str], Coroutine[Any, Any, None]]
 
 ENTITY_DESCRIPTIONS: tuple[UnifiButtonEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class UnifiButtonEntity(UnifiEntity[HandlerT, ApiItemT], ButtonEntity):
+class UnifiButtonEntity[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntity[HandlerT, ApiItemT], ButtonEntity):
     entity_description: UnifiButtonEntityDescription[HandlerT, ApiItemT]
     async def async_press(self) -> None: ...
     @callback

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -1,6 +1,6 @@
 import voluptuous as vol
 from . import UnifiConfigEntry as UnifiConfigEntry
-from .const import CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DOMAIN as UNIFI_DOMAIN
+from .const import CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DOMAIN as DOMAIN
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
 from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
 from _typeshed import Incomplete
@@ -18,7 +18,7 @@ DEFAULT_SITE_ID: str
 DEFAULT_VERIFY_SSL: bool
 MODEL_PORTS: Incomplete
 
-class UnifiFlowHandler(ConfigFlow, domain=UNIFI_DOMAIN):
+class UnifiFlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
     sites: Sites
     @staticmethod

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -1,10 +1,11 @@
 import aiounifi
 from . import UnifiConfigEntry as UnifiConfigEntry
-from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn
+from .const import DOMAIN as DOMAIN
+from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
-from aiounifi.interfaces.api_handlers import ItemEvent
-from aiounifi.models.api import ApiItemT
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, ItemEvent
+from aiounifi.models.api import ApiItem as ApiItem
 from aiounifi.models.event import Event, EventKey
 from collections.abc import Callable as Callable, Mapping
 from dataclasses import dataclass
@@ -35,7 +36,7 @@ def async_client_is_connected_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_device_heartbeat_timedelta_fn(hub: UnifiHub, obj_id: str) -> timedelta: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiTrackerEntityDescription(UnifiEntityDescription[HandlerT, ApiItemT], ScannerEntityDescription):
+class UnifiTrackerEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntityDescription[HandlerT, ApiItemT], ScannerEntityDescription):
     heartbeat_timedelta_fn: Callable[[UnifiHub, str], timedelta]
     ip_address_fn: Callable[[aiounifi.Controller, str], str | None]
     is_connected_fn: Callable[[UnifiHub, str], bool]
@@ -47,7 +48,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiTrackerEntityDescription, ...]
 def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class UnifiScannerEntity(UnifiEntity[HandlerT, ApiItemT], ScannerEntity):
+class UnifiScannerEntity[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntity[HandlerT, ApiItemT], ScannerEntity):
     entity_description: UnifiTrackerEntityDescription
     _event_is_on: set[EventKey]
     _ignore_events: bool

homeassistant-stubs/components/unifi/entity.pyi

@@ -4,8 +4,8 @@ from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, DOMAIN as DOMAIN
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from abc import abstractmethod
-from aiounifi.interfaces.api_handlers import APIHandler, CallbackType, ItemEvent, UnsubscribeType
-from aiounifi.models.api import ApiItemT as ApiItemT
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, CallbackType, ItemEvent, UnsubscribeType
+from aiounifi.models.api import ApiItem as ApiItem
 from aiounifi.models.event import Event as Event, EventKey as EventKey
 from collections.abc import Callable
 from dataclasses import dataclass
@@ -13,11 +13,8 @@ from homeassistant.core import callback as callback
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity import Entity as Entity, EntityDescription as EntityDescription
-from typing import Generic, TypeVar
-
-HandlerT = TypeVar('HandlerT', bound=APIHandler)
-SubscriptionT = Callable[[CallbackType, ItemEvent], UnsubscribeType]
 
+type SubscriptionType = Callable[[CallbackType, ItemEvent], UnsubscribeType]
 @callback
 def async_device_available_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 @callback
@@ -30,7 +27,7 @@ def async_wlan_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
 def async_client_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiEntityDescription(EntityDescription, Generic[HandlerT, ApiItemT]):
+class UnifiEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](EntityDescription):
     api_handler_fn: Callable[[aiounifi.Controller], HandlerT]
     device_info_fn: Callable[[UnifiHub, str], DeviceInfo | None]
     object_fn: Callable[[aiounifi.Controller, str], ApiItemT]
@@ -44,7 +41,7 @@ class UnifiEntityDescription(EntityDescription, Generic[HandlerT, ApiItemT]):
     event_to_subscribe: tuple[EventKey, ...] | None = ...
     should_poll: bool = ...
 
-class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
+class UnifiEntity[HandlerT: APIHandler, ApiItemT: ApiItem](Entity, metaclass=abc.ABCMeta):
     entity_description: UnifiEntityDescription[HandlerT, ApiItemT]
     _attr_unique_id: str
     _obj_id: Incomplete

homeassistant-stubs/components/unifi/hub/api.pyi

@@ -1,10 +1,10 @@
 import aiounifi
 from ..const import CONF_SITE_ID as CONF_SITE_ID, LOGGER as LOGGER
 from ..errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+from collections.abc import Mapping
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers import aiohttp_client as aiohttp_client
-from types import MappingProxyType
 from typing import Any
 
-async def get_unifi_api(hass: HomeAssistant, config: MappingProxyType[str, Any]) -> aiounifi.Controller: ...
+async def get_unifi_api(hass: HomeAssistant, config: Mapping[str, Any]) -> aiounifi.Controller: ...

homeassistant-stubs/components/unifi/hub/entity_loader.pyi

@@ -1,3 +1,4 @@
+from .. import UnifiConfigEntry as UnifiConfigEntry
 from ..const import LOGGER as LOGGER, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
 from ..entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
 from .hub import UnifiHub as UnifiHub
@@ -18,11 +19,11 @@ class UnifiEntityLoader:
     api_updaters: Incomplete
     polling_api_updaters: Incomplete
     wireless_clients: Incomplete
-    _dataUpdateCoordinator: Incomplete
+    _data_update_coordinator: Incomplete
     _update_listener: Incomplete
     platforms: list[tuple[AddEntitiesCallback, type[UnifiEntity], tuple[UnifiEntityDescription, ...], bool]]
     known_objects: set[tuple[str, str]]
-    def __init__(self, hub: UnifiHub) -> None: ...
+    def __init__(self, hub: UnifiHub, config_entry: UnifiConfigEntry) -> None: ...
     async def initialize(self) -> None: ...
     async def _refresh_data(self, updaters: Sequence[Callable[[], Coroutine[Any, Any, None]]]) -> None: ...
     async def _update_pollable_api_data(self) -> None: ...

homeassistant-stubs/components/unifi/hub/hub.pyi

@@ -1,6 +1,6 @@
 import aiounifi
 from .. import UnifiConfigEntry as UnifiConfigEntry
-from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_SITE_ID as CONF_SITE_ID, PLATFORMS as PLATFORMS
+from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_SITE_ID as CONF_SITE_ID, DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
 from .config import UnifiConfig as UnifiConfig
 from .entity_helper import UnifiEntityHelper as UnifiEntityHelper
 from .entity_loader import UnifiEntityLoader as UnifiEntityLoader

homeassistant-stubs/components/unifi/image.pyi

@@ -1,9 +1,9 @@
 from . import UnifiConfigEntry as UnifiConfigEntry
-from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
-from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
-from aiounifi.models.api import ApiItemT
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, ItemEvent as ItemEvent
+from aiounifi.models.api import ApiItem as ApiItem
 from aiounifi.models.wlan import Wlan
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
@@ -16,7 +16,7 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 def async_wlan_qr_code_image_fn(hub: UnifiHub, wlan: Wlan) -> bytes: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiImageEntityDescription(ImageEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
+class UnifiImageEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](ImageEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     image_fn: Callable[[UnifiHub, ApiItemT], bytes]
     value_fn: Callable[[ApiItemT], str | None]
 
@@ -24,7 +24,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiImageEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class UnifiImageEntity(UnifiEntity[HandlerT, ApiItemT], ImageEntity):
+class UnifiImageEntity[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntity[HandlerT, ApiItemT], ImageEntity):
     entity_description: UnifiImageEntityDescription[HandlerT, ApiItemT]
     _attr_content_type: str
     current_image: bytes | None

homeassistant-stubs/components/unifi/light.pyi

@@ -0,0 +1,48 @@
+from . import UnifiConfigEntry as UnifiConfigEntry
+from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn
+from .hub import UnifiHub as UnifiHub
+from _typeshed import Incomplete
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, ItemEvent
+from aiounifi.models.api import ApiItem as ApiItem
+from aiounifi.models.device import Device
+from collections.abc import Callable as Callable, Coroutine
+from dataclasses import dataclass
+from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_RGB_COLOR as ATTR_RGB_COLOR, ColorMode as ColorMode, LightEntity as LightEntity, LightEntityDescription as LightEntityDescription, LightEntityFeature as LightEntityFeature
+from homeassistant.const import EntityCategory as EntityCategory
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from homeassistant.util.color import rgb_hex_to_rgb_list as rgb_hex_to_rgb_list
+from typing import Any
+
+def convert_brightness_to_unifi(ha_brightness: int) -> int: ...
+def convert_brightness_to_ha(unifi_brightness: int) -> int: ...
+def get_device_brightness_or_default(device: Device) -> int: ...
+@callback
+def async_device_led_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+@callback
+def async_device_led_is_on_fn(hub: UnifiHub, device: Device) -> bool: ...
+async def async_device_led_control_fn(hub: UnifiHub, obj_id: str, turn_on: bool, **kwargs: Any) -> None: ...
+
+@dataclass(frozen=True, kw_only=True)
+class UnifiLightEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](LightEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
+    control_fn: Callable[[UnifiHub, str, bool], Coroutine[Any, Any, None]]
+    is_on_fn: Callable[[UnifiHub, ApiItemT], bool]
+
+ENTITY_DESCRIPTIONS: tuple[UnifiLightEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class UnifiLightEntity[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntity[HandlerT, ApiItemT], LightEntity):
+    entity_description: UnifiLightEntityDescription[HandlerT, ApiItemT]
+    _attr_supported_features: Incomplete
+    _attr_supported_color_modes: Incomplete
+    _attr_color_mode: Incomplete
+    @callback
+    def async_initiate_state(self) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    _attr_is_on: Incomplete
+    _attr_brightness: Incomplete
+    _attr_rgb_color: Incomplete
+    @callback
+    def async_update_state(self, event: ItemEvent, obj_id: str) -> None: ...

homeassistant-stubs/components/unifi/sensor.pyi

@@ -1,10 +1,10 @@
 from . import UnifiConfigEntry as UnifiConfigEntry
 from .const import DEVICE_STATES as DEVICE_STATES
-from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_client_device_info_fn as async_client_device_info_fn, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_client_device_info_fn as async_client_device_info_fn, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
-from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
-from aiounifi.models.api import ApiItemT
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, ItemEvent as ItemEvent
+from aiounifi.models.api import ApiItem as ApiItem
 from aiounifi.models.client import Client
 from aiounifi.models.device import Device, TypedDeviceTemperature as TypedDeviceTemperature, TypedDeviceUptimeStatsWanMonitor as TypedDeviceUptimeStatsWanMonitor
 from aiounifi.models.wlan import Wlan
@@ -32,6 +32,8 @@ def async_client_tx_value_fn(hub: UnifiHub, client: Client) -> float: ...
 @callback
 def async_client_uptime_value_fn(hub: UnifiHub, client: Client) -> datetime: ...
 @callback
+def async_wired_client_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+@callback
 def async_wlan_client_value_fn(hub: UnifiHub, wlan: Wlan) -> int: ...
 @callback
 def async_device_clients_value_fn(hub: UnifiHub, device: Device) -> int: ...
@@ -66,7 +68,7 @@ def _device_temperature(temperature_name: str, temperatures: list[TypedDeviceTem
 def make_device_temperatur_sensors() -> tuple[UnifiSensorEntityDescription, ...]: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiSensorEntityDescription(SensorEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
+class UnifiSensorEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](SensorEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     value_fn: Callable[[UnifiHub, ApiItemT], datetime | float | str | None]
     is_connected_fn: Callable[[UnifiHub, str], bool] | None = ...
     value_changed_fn: Callable[[StateType | date | datetime | Decimal, datetime | float | str | None], bool] = ...
@@ -75,7 +77,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiSensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class UnifiSensorEntity(UnifiEntity[HandlerT, ApiItemT], SensorEntity):
+class UnifiSensorEntity[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntity[HandlerT, ApiItemT], SensorEntity):
     entity_description: UnifiSensorEntityDescription[HandlerT, ApiItemT]
     _attr_available: bool
     @callback

homeassistant-stubs/components/unifi/services.pyi

@@ -1,3 +1,4 @@
+from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID

homeassistant-stubs/components/unifi/switch.pyi

@@ -1,11 +1,11 @@
 import aiounifi
 from . import UnifiConfigEntry as UnifiConfigEntry
-from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER
-from .entity import HandlerT as HandlerT, SubscriptionT as SubscriptionT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_client_device_info_fn as async_client_device_info_fn, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, DOMAIN as DOMAIN
+from .entity import SubscriptionType as SubscriptionType, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_client_device_info_fn as async_client_device_info_fn, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
-from aiounifi.interfaces.api_handlers import ItemEvent
-from aiounifi.models.api import ApiItemT
+from aiounifi.interfaces.api_handlers import APIHandler as APIHandler, ItemEvent
+from aiounifi.models.api import ApiItem as ApiItem
 from aiounifi.models.dpi_restriction_group import DPIRestrictionGroup
 from aiounifi.models.event import Event as Event
 from collections.abc import Callable as Callable, Coroutine
@@ -35,18 +35,21 @@ async def async_firewall_policy_control_fn(hub: UnifiHub, obj_id: str, target: b
 def async_firewall_policy_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 @callback
 def async_outlet_switching_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+@callback
+def async_port_control_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 async def async_outlet_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_poe_port_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_port_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_port_forward_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_traffic_rule_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_traffic_route_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_wlan_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiSwitchEntityDescription(SwitchEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
+class UnifiSwitchEntityDescription[HandlerT: APIHandler, ApiItemT: ApiItem](SwitchEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     control_fn: Callable[[UnifiHub, str, bool], Coroutine[Any, Any, None]]
     is_on_fn: Callable[[UnifiHub, ApiItemT], bool]
-    custom_subscribe: Callable[[aiounifi.Controller], SubscriptionT] | None = ...
+    custom_subscribe: Callable[[aiounifi.Controller], SubscriptionType] | None = ...
     only_event_for_state_change: bool = ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSwitchEntityDescription, ...]
@@ -55,7 +58,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiSwitchEntityDescription, ...]
 def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class UnifiSwitchEntity(UnifiEntity[HandlerT, ApiItemT], SwitchEntity):
+class UnifiSwitchEntity[HandlerT: APIHandler, ApiItemT: ApiItem](UnifiEntity[HandlerT, ApiItemT], SwitchEntity):
     entity_description: UnifiSwitchEntityDescription[HandlerT, ApiItemT]
     @callback
     def async_initiate_state(self) -> None: ...

homeassistant-stubs/components/unifi/update.pyi

@@ -10,16 +10,14 @@ from dataclasses import dataclass
 from homeassistant.components.update import UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityDescription as UpdateEntityDescription, UpdateEntityFeature as UpdateEntityFeature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from typing import Any, TypeVar
+from typing import Any
 
 LOGGER: Incomplete
-_DataT = TypeVar('_DataT', bound=Device)
-_HandlerT = TypeVar('_HandlerT', bound=Devices)
 
 async def async_device_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
-class UnifiUpdateEntityDescription(UpdateEntityDescription, UnifiEntityDescription[_HandlerT, _DataT]):
+class UnifiUpdateEntityDescription[_HandlerT: Devices, _DataT: Device](UpdateEntityDescription, UnifiEntityDescription[_HandlerT, _DataT]):
     control_fn: Callable[[aiounifi.Controller, str], Coroutine[Any, Any, None]]
     state_fn: Callable[[aiounifi.Controller, _DataT], bool]
 
@@ -27,7 +25,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiUpdateEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class UnifiDeviceUpdateEntity(UnifiEntity[_HandlerT, _DataT], UpdateEntity):
+class UnifiDeviceUpdateEntity[_HandlerT: Devices, _DataT: Device](UnifiEntity[_HandlerT, _DataT], UpdateEntity):
     entity_description: UnifiUpdateEntityDescription[_HandlerT, _DataT]
     _attr_supported_features: Incomplete
     @callback

homeassistant-stubs/components/unifiprotect/__init__.pyi

@@ -1,14 +1,15 @@
-from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_ALLOW_EA as CONF_ALLOW_EA, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN, MIN_REQUIRED_PROTECT_V as MIN_REQUIRED_PROTECT_V, OUTDATED_LOG_MESSAGE as OUTDATED_LOG_MESSAGE, PLATFORMS as PLATFORMS
+from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_ALLOW_EA as CONF_ALLOW_EA, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN, MIN_REQUIRED_PROTECT_V as MIN_REQUIRED_PROTECT_V, PLATFORMS as PLATFORMS
 from .data import ProtectData as ProtectData, UFPConfigEntry as UFPConfigEntry
-from .discovery import async_start_discovery as async_start_discovery
+from .discovery import DATA_UNIFIPROTECT as DATA_UNIFIPROTECT, UniFiProtectRuntimeData as UniFiProtectRuntimeData, async_start_discovery as async_start_discovery
 from .migrate import async_migrate_data as async_migrate_data
 from .services import async_setup_services as async_setup_services
 from .utils import _async_unifi_mac_from_hass as _async_unifi_mac_from_hass, async_create_api_client as async_create_api_client, async_get_devices as async_get_devices
 from .views import SnapshotProxyView as SnapshotProxyView, ThumbnailProxyView as ThumbnailProxyView, VideoEventProxyView as VideoEventProxyView, VideoProxyView as VideoProxyView
 from _typeshed import Incomplete
-from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
+from homeassistant.const import CONF_API_KEY as CONF_API_KEY, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryError as ConfigEntryError, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers import device_registry as dr
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity
 from homeassistant.helpers.typing import ConfigType as ConfigType
@@ -18,11 +19,11 @@ from uiprotect.test_util.anonymize import anonymize_data as anonymize_data
 _LOGGER: Incomplete
 SCAN_INTERVAL: Incomplete
 CONFIG_SCHEMA: Incomplete
-EARLY_ACCESS_URL: str
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...
 async def _async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, data_service: ProtectData, bootstrap: Bootstrap) -> None: ...
-async def _async_options_updated(hass: HomeAssistant, entry: UFPConfigEntry) -> None: ...
 async def async_unload_entry(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...
+async def async_remove_entry(hass: HomeAssistant, entry: UFPConfigEntry) -> None: ...
 async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: UFPConfigEntry, device_entry: dr.DeviceEntry) -> bool: ...
+async def async_migrate_entry(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...

homeassistant-stubs/components/unifiprotect/binary_sensor.pyi

@@ -11,6 +11,7 @@ from uiprotect.data import ModelType, NVR as NVR, ProtectAdoptableDeviceModel as
 from uiprotect.data.nvr import UOSDisk as UOSDisk
 
 _KEY_DOOR: str
+PARALLEL_UPDATES: int
 
 @dataclasses.dataclass(frozen=True, kw_only=True)
 class ProtectBinaryEntityDescription(ProtectEntityDescription, BinarySensorEntityDescription): ...