homeassistant-stubs 2023.8.3__py3-none-any.whl → 2023.9.0__py3-none-any.whl

homeassistant-stubs/components/assist_pipeline/__init__.pyi

@@ -1,10 +1,10 @@
 from .const import DOMAIN as DOMAIN
 from .error import PipelineNotFound as PipelineNotFound
-from .pipeline import Pipeline as Pipeline, PipelineEvent as PipelineEvent, PipelineEventCallback, PipelineEventType as PipelineEventType, async_create_default_pipeline as async_create_default_pipeline, async_get_pipelines as async_get_pipelines
+from .pipeline import Pipeline as Pipeline, PipelineEvent as PipelineEvent, PipelineEventCallback, PipelineEventType as PipelineEventType, PipelineStage, WakeWordSettings as WakeWordSettings, async_create_default_pipeline as async_create_default_pipeline, async_get_pipelines as async_get_pipelines
 from collections.abc import AsyncIterable
 from homeassistant.components import stt
 from homeassistant.core import Context, HomeAssistant
 from homeassistant.helpers.typing import ConfigType
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
-async def async_pipeline_from_audio_stream(hass: HomeAssistant, context: Context, event_callback: PipelineEventCallback, stt_metadata: stt.SpeechMetadata, stt_stream: AsyncIterable[bytes], pipeline_id: str | None = ..., conversation_id: str | None = ..., tts_audio_output: str | None = ..., device_id: str | None = ...) -> None: ...
+async def async_pipeline_from_audio_stream(hass: HomeAssistant, *, context: Context, event_callback: PipelineEventCallback, stt_metadata: stt.SpeechMetadata, stt_stream: AsyncIterable[bytes], pipeline_id: str | None = ..., conversation_id: str | None = ..., tts_audio_output: str | None = ..., wake_word_settings: WakeWordSettings | None = ..., device_id: str | None = ..., start_stage: PipelineStage = ..., end_stage: PipelineStage = ...) -> None: ...

homeassistant-stubs/components/assist_pipeline/const.pyi

@@ -1 +1,4 @@
+from _typeshed import Incomplete
+
 DOMAIN: str
+DATA_CONFIG: Incomplete

homeassistant-stubs/components/assist_pipeline/error.pyi

@@ -7,6 +7,8 @@ class PipelineError(HomeAssistantError):
     def __init__(self, code: str, message: str) -> None: ...
 
 class PipelineNotFound(PipelineError): ...
+class WakeWordDetectionError(PipelineError): ...
+class WakeWordTimeoutError(WakeWordDetectionError): ...
 class SpeechToTextError(PipelineError): ...
 class IntentRecognitionError(PipelineError): ...
 class TextToSpeechError(PipelineError): ...

homeassistant-stubs/components/assist_pipeline/pipeline.pyi

@@ -1,15 +1,20 @@
-from .const import DOMAIN as DOMAIN
-from .error import IntentRecognitionError as IntentRecognitionError, PipelineError as PipelineError, PipelineNotFound as PipelineNotFound, SpeechToTextError as SpeechToTextError, TextToSpeechError as TextToSpeechError
+from .const import DATA_CONFIG as DATA_CONFIG, DOMAIN as DOMAIN
+from .error import IntentRecognitionError as IntentRecognitionError, PipelineError as PipelineError, PipelineNotFound as PipelineNotFound, SpeechToTextError as SpeechToTextError, TextToSpeechError as TextToSpeechError, WakeWordDetectionError as WakeWordDetectionError, WakeWordTimeoutError as WakeWordTimeoutError
+from .ring_buffer import RingBuffer as RingBuffer
+from .vad import VoiceActivityTimeout as VoiceActivityTimeout, VoiceCommandSegmenter as VoiceCommandSegmenter
 from _typeshed import Incomplete
-from collections.abc import AsyncIterable, Callable, Iterable
+from collections.abc import AsyncGenerator, AsyncIterable, Callable, Iterable
 from enum import StrEnum
-from homeassistant.components import conversation as conversation, media_source as media_source, stt as stt, tts as tts, websocket_api as websocket_api
+from homeassistant.components import conversation as conversation, media_source as media_source, stt as stt, tts as tts, wake_word as wake_word, websocket_api as websocket_api
 from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.collection import CollectionError as CollectionError, ItemNotFound as ItemNotFound, SerializedStorageCollection as SerializedStorageCollection, StorageCollection as StorageCollection, StorageCollectionWebsocket as StorageCollectionWebsocket
 from homeassistant.helpers.singleton import singleton as singleton
 from homeassistant.helpers.storage import Store as Store
 from homeassistant.util.limited_size_dict import LimitedSizeDict as LimitedSizeDict
+from pathlib import Path
+from queue import Queue
+from threading import Thread
 from typing import Any
 
 _LOGGER: Incomplete
@@ -32,7 +37,11 @@ def async_get_pipelines(hass: HomeAssistant) -> Iterable[Pipeline]: ...
 class PipelineEventType(StrEnum):
     RUN_START: str
     RUN_END: str
+    WAKE_WORD_START: str
+    WAKE_WORD_END: str
     STT_START: str
+    STT_VAD_START: str
+    STT_VAD_END: str
     STT_END: str
     INTENT_START: str
     INTENT_END: str
@@ -45,7 +54,6 @@ class PipelineEvent:
     data: dict[str, Any] | None
     timestamp: str
     def __init__(self, type, data, timestamp) -> None: ...
-    def __mypy-replace(*, type, data, timestamp) -> None: ...
 PipelineEventCallback = Callable[[PipelineEvent], None]
 
 class Pipeline:
@@ -61,9 +69,9 @@ class Pipeline:
     id: str
     def to_json(self) -> dict[str, Any]: ...
     def __init__(self, conversation_engine, conversation_language, language, name, stt_engine, stt_language, tts_engine, tts_language, tts_voice, id) -> None: ...
-    def __mypy-replace(*, conversation_engine, conversation_language, language, name, stt_engine, stt_language, tts_engine, tts_language, tts_voice, id) -> None: ...
 
 class PipelineStage(StrEnum):
+    WAKE_WORD: str
     STT: str
     INTENT: str
     TTS: str
@@ -75,6 +83,11 @@ class PipelineRunValidationError(Exception): ...
 class InvalidPipelineStagesError(PipelineRunValidationError):
     def __init__(self, start_stage: PipelineStage, end_stage: PipelineStage) -> None: ...
 
+class WakeWordSettings:
+    timeout: float | None
+    audio_seconds_to_buffer: float
+    def __init__(self, timeout, audio_seconds_to_buffer) -> None: ...
+
 class PipelineRun:
     hass: HomeAssistant
     context: Context
@@ -86,23 +99,34 @@ class PipelineRun:
     runner_data: Any | None
     intent_agent: str | None
     tts_audio_output: str | None
+    wake_word_settings: WakeWordSettings | None
     id: str
     stt_provider: stt.SpeechToTextEntity | stt.Provider
     tts_engine: str
     tts_options: dict | None
+    wake_word_engine: str
+    wake_word_provider: wake_word.WakeWordDetectionEntity
+    debug_recording_thread: Thread | None
+    debug_recording_queue: Queue[str | bytes | None] | None
     def __post_init__(self) -> None: ...
     def process_event(self, event: PipelineEvent) -> None: ...
-    def start(self) -> None: ...
-    def end(self) -> None: ...
+    def start(self, device_id: str | None) -> None: ...
+    async def end(self) -> None: ...
+    async def prepare_wake_word_detection(self) -> None: ...
+    async def wake_word_detection(self, stream: AsyncIterable[bytes], audio_chunks_for_stt: list[bytes]) -> wake_word.DetectionResult | None: ...
+    async def _wake_word_audio_stream(self, audio_stream: AsyncIterable[bytes], stt_audio_buffer: RingBuffer | None, wake_word_vad: VoiceActivityTimeout | None, sample_rate: int = ..., sample_width: int = ...) -> AsyncIterable[tuple[bytes, int]]: ...
     async def prepare_speech_to_text(self, metadata: stt.SpeechMetadata) -> None: ...
     async def speech_to_text(self, metadata: stt.SpeechMetadata, stream: AsyncIterable[bytes]) -> str: ...
+    async def _speech_to_text_stream(self, audio_stream: AsyncIterable[bytes], stt_vad: VoiceCommandSegmenter | None, sample_rate: int = ..., sample_width: int = ...) -> AsyncGenerator[bytes, None]: ...
     async def prepare_recognize_intent(self) -> None: ...
     async def recognize_intent(self, intent_input: str, conversation_id: str | None, device_id: str | None) -> str: ...
     async def prepare_text_to_speech(self) -> None: ...
     async def text_to_speech(self, tts_input: str) -> str: ...
-    def __init__(self, hass, context, pipeline, start_stage, end_stage, event_callback, language, runner_data, intent_agent, tts_audio_output, id) -> None: ...
-    def __mypy-replace(*, hass, context, pipeline, start_stage, end_stage, event_callback, language, runner_data, intent_agent, tts_audio_output, id, stt_provider, tts_engine, tts_options) -> None: ...
-    def __mypy-post_init(self) -> None: ...
+    def _start_debug_recording_thread(self, device_id: str | None) -> None: ...
+    async def _stop_debug_recording_thread(self) -> None: ...
+    def __init__(self, hass, context, pipeline, start_stage, end_stage, event_callback, language, runner_data, intent_agent, tts_audio_output, wake_word_settings, id, debug_recording_thread, debug_recording_queue) -> None: ...
+
+def _pipeline_debug_recording_thread_proc(run_recording_dir: Path, queue: Queue[str | bytes | None], message_timeout: float = ...) -> None: ...
 
 class PipelineInput:
     run: PipelineRun
@@ -115,7 +139,6 @@ class PipelineInput:
     async def execute(self) -> None: ...
     async def validate(self) -> None: ...
     def __init__(self, run, stt_metadata, stt_stream, intent_input, tts_input, conversation_id, device_id) -> None: ...
-    def __mypy-replace(*, run, stt_metadata, stt_stream, intent_input, tts_input, conversation_id, device_id) -> None: ...
 
 class PipelinePreferred(CollectionError):
     item_id: Incomplete
@@ -150,12 +173,10 @@ class PipelineData:
     pipeline_store: PipelineStorageCollection
     pipeline_devices: set[str]
     def __init__(self, pipeline_runs, pipeline_store) -> None: ...
-    def __mypy-replace(*, pipeline_runs, pipeline_store, pipeline_devices) -> None: ...
 
 class PipelineRunDebug:
     events: list[PipelineEvent]
     timestamp: str
     def __init__(self) -> None: ...
-    def __mypy-replace(*, events, timestamp) -> None: ...
 
 async def async_setup_pipeline_store(hass: HomeAssistant) -> PipelineData: ...

homeassistant-stubs/components/assist_pipeline/ring_buffer.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+
+class RingBuffer:
+    _buffer: Incomplete
+    _pos: int
+    _length: int
+    _maxlen: Incomplete
+    def __init__(self, maxlen: int) -> None: ...
+    @property
+    def maxlen(self) -> int: ...
+    @property
+    def pos(self) -> int: ...
+    def __len__(self) -> int: ...
+    def put(self, data: bytes) -> None: ...
+    def getvalue(self) -> bytes: ...

homeassistant-stubs/components/assist_pipeline/vad.pyi

@@ -1,7 +1,11 @@
 import webrtcvad
+from _typeshed import Incomplete
+from collections.abc import Iterable
 from enum import StrEnum
+from typing import Final
 
-_SAMPLE_RATE: int
+_SAMPLE_RATE: Final[int]
+_SAMPLE_WIDTH: Final[int]
 
 class VadSensitivity(StrEnum):
     DEFAULT: str
@@ -10,9 +14,21 @@ class VadSensitivity(StrEnum):
     @staticmethod
     def to_seconds(sensitivity: VadSensitivity | str) -> float: ...
 
+class AudioBuffer:
+    _buffer: Incomplete
+    _length: int
+    def __init__(self, maxlen: int) -> None: ...
+    @property
+    def length(self) -> int: ...
+    def clear(self) -> None: ...
+    def append(self, data: bytes) -> None: ...
+    def bytes(self) -> bytes: ...
+    def __len__(self) -> int: ...
+    def __bool__(self) -> bool: ...
+
 class VoiceCommandSegmenter:
     vad_mode: int
-    vad_frames: int
+    vad_samples_per_chunk: int
     speech_seconds: float
     silence_seconds: float
     timeout_seconds: float
@@ -23,13 +39,32 @@ class VoiceCommandSegmenter:
     _timeout_seconds_left: float
     _reset_seconds_left: float
     _vad: webrtcvad.Vad
-    _audio_buffer: bytes
+    _leftover_chunk_buffer: AudioBuffer
     _bytes_per_chunk: int
     _seconds_per_chunk: float
     def __post_init__(self) -> None: ...
     def reset(self) -> None: ...
     def process(self, samples: bytes) -> bool: ...
+    @property
+    def audio_buffer(self) -> bytes: ...
     def _process_chunk(self, chunk: bytes) -> bool: ...
-    def __init__(self, vad_mode, vad_frames, speech_seconds, silence_seconds, timeout_seconds, reset_seconds, in_command, _speech_seconds_left, _silence_seconds_left, _timeout_seconds_left, _reset_seconds_left, _vad, _audio_buffer, _bytes_per_chunk, _seconds_per_chunk) -> None: ...
-    def __mypy-replace(*, vad_mode, vad_frames, speech_seconds, silence_seconds, timeout_seconds, reset_seconds, in_command, _speech_seconds_left, _silence_seconds_left, _timeout_seconds_left, _reset_seconds_left, _vad, _audio_buffer, _bytes_per_chunk, _seconds_per_chunk) -> None: ...
-    def __mypy-post_init(self) -> None: ...
+    def __init__(self, vad_mode, vad_samples_per_chunk, speech_seconds, silence_seconds, timeout_seconds, reset_seconds, in_command, _speech_seconds_left, _silence_seconds_left, _timeout_seconds_left, _reset_seconds_left, _vad) -> None: ...
+
+class VoiceActivityTimeout:
+    silence_seconds: float
+    reset_seconds: float
+    vad_mode: int
+    vad_samples_per_chunk: int
+    _silence_seconds_left: float
+    _reset_seconds_left: float
+    _vad: webrtcvad.Vad
+    _leftover_chunk_buffer: AudioBuffer
+    _bytes_per_chunk: int
+    _seconds_per_chunk: float
+    def __post_init__(self) -> None: ...
+    def reset(self) -> None: ...
+    def process(self, samples: bytes) -> bool: ...
+    def _process_chunk(self, chunk: bytes) -> bool: ...
+    def __init__(self, silence_seconds, reset_seconds, vad_mode, vad_samples_per_chunk, _silence_seconds_left, _reset_seconds_left, _vad) -> None: ...
+
+def chunk_samples(samples: bytes, bytes_per_chunk: int, leftover_chunk_buffer: AudioBuffer) -> Iterable[bytes]: ...

homeassistant-stubs/components/assist_pipeline/websocket_api.pyi

@@ -1,7 +1,6 @@
 from .const import DOMAIN as DOMAIN
 from .error import PipelineNotFound as PipelineNotFound
-from .pipeline import PipelineData as PipelineData, PipelineError as PipelineError, PipelineEvent as PipelineEvent, PipelineEventType as PipelineEventType, PipelineInput as PipelineInput, PipelineRun as PipelineRun, PipelineStage as PipelineStage, async_get_pipeline as async_get_pipeline
-from .vad import VoiceCommandSegmenter as VoiceCommandSegmenter
+from .pipeline import PipelineData as PipelineData, PipelineError as PipelineError, PipelineEvent as PipelineEvent, PipelineEventType as PipelineEventType, PipelineInput as PipelineInput, PipelineRun as PipelineRun, PipelineStage as PipelineStage, WakeWordSettings as WakeWordSettings, async_get_pipeline as async_get_pipeline
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from homeassistant.components import conversation as conversation, stt as stt, tts as tts, websocket_api as websocket_api
@@ -10,6 +9,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from typing import Any
 
 DEFAULT_TIMEOUT: int
+DEFAULT_WAKE_WORD_TIMEOUT: int
 _LOGGER: Incomplete
 
 def async_register_websocket_api(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/asuswrt/router.pyi

@@ -7,9 +7,8 @@ from homeassistant.components.device_tracker import CONF_CONSIDER_HOME as CONF_C
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
-from homeassistant.helpers.device_registry import format_mac as format_mac
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo, format_mac as format_mac
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
-from homeassistant.helpers.entity import DeviceInfo as DeviceInfo
 from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 from homeassistant.util import slugify as slugify

homeassistant-stubs/components/asuswrt/sensor.pyi

@@ -12,7 +12,6 @@ from homeassistant.util import slugify as slugify
 class AsusWrtSensorEntityDescription(SensorEntityDescription):
     factor: int | None
     def __init__(self, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, factor) -> None: ...
-    def __mypy-replace(*, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, factor) -> None: ...
 
 UNIT_DEVICES: str
 CONNECTION_SENSORS: tuple[AsusWrtSensorEntityDescription, ...]

homeassistant-stubs/components/auth/__init__.pyi

@@ -3,7 +3,7 @@ from _typeshed import Incomplete
 from aiohttp import web
 from collections.abc import Callable
 from homeassistant.auth import InvalidAuthError as InvalidAuthError
-from homeassistant.auth.models import Credentials as Credentials, TOKEN_TYPE_LONG_LIVED_ACCESS_TOKEN as TOKEN_TYPE_LONG_LIVED_ACCESS_TOKEN, User as User
+from homeassistant.auth.models import Credentials as Credentials, RefreshToken as RefreshToken, TOKEN_TYPE_LONG_LIVED_ACCESS_TOKEN as TOKEN_TYPE_LONG_LIVED_ACCESS_TOKEN, User as User
 from homeassistant.components import websocket_api as websocket_api
 from homeassistant.components.http.auth import async_sign_path as async_sign_path, async_user_not_allowed_do_auth as async_user_not_allowed_do_auth
 from homeassistant.components.http.ban import log_invalid_auth as log_invalid_auth
@@ -54,4 +54,5 @@ async def websocket_current_user(hass: HomeAssistant, connection: websocket_api.
 async def websocket_create_long_lived_access_token(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def websocket_refresh_tokens(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def websocket_delete_refresh_token(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def websocket_delete_all_refresh_tokens(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def websocket_sign_path(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/automation/__init__.pyi

@@ -146,7 +146,6 @@ class AutomationEntityConfig:
     raw_config: ConfigType | None
     validation_failed: bool
     def __init__(self, config_block, list_no, raw_blueprint_inputs, raw_config, validation_failed) -> None: ...
-    def __mypy-replace(*, config_block, list_no, raw_blueprint_inputs, raw_config, validation_failed) -> None: ...
 
 async def _prepare_automation_config(hass: HomeAssistant, config: ConfigType) -> list[AutomationEntityConfig]: ...
 def _automation_name(automation_config: AutomationEntityConfig) -> str: ...

homeassistant-stubs/components/awair/__init__.pyi

@@ -21,7 +21,6 @@ class AwairResult:
     device: AwairBaseDevice
     air_data: AirData
     def __init__(self, device, air_data) -> None: ...
-    def __mypy-replace(*, device, air_data) -> None: ...
 
 class AwairDataUpdateCoordinator(DataUpdateCoordinator[dict[str, AwairResult]]):
     _config_entry: Incomplete

homeassistant-stubs/components/awair/sensor.pyi

@@ -5,7 +5,7 @@ from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceCla
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_CONNECTIONS as ATTR_CONNECTIONS, ATTR_SW_VERSION as ATTR_SW_VERSION, CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_BILLION as CONCENTRATION_PARTS_PER_BILLION, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, LIGHT_LUX as LIGHT_LUX, PERCENTAGE as PERCENTAGE, UnitOfSoundPressure as UnitOfSoundPressure, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity import DeviceInfo as DeviceInfo
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from python_awair.air_data import AirData as AirData
@@ -17,11 +17,9 @@ DUST_ALIASES: Incomplete
 class AwairRequiredKeysMixin:
     unique_id_tag: str
     def __init__(self, unique_id_tag) -> None: ...
-    def __mypy-replace(*, unique_id_tag) -> None: ...
 
 class AwairSensorEntityDescription(SensorEntityDescription, AwairRequiredKeysMixin):
     def __init__(self, unique_id_tag, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
-    def __mypy-replace(*, unique_id_tag, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
 
 SENSOR_TYPE_SCORE: Incomplete
 SENSOR_TYPES: tuple[AwairSensorEntityDescription, ...]

homeassistant-stubs/components/backup/manager.pyi

@@ -18,7 +18,6 @@ class Backup:
     size: float
     def as_dict(self) -> dict: ...
     def __init__(self, slug, name, date, path, size) -> None: ...
-    def __mypy-replace(*, slug, name, date, path, size) -> None: ...
 
 class BackupPlatformProtocol(Protocol):
     async def async_pre_backup(self, hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/baf/binary_sensor.pyi

@@ -12,11 +12,9 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 class BAFBinarySensorDescriptionMixin:
     value_fn: Callable[[Device], bool | None]
     def __init__(self, value_fn) -> None: ...
-    def __mypy-replace(*, value_fn) -> None: ...
 
 class BAFBinarySensorDescription(BinarySensorEntityDescription, BAFBinarySensorDescriptionMixin):
     def __init__(self, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
-    def __mypy-replace(*, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
 
 OCCUPANCY_SENSORS: Incomplete
 

homeassistant-stubs/components/baf/entity.pyi

@@ -1,8 +1,8 @@
 from _typeshed import Incomplete
 from aiobafi6 import Device as Device
 from homeassistant.core import callback as callback
-from homeassistant.helpers.device_registry import format_mac as format_mac
-from homeassistant.helpers.entity import DeviceInfo as DeviceInfo, Entity as Entity
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo, format_mac as format_mac
+from homeassistant.helpers.entity import Entity as Entity
 
 class BAFEntity(Entity):
     _attr_should_poll: bool

homeassistant-stubs/components/baf/light.pyi

@@ -13,6 +13,7 @@ from typing import Any
 async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFLight(BAFEntity, LightEntity):
+    _attr_name: Incomplete
     _attr_is_on: Incomplete
     _attr_brightness: Incomplete
     def _async_update_attrs(self) -> None: ...
@@ -20,10 +21,8 @@ class BAFLight(BAFEntity, LightEntity):
     async def async_turn_off(self, **kwargs: Any) -> None: ...
 
 class BAFFanLight(BAFLight):
-    _attr_name: Incomplete
     _attr_supported_color_modes: Incomplete
     _attr_color_mode: Incomplete
-    def __init__(self, device: Device) -> None: ...
 
 class BAFStandaloneLight(BAFLight):
     _attr_supported_color_modes: Incomplete

homeassistant-stubs/components/baf/models.pyi

@@ -5,7 +5,6 @@ class BAFData:
     device: Device
     run_future: asyncio.Future
     def __init__(self, device, run_future) -> None: ...
-    def __mypy-replace(*, device, run_future) -> None: ...
 
 class BAFDiscovery:
     ip_address: str
@@ -13,4 +12,3 @@ class BAFDiscovery:
     uuid: str
     model: str
     def __init__(self, ip_address, name, uuid, model) -> None: ...
-    def __mypy-replace(*, ip_address, name, uuid, model) -> None: ...

homeassistant-stubs/components/baf/number.pyi

@@ -13,11 +13,9 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 class BAFNumberDescriptionMixin:
     value_fn: Callable[[Device], int | None]
     def __init__(self, value_fn) -> None: ...
-    def __mypy-replace(*, value_fn) -> None: ...
 
 class BAFNumberDescription(NumberEntityDescription, BAFNumberDescriptionMixin):
     def __init__(self, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step) -> None: ...
-    def __mypy-replace(*, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step) -> None: ...
 
 AUTO_COMFORT_NUMBER_DESCRIPTIONS: Incomplete
 FAN_NUMBER_DESCRIPTIONS: Incomplete

homeassistant-stubs/components/baf/sensor.pyi

@@ -13,11 +13,9 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 class BAFSensorDescriptionMixin:
     value_fn: Callable[[Device], int | float | str | None]
     def __init__(self, value_fn) -> None: ...
-    def __mypy-replace(*, value_fn) -> None: ...
 
 class BAFSensorDescription(SensorEntityDescription, BAFSensorDescriptionMixin):
     def __init__(self, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
-    def __mypy-replace(*, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
 
 AUTO_COMFORT_SENSORS: Incomplete
 DEFINED_ONLY_SENSORS: Incomplete

homeassistant-stubs/components/baf/switch.pyi

@@ -14,11 +14,9 @@ from typing import Any
 class BAFSwitchDescriptionMixin:
     value_fn: Callable[[Device], bool | None]
     def __init__(self, value_fn) -> None: ...
-    def __mypy-replace(*, value_fn) -> None: ...
 
 class BAFSwitchDescription(SwitchEntityDescription, BAFSwitchDescriptionMixin):
     def __init__(self, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
-    def __mypy-replace(*, value_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
 
 BASE_SWITCHES: Incomplete
 AUTO_COMFORT_SWITCHES: Incomplete

homeassistant-stubs/components/bayesian/helpers.pyi

@@ -19,4 +19,3 @@ class Observation:
     @property
     def template(self) -> str | None: ...
     def __init__(self, entity_id, platform, prob_given_true, prob_given_false, to_state, above, below, value_template, observed, id) -> None: ...
-    def __mypy-replace(*, entity_id, platform, prob_given_true, prob_given_false, to_state, above, below, value_template, observed, id) -> None: ...

homeassistant-stubs/components/binary_sensor/__init__.pyi

@@ -82,7 +82,6 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: .
 class BinarySensorEntityDescription(EntityDescription):
     device_class: BinarySensorDeviceClass | None
     def __init__(self, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
-    def __mypy-replace(*, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
 
 class BinarySensorEntity(Entity):
     entity_description: BinarySensorEntityDescription

homeassistant-stubs/components/bluetooth/base_scanner.pyi

@@ -23,7 +23,6 @@ class BluetoothScannerDevice:
     ble_device: BLEDevice
     advertisement: AdvertisementData
     def __init__(self, scanner, ble_device, advertisement) -> None: ...
-    def __mypy-replace(*, scanner, ble_device, advertisement) -> None: ...
 
 class BaseHaScanner(ABC, metaclass=abc.ABCMeta):
     __slots__: Incomplete

homeassistant-stubs/components/bluetooth/match.pyi

@@ -33,7 +33,6 @@ class IntegrationMatchHistory:
     service_data: set[str]
     service_uuids: set[str]
     def __init__(self, manufacturer_data, service_data, service_uuids) -> None: ...
-    def __mypy-replace(*, manufacturer_data, service_data, service_uuids) -> None: ...
 
 def seen_all_fields(previous_match: IntegrationMatchHistory, advertisement_data: AdvertisementData) -> bool: ...
 

homeassistant-stubs/components/bluetooth/models.pyi

@@ -15,7 +15,6 @@ class HaBluetoothConnector:
     source: str
     can_connect: Callable[[], bool]
     def __init__(self, client, source, can_connect) -> None: ...
-    def __mypy-replace(*, client, source, can_connect) -> None: ...
 
 class BluetoothScanningMode(Enum):
     PASSIVE: str

homeassistant-stubs/components/bluetooth/passive_update_processor.pyi

@@ -1,57 +1,96 @@
 import logging
-from . import BluetoothChange as BluetoothChange, BluetoothScanningMode as BluetoothScanningMode, BluetoothServiceInfoBleak as BluetoothServiceInfoBleak
 from .const import DOMAIN as DOMAIN
+from .models import BluetoothChange as BluetoothChange, BluetoothScanningMode as BluetoothScanningMode, BluetoothServiceInfoBleak as BluetoothServiceInfoBleak
 from .update_coordinator import BasePassiveBluetoothCoordinator as BasePassiveBluetoothCoordinator
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Mapping
-from homeassistant.const import ATTR_IDENTIFIERS as ATTR_IDENTIFIERS, ATTR_NAME as ATTR_NAME
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity import DeviceInfo as DeviceInfo, Entity as Entity, EntityDescription as EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from typing import Any, Generic, TypeVar
+from collections.abc import Callable as Callable
+from homeassistant import config_entries as config_entries
+from homeassistant.const import ATTR_IDENTIFIERS as ATTR_IDENTIFIERS, ATTR_NAME as ATTR_NAME, CONF_ENTITY_CATEGORY as CONF_ENTITY_CATEGORY, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, EntityCategory as EntityCategory
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity import Entity as Entity, EntityDescription as EntityDescription
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback, async_get_current_platform as async_get_current_platform
+from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
+from homeassistant.helpers.storage import Store as Store
+from homeassistant.util.enum import try_parse_enum as try_parse_enum
+from typing import Any, Generic, TypeVar, TypedDict
+
+STORAGE_KEY: str
+STORAGE_VERSION: int
+STORAGE_SAVE_INTERVAL: Incomplete
+PASSIVE_UPDATE_PROCESSOR: str
+_T = TypeVar('_T')
 
 class PassiveBluetoothEntityKey:
     key: str
     device_id: str | None
+    def to_string(self) -> str: ...
+    @classmethod
+    def from_string(cls, key: str) -> PassiveBluetoothEntityKey: ...
     def __init__(self, key, device_id) -> None: ...
-    def __mypy-replace(*, key, device_id) -> None: ...
-_T = TypeVar('_T')
+
+class PassiveBluetoothProcessorData:
+    coordinators: set[PassiveBluetoothProcessorCoordinator]
+    all_restore_data: dict[str, dict[str, RestoredPassiveBluetoothDataUpdate]]
+    def __init__(self, coordinators, all_restore_data) -> None: ...
+
+class RestoredPassiveBluetoothDataUpdate(TypedDict):
+    devices: dict[str, DeviceInfo]
+    entity_descriptions: dict[str, dict[str, Any]]
+    entity_names: dict[str, str | None]
+    entity_data: dict[str, Any]
+
+cached_fields: Incomplete
+
+def deserialize_entity_description(descriptions_class: type[EntityDescription], data: dict[str, Any]) -> EntityDescription: ...
+def serialize_entity_description(description: EntityDescription) -> dict[str, Any]: ...
 
 class PassiveBluetoothDataUpdate(Generic[_T]):
     devices: dict[str | None, DeviceInfo]
-    entity_descriptions: Mapping[PassiveBluetoothEntityKey, EntityDescription]
-    entity_names: Mapping[PassiveBluetoothEntityKey, str | None]
-    entity_data: Mapping[PassiveBluetoothEntityKey, _T]
+    entity_descriptions: dict[PassiveBluetoothEntityKey, EntityDescription]
+    entity_names: dict[PassiveBluetoothEntityKey, str | None]
+    entity_data: dict[PassiveBluetoothEntityKey, _T]
+    def update(self, new_data: PassiveBluetoothDataUpdate[_T]) -> None: ...
+    def async_get_restore_data(self) -> RestoredPassiveBluetoothDataUpdate: ...
+    def async_set_restore_data(self, restore_data: RestoredPassiveBluetoothDataUpdate, entity_description_class: type[EntityDescription]) -> None: ...
     def __init__(self, devices, entity_descriptions, entity_names, entity_data) -> None: ...
-    def __mypy-replace(*, devices, entity_descriptions, entity_names, entity_data) -> None: ...
+
+def async_register_coordinator_for_restore(hass: HomeAssistant, coordinator: PassiveBluetoothProcessorCoordinator) -> CALLBACK_TYPE: ...
+async def async_setup(hass: HomeAssistant) -> None: ...
 
 class PassiveBluetoothProcessorCoordinator(BasePassiveBluetoothCoordinator, Generic[_T]):
     _processors: Incomplete
     _update_method: Incomplete
     last_update_success: bool
+    restore_data: Incomplete
+    restore_key: Incomplete
     def __init__(self, hass: HomeAssistant, logger: logging.Logger, address: str, mode: BluetoothScanningMode, update_method: Callable[[BluetoothServiceInfoBleak], _T], connectable: bool = ...) -> None: ...
     @property
     def available(self) -> bool: ...
-    def async_register_processor(self, processor: PassiveBluetoothDataProcessor) -> Callable[[], None]: ...
+    def async_get_restore_data(self) -> dict[str, RestoredPassiveBluetoothDataUpdate]: ...
+    def async_register_processor(self, processor: PassiveBluetoothDataProcessor, entity_description_class: type[EntityDescription] | None = ...) -> Callable[[], None]: ...
     def _async_handle_unavailable(self, service_info: BluetoothServiceInfoBleak) -> None: ...
     def _async_handle_bluetooth_event(self, service_info: BluetoothServiceInfoBleak, change: BluetoothChange) -> None: ...
 _PassiveBluetoothDataProcessorT = TypeVar('_PassiveBluetoothDataProcessorT', bound='PassiveBluetoothDataProcessor[Any]')
 
 class PassiveBluetoothDataProcessor(Generic[_T]):
     coordinator: PassiveBluetoothProcessorCoordinator
+    data: PassiveBluetoothDataUpdate[_T]
+    entity_names: dict[PassiveBluetoothEntityKey, str | None]
+    entity_data: dict[PassiveBluetoothEntityKey, _T]
+    entity_descriptions: dict[PassiveBluetoothEntityKey, EntityDescription]
+    devices: dict[str | None, DeviceInfo]
+    restore_key: str | None
     _listeners: Incomplete
     _entity_key_listeners: Incomplete
     update_method: Incomplete
-    entity_names: Incomplete
-    entity_data: Incomplete
-    entity_descriptions: Incomplete
-    devices: Incomplete
     last_update_success: bool
-    def __init__(self, update_method: Callable[[_T], PassiveBluetoothDataUpdate[_T]]) -> None: ...
+    def __init__(self, update_method: Callable[[_T], PassiveBluetoothDataUpdate[_T]], restore_key: str | None = ...) -> None: ...
+    def async_register_coordinator(self, coordinator: PassiveBluetoothProcessorCoordinator, entity_description_class: type[EntityDescription] | None) -> None: ...
     @property
     def available(self) -> bool: ...
     def async_handle_unavailable(self) -> None: ...
-    def async_add_entities_listener(self, entity_class: type[PassiveBluetoothProcessorEntity], async_add_entites: AddEntitiesCallback) -> Callable[[], None]: ...
+    def async_add_entities_listener(self, entity_class: type[PassiveBluetoothProcessorEntity], async_add_entities: AddEntitiesCallback) -> Callable[[], None]: ...
     def async_add_listener(self, update_callback: Callable[[PassiveBluetoothDataUpdate[_T] | None], None]) -> Callable[[], None]: ...
     def async_add_entity_key_listener(self, update_callback: Callable[[PassiveBluetoothDataUpdate[_T] | None], None], entity_key: PassiveBluetoothEntityKey) -> Callable[[], None]: ...
     def async_update_listeners(self, data: PassiveBluetoothDataUpdate[_T] | None) -> None: ...