homeassistant-stubs 2024.10.4__py3-none-any.whl → 2024.11.0b0__py3-none-any.whl

homeassistant-stubs/components/mqtt/discovery.pyi

@@ -1,16 +1,17 @@
 from .abbreviations import ABBREVIATIONS as ABBREVIATIONS, DEVICE_ABBREVIATIONS as DEVICE_ABBREVIATIONS, ORIGIN_ABBREVIATIONS as ORIGIN_ABBREVIATIONS
 from .client import async_subscribe_internal as async_subscribe_internal
-from .const import ATTR_DISCOVERY_HASH as ATTR_DISCOVERY_HASH, ATTR_DISCOVERY_PAYLOAD as ATTR_DISCOVERY_PAYLOAD, ATTR_DISCOVERY_TOPIC as ATTR_DISCOVERY_TOPIC, CONF_AVAILABILITY as CONF_AVAILABILITY, CONF_ORIGIN as CONF_ORIGIN, CONF_TOPIC as CONF_TOPIC, DOMAIN as DOMAIN, SUPPORTED_COMPONENTS as SUPPORTED_COMPONENTS
-from .models import DATA_MQTT as DATA_MQTT, MqttOriginInfo as MqttOriginInfo, ReceiveMessage as ReceiveMessage
-from .schemas import MQTT_ORIGIN_INFO_SCHEMA as MQTT_ORIGIN_INFO_SCHEMA
+from .const import ATTR_DISCOVERY_HASH as ATTR_DISCOVERY_HASH, ATTR_DISCOVERY_PAYLOAD as ATTR_DISCOVERY_PAYLOAD, ATTR_DISCOVERY_TOPIC as ATTR_DISCOVERY_TOPIC, CONF_AVAILABILITY as CONF_AVAILABILITY, CONF_COMPONENTS as CONF_COMPONENTS, CONF_ORIGIN as CONF_ORIGIN, CONF_TOPIC as CONF_TOPIC, DOMAIN as DOMAIN, SUPPORTED_COMPONENTS as SUPPORTED_COMPONENTS
+from .models import DATA_MQTT as DATA_MQTT, MqttComponentConfig as MqttComponentConfig, MqttOriginInfo as MqttOriginInfo, ReceiveMessage as ReceiveMessage
+from .schemas import DEVICE_DISCOVERY_SCHEMA as DEVICE_DISCOVERY_SCHEMA, MQTT_ORIGIN_INFO_SCHEMA as MQTT_ORIGIN_INFO_SCHEMA, SHARED_OPTIONS as SHARED_OPTIONS
 from .util import async_forward_entry_setup_and_setup_discovery as async_forward_entry_setup_and_setup_discovery
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, SOURCE_MQTT as SOURCE_MQTT, signal_discovered_config_entry_removed as signal_discovered_config_entry_removed
 from homeassistant.const import CONF_DEVICE as CONF_DEVICE, CONF_PLATFORM as CONF_PLATFORM
 from homeassistant.core import HassJobType as HassJobType, HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.data_entry_flow import FlowResultType as FlowResultType
+from homeassistant.helpers import discovery_flow as discovery_flow
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
-from homeassistant.helpers.service_info.mqtt import MqttServiceInfo as MqttServiceInfo
+from homeassistant.helpers.service_info.mqtt import MqttServiceInfo as MqttServiceInfo, ReceivePayloadType as ReceivePayloadType
 from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
 from homeassistant.loader import async_get_mqtt as async_get_mqtt
 from homeassistant.util.json import json_loads_object as json_loads_object
@@ -26,16 +27,32 @@ MQTT_DISCOVERY_UPDATED: SignalTypeFormat[MQTTDiscoveryPayload]
 MQTT_DISCOVERY_NEW: SignalTypeFormat[MQTTDiscoveryPayload]
 MQTT_DISCOVERY_DONE: SignalTypeFormat[Any]
 TOPIC_BASE: str
+CONF_MIGRATE_DISCOVERY: str
+MIGRATE_DISCOVERY_SCHEMA: Incomplete
 
 class MQTTDiscoveryPayload(dict[str, Any]):
+    device_discovery: bool
+    migrate_discovery: bool
     discovery_data: DiscoveryInfoType
 
+@dataclass(frozen=True)
+class MQTTIntegrationDiscoveryConfig:
+    integration: str
+    msg: ReceiveMessage
+    def __init__(self, integration, msg) -> None: ...
+
+def _async_process_discovery_migration(payload: MQTTDiscoveryPayload) -> bool: ...
 def clear_discovery_hash(hass: HomeAssistant, discovery_hash: tuple[str, str]) -> None: ...
 def set_discovery_hash(hass: HomeAssistant, discovery_hash: tuple[str, str]) -> None: ...
+def get_origin_log_string(discovery_payload: MQTTDiscoveryPayload, *, include_url: bool) -> str: ...
+def get_origin_support_url(discovery_payload: MQTTDiscoveryPayload) -> str | None: ...
 def async_log_discovery_origin_info(message: str, discovery_payload: MQTTDiscoveryPayload, level: int = ...) -> None: ...
-def _replace_abbreviations(payload: Any | dict[str, Any], abbreviations: dict[str, str], abbreviations_set: set[str]) -> None: ...
-def _replace_all_abbreviations(discovery_payload: Any | dict[str, Any]) -> None: ...
-def _replace_topic_base(discovery_payload: dict[str, Any]) -> None: ...
+def _replace_abbreviations(payload: dict[str, Any] | str, abbreviations: dict[str, str], abbreviations_set: set[str]) -> None: ...
+def _replace_all_abbreviations(discovery_payload: dict[str, Any], component_only: bool = False) -> None: ...
+def _replace_topic_base(discovery_payload: MQTTDiscoveryPayload) -> None: ...
+def _generate_device_config(hass: HomeAssistant, object_id: str, node_id: str | None, migrate_discovery: bool = False) -> MQTTDiscoveryPayload: ...
+def _parse_device_payload(hass: HomeAssistant, payload: ReceivePayloadType, object_id: str, node_id: str | None) -> MQTTDiscoveryPayload: ...
 def _valid_origin_info(discovery_payload: MQTTDiscoveryPayload) -> bool: ...
+def _merge_common_device_options(component_config: MQTTDiscoveryPayload, device_config: dict[str, Any]) -> None: ...
 async def async_start(hass: HomeAssistant, discovery_topic: str, config_entry: ConfigEntry) -> None: ...
 async def async_stop(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/mqtt/entity.pyi

@@ -2,9 +2,9 @@ import abc
 import voluptuous as vol
 from . import debug_info as debug_info, subscription as subscription
 from .client import async_publish as async_publish
-from .const import ATTR_DISCOVERY_HASH as ATTR_DISCOVERY_HASH, ATTR_DISCOVERY_PAYLOAD as ATTR_DISCOVERY_PAYLOAD, ATTR_DISCOVERY_TOPIC as ATTR_DISCOVERY_TOPIC, AVAILABILITY_ALL as AVAILABILITY_ALL, AVAILABILITY_ANY as AVAILABILITY_ANY, CONF_AVAILABILITY as CONF_AVAILABILITY, CONF_AVAILABILITY_MODE as CONF_AVAILABILITY_MODE, CONF_AVAILABILITY_TEMPLATE as CONF_AVAILABILITY_TEMPLATE, CONF_AVAILABILITY_TOPIC as CONF_AVAILABILITY_TOPIC, CONF_CONFIGURATION_URL as CONF_CONFIGURATION_URL, CONF_CONNECTIONS as CONF_CONNECTIONS, CONF_ENABLED_BY_DEFAULT as CONF_ENABLED_BY_DEFAULT, CONF_ENCODING as CONF_ENCODING, CONF_HW_VERSION as CONF_HW_VERSION, CONF_IDENTIFIERS as CONF_IDENTIFIERS, CONF_JSON_ATTRS_TEMPLATE as CONF_JSON_ATTRS_TEMPLATE, CONF_JSON_ATTRS_TOPIC as CONF_JSON_ATTRS_TOPIC, CONF_MANUFACTURER as CONF_MANUFACTURER, CONF_OBJECT_ID as CONF_OBJECT_ID, CONF_PAYLOAD_AVAILABLE as CONF_PAYLOAD_AVAILABLE, CONF_PAYLOAD_NOT_AVAILABLE as CONF_PAYLOAD_NOT_AVAILABLE, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_SCHEMA as CONF_SCHEMA, CONF_SERIAL_NUMBER as CONF_SERIAL_NUMBER, CONF_SUGGESTED_AREA as CONF_SUGGESTED_AREA, CONF_SW_VERSION as CONF_SW_VERSION, CONF_TOPIC as CONF_TOPIC, CONF_VIA_DEVICE as CONF_VIA_DEVICE, DEFAULT_ENCODING as DEFAULT_ENCODING, DOMAIN as DOMAIN, MQTT_CONNECTION_STATE as MQTT_CONNECTION_STATE
+from .const import ATTR_DISCOVERY_HASH as ATTR_DISCOVERY_HASH, ATTR_DISCOVERY_PAYLOAD as ATTR_DISCOVERY_PAYLOAD, ATTR_DISCOVERY_TOPIC as ATTR_DISCOVERY_TOPIC, AVAILABILITY_ALL as AVAILABILITY_ALL, AVAILABILITY_ANY as AVAILABILITY_ANY, CONF_AVAILABILITY as CONF_AVAILABILITY, CONF_AVAILABILITY_MODE as CONF_AVAILABILITY_MODE, CONF_AVAILABILITY_TEMPLATE as CONF_AVAILABILITY_TEMPLATE, CONF_AVAILABILITY_TOPIC as CONF_AVAILABILITY_TOPIC, CONF_CONFIGURATION_URL as CONF_CONFIGURATION_URL, CONF_CONNECTIONS as CONF_CONNECTIONS, CONF_ENABLED_BY_DEFAULT as CONF_ENABLED_BY_DEFAULT, CONF_ENCODING as CONF_ENCODING, CONF_ENTITY_PICTURE as CONF_ENTITY_PICTURE, CONF_HW_VERSION as CONF_HW_VERSION, CONF_IDENTIFIERS as CONF_IDENTIFIERS, CONF_JSON_ATTRS_TEMPLATE as CONF_JSON_ATTRS_TEMPLATE, CONF_JSON_ATTRS_TOPIC as CONF_JSON_ATTRS_TOPIC, CONF_MANUFACTURER as CONF_MANUFACTURER, CONF_OBJECT_ID as CONF_OBJECT_ID, CONF_PAYLOAD_AVAILABLE as CONF_PAYLOAD_AVAILABLE, CONF_PAYLOAD_NOT_AVAILABLE as CONF_PAYLOAD_NOT_AVAILABLE, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_SCHEMA as CONF_SCHEMA, CONF_SERIAL_NUMBER as CONF_SERIAL_NUMBER, CONF_SUGGESTED_AREA as CONF_SUGGESTED_AREA, CONF_SW_VERSION as CONF_SW_VERSION, CONF_TOPIC as CONF_TOPIC, CONF_VIA_DEVICE as CONF_VIA_DEVICE, DEFAULT_ENCODING as DEFAULT_ENCODING, DOMAIN as DOMAIN, MQTT_CONNECTION_STATE as MQTT_CONNECTION_STATE
 from .debug_info import log_message as log_message
-from .discovery import MQTTDiscoveryPayload as MQTTDiscoveryPayload, MQTT_DISCOVERY_DONE as MQTT_DISCOVERY_DONE, MQTT_DISCOVERY_NEW as MQTT_DISCOVERY_NEW, MQTT_DISCOVERY_UPDATED as MQTT_DISCOVERY_UPDATED, clear_discovery_hash as clear_discovery_hash, set_discovery_hash as set_discovery_hash
+from .discovery import MQTTDiscoveryPayload as MQTTDiscoveryPayload, MQTT_DISCOVERY_DONE as MQTT_DISCOVERY_DONE, MQTT_DISCOVERY_NEW as MQTT_DISCOVERY_NEW, MQTT_DISCOVERY_UPDATED as MQTT_DISCOVERY_UPDATED, clear_discovery_hash as clear_discovery_hash, get_origin_log_string as get_origin_log_string, get_origin_support_url as get_origin_support_url, set_discovery_hash as set_discovery_hash
 from .models import DATA_MQTT as DATA_MQTT, MessageCallbackType as MessageCallbackType, MqttValueTemplate as MqttValueTemplate, MqttValueTemplateException as MqttValueTemplateException, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage
 from .subscription import EntitySubscription as EntitySubscription, async_prepare_subscribe_topics as async_prepare_subscribe_topics, async_subscribe_topics_internal as async_subscribe_topics_internal, async_unsubscribe_topics as async_unsubscribe_topics
 from .util import mqtt_config_entry_enabled as mqtt_config_entry_enabled
@@ -89,6 +89,7 @@ class MqttDiscoveryDeviceUpdateMixin(ABC, metaclass=abc.ABCMeta):
     _config_entry: Incomplete
     _config_entry_id: Incomplete
     _skip_device_removal: bool
+    _migrate_discovery: Incomplete
     _remove_discovery_updated: Incomplete
     _remove_device_updated: Incomplete
     def __init__(self, hass: HomeAssistant, discovery_data: DiscoveryInfoType, device_id: str | None, config_entry: ConfigEntry, log_name: str) -> None: ...
@@ -107,6 +108,7 @@ class MqttDiscoveryUpdateMixin(Entity):
     _remove_discovery_updated: Incomplete
     _removed_from_hass: bool
     _registry_hooks: Incomplete
+    _migrate_discovery: Incomplete
     def __init__(self, hass: HomeAssistant, discovery_data: DiscoveryInfoType | None, discovery_update: Callable[[MQTTDiscoveryPayload], Coroutine[Any, Any, None]] | None = None) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     async def _async_remove_state_and_registry_entry(self) -> None: ...
@@ -157,6 +159,7 @@ class MqttEntity(MqttAttributesMixin, MqttAvailabilityMixin, MqttDiscoveryUpdate
     _attr_entity_category: Incomplete
     _attr_entity_registry_enabled_default: Incomplete
     _attr_icon: Incomplete
+    _attr_entity_picture: Incomplete
     def _setup_common_attributes_from_config(self, config: ConfigType) -> None: ...
     def _setup_from_config(self, config: ConfigType) -> None: ...
     @abstractmethod

homeassistant-stubs/components/mqtt/models.pyi

@@ -7,6 +7,7 @@ from .discovery import MQTTDiscoveryPayload as MQTTDiscoveryPayload
 from .tag import MQTTTagScanner as MQTTTagScanner
 from _typeshed import Incomplete
 from collections import deque
+from collections.abc import Callable
 from dataclasses import dataclass
 from enum import StrEnum
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_NAME as ATTR_NAME, Platform as Platform
@@ -26,6 +27,7 @@ class PayloadSentinel(StrEnum):
 
 _LOGGER: Incomplete
 ATTR_THIS: str
+type PublishPayloadType = str | bytes | int | float | None
 
 def convert_outgoing_mqtt_payload(payload: PublishPayloadType) -> PublishPayloadType: ...
 
@@ -46,6 +48,7 @@ class ReceiveMessage:
     subscribed_topic: str
     timestamp: float
     def __init__(self, topic, payload, qos, retain, subscribed_topic, timestamp) -> None: ...
+type MessageCallbackType = Callable[[ReceiveMessage], None]
 
 class SubscriptionDebugInfo(TypedDict):
     messages: deque[ReceiveMessage]
@@ -128,5 +131,13 @@ class MqttData:
     tags: dict[str, dict[str, MQTTTagScanner]] = ...
     def __init__(self, client, config, debug_info_entities=..., debug_info_triggers=..., device_triggers=..., data_config_flow_lock=..., discovery_already_discovered=..., discovery_pending_discovered=..., discovery_registry_hooks=..., discovery_unsubscribe=..., integration_unsubscribe=..., last_discovery=..., platforms_loaded=..., reload_dispatchers=..., reload_handlers=..., reload_schema=..., state_write_requests=..., subscriptions_to_restore=..., tags=...) -> None: ...
 
+@dataclass(slots=True)
+class MqttComponentConfig:
+    component: str
+    object_id: str
+    node_id: str | None
+    discovery_payload: MQTTDiscoveryPayload
+    def __init__(self, component, object_id, node_id, discovery_payload) -> None: ...
+
 DATA_MQTT: HassKey[MqttData]
 DATA_MQTT_AVAILABLE: HassKey[asyncio.Future[bool]]

homeassistant-stubs/components/mqtt/schemas.pyi

@@ -1,16 +1,24 @@
-from .const import AVAILABILITY_LATEST as AVAILABILITY_LATEST, AVAILABILITY_MODES as AVAILABILITY_MODES, CONF_AVAILABILITY as CONF_AVAILABILITY, CONF_AVAILABILITY_MODE as CONF_AVAILABILITY_MODE, CONF_AVAILABILITY_TEMPLATE as CONF_AVAILABILITY_TEMPLATE, CONF_AVAILABILITY_TOPIC as CONF_AVAILABILITY_TOPIC, CONF_CONFIGURATION_URL as CONF_CONFIGURATION_URL, CONF_CONNECTIONS as CONF_CONNECTIONS, CONF_DEPRECATED_VIA_HUB as CONF_DEPRECATED_VIA_HUB, CONF_ENABLED_BY_DEFAULT as CONF_ENABLED_BY_DEFAULT, CONF_HW_VERSION as CONF_HW_VERSION, CONF_IDENTIFIERS as CONF_IDENTIFIERS, CONF_JSON_ATTRS_TEMPLATE as CONF_JSON_ATTRS_TEMPLATE, CONF_JSON_ATTRS_TOPIC as CONF_JSON_ATTRS_TOPIC, CONF_MANUFACTURER as CONF_MANUFACTURER, CONF_OBJECT_ID as CONF_OBJECT_ID, CONF_ORIGIN as CONF_ORIGIN, CONF_PAYLOAD_AVAILABLE as CONF_PAYLOAD_AVAILABLE, CONF_PAYLOAD_NOT_AVAILABLE as CONF_PAYLOAD_NOT_AVAILABLE, CONF_SERIAL_NUMBER as CONF_SERIAL_NUMBER, CONF_SUGGESTED_AREA as CONF_SUGGESTED_AREA, CONF_SUPPORT_URL as CONF_SUPPORT_URL, CONF_SW_VERSION as CONF_SW_VERSION, CONF_TOPIC as CONF_TOPIC, CONF_VIA_DEVICE as CONF_VIA_DEVICE, DEFAULT_PAYLOAD_AVAILABLE as DEFAULT_PAYLOAD_AVAILABLE, DEFAULT_PAYLOAD_NOT_AVAILABLE as DEFAULT_PAYLOAD_NOT_AVAILABLE
-from .util import valid_subscribe_topic as valid_subscribe_topic
+from .const import AVAILABILITY_LATEST as AVAILABILITY_LATEST, AVAILABILITY_MODES as AVAILABILITY_MODES, CONF_AVAILABILITY as CONF_AVAILABILITY, CONF_AVAILABILITY_MODE as CONF_AVAILABILITY_MODE, CONF_AVAILABILITY_TEMPLATE as CONF_AVAILABILITY_TEMPLATE, CONF_AVAILABILITY_TOPIC as CONF_AVAILABILITY_TOPIC, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_COMPONENTS as CONF_COMPONENTS, CONF_CONFIGURATION_URL as CONF_CONFIGURATION_URL, CONF_CONNECTIONS as CONF_CONNECTIONS, CONF_DEPRECATED_VIA_HUB as CONF_DEPRECATED_VIA_HUB, CONF_ENABLED_BY_DEFAULT as CONF_ENABLED_BY_DEFAULT, CONF_ENCODING as CONF_ENCODING, CONF_ENTITY_PICTURE as CONF_ENTITY_PICTURE, CONF_HW_VERSION as CONF_HW_VERSION, CONF_IDENTIFIERS as CONF_IDENTIFIERS, CONF_JSON_ATTRS_TEMPLATE as CONF_JSON_ATTRS_TEMPLATE, CONF_JSON_ATTRS_TOPIC as CONF_JSON_ATTRS_TOPIC, CONF_MANUFACTURER as CONF_MANUFACTURER, CONF_OBJECT_ID as CONF_OBJECT_ID, CONF_ORIGIN as CONF_ORIGIN, CONF_PAYLOAD_AVAILABLE as CONF_PAYLOAD_AVAILABLE, CONF_PAYLOAD_NOT_AVAILABLE as CONF_PAYLOAD_NOT_AVAILABLE, CONF_QOS as CONF_QOS, CONF_SERIAL_NUMBER as CONF_SERIAL_NUMBER, CONF_STATE_TOPIC as CONF_STATE_TOPIC, CONF_SUGGESTED_AREA as CONF_SUGGESTED_AREA, CONF_SUPPORT_URL as CONF_SUPPORT_URL, CONF_SW_VERSION as CONF_SW_VERSION, CONF_TOPIC as CONF_TOPIC, CONF_VIA_DEVICE as CONF_VIA_DEVICE, DEFAULT_PAYLOAD_AVAILABLE as DEFAULT_PAYLOAD_AVAILABLE, DEFAULT_PAYLOAD_NOT_AVAILABLE as DEFAULT_PAYLOAD_NOT_AVAILABLE, ENTITY_PLATFORMS as ENTITY_PLATFORMS, SUPPORTED_COMPONENTS as SUPPORTED_COMPONENTS
+from .util import valid_publish_topic as valid_publish_topic, valid_qos_schema as valid_qos_schema, valid_subscribe_topic as valid_subscribe_topic
 from _typeshed import Incomplete
-from homeassistant.const import CONF_DEVICE as CONF_DEVICE, CONF_ENTITY_CATEGORY as CONF_ENTITY_CATEGORY, CONF_ICON as CONF_ICON, CONF_MODEL as CONF_MODEL, CONF_MODEL_ID as CONF_MODEL_ID, CONF_NAME as CONF_NAME, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE
+from homeassistant.const import CONF_DEVICE as CONF_DEVICE, CONF_ENTITY_CATEGORY as CONF_ENTITY_CATEGORY, CONF_ICON as CONF_ICON, CONF_MODEL as CONF_MODEL, CONF_MODEL_ID as CONF_MODEL_ID, CONF_NAME as CONF_NAME, CONF_PLATFORM as CONF_PLATFORM, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE
 from homeassistant.helpers.entity import ENTITY_CATEGORIES_SCHEMA as ENTITY_CATEGORIES_SCHEMA
 from homeassistant.helpers.typing import ConfigType as ConfigType
+from typing import Any
 
-MQTT_AVAILABILITY_SINGLE_SCHEMA: Incomplete
-MQTT_AVAILABILITY_LIST_SCHEMA: Incomplete
-MQTT_AVAILABILITY_SCHEMA: Incomplete
+SHARED_OPTIONS: Incomplete
+MQTT_ORIGIN_INFO_SCHEMA: Incomplete
+_MQTT_AVAILABILITY_SINGLE_SCHEMA: Incomplete
+_MQTT_AVAILABILITY_LIST_SCHEMA: Incomplete
+_MQTT_AVAILABILITY_SCHEMA: Incomplete
 
 def validate_device_has_at_least_one_identifier(value: ConfigType) -> ConfigType: ...
 
 MQTT_ENTITY_DEVICE_INFO_SCHEMA: Incomplete
-MQTT_ORIGIN_INFO_SCHEMA: Incomplete
 MQTT_ENTITY_COMMON_SCHEMA: Incomplete
+_UNIQUE_ID_SCHEMA: Incomplete
+
+def check_unique_id(config: dict[str, Any]) -> dict[str, Any]: ...
+
+_COMPONENT_CONFIG_SCHEMA: Incomplete
+DEVICE_DISCOVERY_SCHEMA: Incomplete

homeassistant-stubs/components/mqtt/update.pyi

@@ -15,11 +15,11 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
 from homeassistant.helpers.typing import ConfigType as ConfigType, VolSchemaType as VolSchemaType
 from homeassistant.util.json import JSON_DECODE_EXCEPTIONS as JSON_DECODE_EXCEPTIONS, json_loads as json_loads
-from typing import Any, TypedDict
+from typing import Any
 
 _LOGGER: Incomplete
 DEFAULT_NAME: str
-CONF_ENTITY_PICTURE: str
+CONF_DISPLAY_PRECISION: str
 CONF_LATEST_VERSION_TEMPLATE: str
 CONF_LATEST_VERSION_TOPIC: str
 CONF_PAYLOAD_INSTALL: str
@@ -28,26 +28,19 @@ CONF_RELEASE_URL: str
 CONF_TITLE: str
 PLATFORM_SCHEMA_MODERN: Incomplete
 DISCOVERY_SCHEMA: Incomplete
-
-class _MqttUpdatePayloadType(TypedDict, total=False):
-    installed_version: str
-    latest_version: str
-    title: str
-    release_summary: str
-    release_url: str
-    entity_picture: str
+MQTT_JSON_UPDATE_SCHEMA: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class MqttUpdate(MqttEntity, UpdateEntity, RestoreEntity):
     _default_name = DEFAULT_NAME
     _entity_id_format: Incomplete
-    _entity_picture: str | None
     @property
     def entity_picture(self) -> str | None: ...
     @staticmethod
     def config_schema() -> VolSchemaType: ...
     _attr_device_class: Incomplete
+    _attr_display_precision: Incomplete
     _attr_release_summary: Incomplete
     _attr_release_url: Incomplete
     _attr_title: Incomplete
@@ -55,6 +48,9 @@ class MqttUpdate(MqttEntity, UpdateEntity, RestoreEntity):
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_installed_version: Incomplete
     _attr_latest_version: Incomplete
+    _attr_entity_picture: Incomplete
+    _attr_update_percentage: Incomplete
+    _attr_in_progress: Incomplete
     def _handle_state_message_received(self, msg: ReceiveMessage) -> None: ...
     def _handle_latest_version_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...

homeassistant-stubs/components/music_assistant/__init__.pyi

@@ -0,0 +1,27 @@
+import asyncio
+from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
+from homeassistant.const import CONF_URL as CONF_URL, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue, async_delete_issue as async_delete_issue
+from music_assistant_client import MusicAssistantClient
+from music_assistant_models.event import MassEvent as MassEvent
+
+type MusicAssistantConfigEntry = ConfigEntry[MusicAssistantEntryData]
+PLATFORMS: Incomplete
+CONNECT_TIMEOUT: int
+LISTEN_READY_TIMEOUT: int
+
+@dataclass
+class MusicAssistantEntryData:
+    mass: MusicAssistantClient
+    listen_task: asyncio.Task
+    def __init__(self, mass, listen_task) -> None: ...
+
+async def async_setup_entry(hass: HomeAssistant, entry: MusicAssistantConfigEntry) -> bool: ...
+async def _client_listen(hass: HomeAssistant, entry: ConfigEntry, mass: MusicAssistantClient, init_ready: asyncio.Event) -> None: ...
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/music_assistant/config_flow.pyi

@@ -0,0 +1,24 @@
+import voluptuous as vol
+from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
+from _typeshed import Incomplete
+from homeassistant.components import zeroconf as zeroconf
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.const import CONF_URL as CONF_URL
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers import aiohttp_client as aiohttp_client
+from music_assistant_models.api import ServerInfoMessage
+from typing import Any
+
+DEFAULT_URL: str
+DEFAULT_TITLE: str
+
+def get_manual_schema(user_input: dict[str, Any]) -> vol.Schema: ...
+async def get_server_info(hass: HomeAssistant, url: str) -> ServerInfoMessage: ...
+
+class MusicAssistantConfigFlow(ConfigFlow, domain=DOMAIN):
+    VERSION: int
+    server_info: Incomplete
+    def __init__(self) -> None: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_zeroconf(self, discovery_info: zeroconf.ZeroconfServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_discovery_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/music_assistant/const.pyi

@@ -0,0 +1,12 @@
+from _typeshed import Incomplete
+
+DOMAIN: str
+DOMAIN_EVENT: Incomplete
+DEFAULT_NAME: str
+ATTR_IS_GROUP: str
+ATTR_GROUP_MEMBERS: str
+ATTR_GROUP_PARENTS: str
+ATTR_MASS_PLAYER_TYPE: str
+ATTR_ACTIVE_QUEUE: str
+ATTR_STREAM_TITLE: str
+LOGGER: Incomplete

homeassistant-stubs/components/music_assistant/entity.pyi

@@ -0,0 +1,24 @@
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity import Entity as Entity
+from music_assistant_client import MusicAssistantClient as MusicAssistantClient
+from music_assistant_models.event import MassEvent as MassEvent
+from music_assistant_models.player import Player as Player
+
+class MusicAssistantEntity(Entity):
+    _attr_has_entity_name: bool
+    _attr_should_poll: bool
+    mass: Incomplete
+    player_id: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, mass: MusicAssistantClient, player_id: str) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    @property
+    def player(self) -> Player: ...
+    @property
+    def unique_id(self) -> str | None: ...
+    @property
+    def available(self) -> bool: ...
+    async def __on_mass_update(self, event: MassEvent) -> None: ...
+    async def async_on_update(self) -> None: ...

homeassistant-stubs/components/music_assistant/media_player.pyi

@@ -0,0 +1,91 @@
+from . import MusicAssistantConfigEntry as MusicAssistantConfigEntry
+from .const import ATTR_ACTIVE_QUEUE as ATTR_ACTIVE_QUEUE, ATTR_MASS_PLAYER_TYPE as ATTR_MASS_PLAYER_TYPE, DOMAIN as DOMAIN
+from .entity import MusicAssistantEntity as MusicAssistantEntity
+from _typeshed import Incomplete
+from collections.abc import Awaitable, Callable as Callable, Coroutine, Mapping
+from homeassistant.components import media_source as media_source
+from homeassistant.components.media_player import ATTR_MEDIA_EXTRA as ATTR_MEDIA_EXTRA, BrowseMedia as BrowseMedia, MediaPlayerDeviceClass as MediaPlayerDeviceClass, MediaPlayerEnqueue as MediaPlayerEnqueue, MediaPlayerEntity as MediaPlayerEntity, MediaPlayerEntityFeature as MediaPlayerEntityFeature, MediaPlayerState as MediaPlayerState, RepeatMode as RepeatMode, async_process_play_media_url as async_process_play_media_url
+from homeassistant.const import STATE_OFF as STATE_OFF
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.util.dt import utc_from_timestamp as utc_from_timestamp
+from music_assistant_client import MusicAssistantClient as MusicAssistantClient
+from music_assistant_models.enums import MediaType, QueueOption
+from music_assistant_models.event import MassEvent as MassEvent
+from music_assistant_models.player import Player as Player
+from music_assistant_models.player_queue import PlayerQueue as PlayerQueue
+from typing import Any
+
+SUPPORTED_FEATURES: Incomplete
+QUEUE_OPTION_MAP: Incomplete
+ATTR_RADIO_MODE: str
+ATTR_MEDIA_ID: str
+ATTR_MEDIA_TYPE: str
+ATTR_ARTIST: str
+ATTR_ALBUM: str
+ATTR_URL: str
+ATTR_USE_PRE_ANNOUNCE: str
+ATTR_ANNOUNCE_VOLUME: str
+ATTR_SOURCE_PLAYER: str
+ATTR_AUTO_PLAY: str
+
+def catch_musicassistant_error[_R, **P](func: Callable[..., Awaitable[_R]]) -> Callable[..., Coroutine[Any, Any, _R | None]]: ...
+async def async_setup_entry(hass: HomeAssistant, entry: MusicAssistantConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class MusicAssistantPlayer(MusicAssistantEntity, MediaPlayerEntity):
+    _attr_name: Incomplete
+    _attr_media_image_remotely_accessible: bool
+    _attr_media_content_type: Incomplete
+    _attr_icon: Incomplete
+    _attr_supported_features: Incomplete
+    _attr_device_class: Incomplete
+    _prev_time: int
+    def __init__(self, mass: MusicAssistantClient, player_id: str) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    @property
+    def active_queue(self) -> PlayerQueue | None: ...
+    @property
+    def extra_state_attributes(self) -> Mapping[str, Any]: ...
+    _attr_state: Incomplete
+    _attr_group_members: Incomplete
+    _attr_volume_level: Incomplete
+    _attr_is_volume_muted: Incomplete
+    async def async_on_update(self) -> None: ...
+    async def async_media_play(self) -> None: ...
+    async def async_media_pause(self) -> None: ...
+    async def async_media_stop(self) -> None: ...
+    async def async_media_next_track(self) -> None: ...
+    async def async_media_previous_track(self) -> None: ...
+    async def async_media_seek(self, position: float) -> None: ...
+    async def async_mute_volume(self, mute: bool) -> None: ...
+    async def async_set_volume_level(self, volume: float) -> None: ...
+    async def async_volume_up(self) -> None: ...
+    async def async_volume_down(self) -> None: ...
+    async def async_turn_on(self) -> None: ...
+    async def async_turn_off(self) -> None: ...
+    async def async_set_shuffle(self, shuffle: bool) -> None: ...
+    async def async_set_repeat(self, repeat: RepeatMode) -> None: ...
+    async def async_clear_playlist(self) -> None: ...
+    async def async_play_media(self, media_type: MediaType | str, media_id: str, enqueue: MediaPlayerEnqueue | None = None, announce: bool | None = None, **kwargs: Any) -> None: ...
+    async def async_join_players(self, group_members: list[str]) -> None: ...
+    async def async_unjoin_player(self) -> None: ...
+    async def _async_handle_play_media(self, media_id: list[str], enqueue: MediaPlayerEnqueue | QueueOption | None = None, radio_mode: bool | None = None, media_type: str | None = None) -> None: ...
+    async def _async_handle_play_announcement(self, url: str, use_pre_announce: bool | None = None, announce_volume: int | None = None) -> None: ...
+    async def async_browse_media(self, media_content_type: MediaType | str | None = None, media_content_id: str | None = None) -> BrowseMedia: ...
+    _attr_media_image_url: Incomplete
+    def _update_media_image_url(self, player: Player, queue: PlayerQueue | None) -> None: ...
+    _attr_media_artist: Incomplete
+    _attr_media_album_artist: Incomplete
+    _attr_media_album_name: Incomplete
+    _attr_media_title: Incomplete
+    _attr_media_content_id: Incomplete
+    _attr_media_duration: Incomplete
+    _attr_media_position: Incomplete
+    _attr_media_position_updated_at: Incomplete
+    _attr_app_id: Incomplete
+    _attr_shuffle: Incomplete
+    _attr_repeat: Incomplete
+    _attr_source: Incomplete
+    def _update_media_attributes(self, player: Player, queue: PlayerQueue | None) -> None: ...
+    def _convert_queueoption_to_media_player_enqueue(self, queue_option: MediaPlayerEnqueue | QueueOption | None) -> QueueOption | None: ...

homeassistant-stubs/components/mysensors/const.pyi

@@ -1,6 +1,6 @@
 from _typeshed import Incomplete
 from homeassistant.const import Platform as Platform
-from typing import Final, TypedDict
+from typing import Final, Literal, TypedDict
 
 ATTR_DEVICES: Final[str]
 ATTR_GATEWAY_ID: Final[str]
@@ -13,6 +13,7 @@ CONF_TOPIC_IN_PREFIX: Final[str]
 CONF_TOPIC_OUT_PREFIX: Final[str]
 CONF_VERSION: Final[str]
 CONF_GATEWAY_TYPE: Final[str]
+type ConfGatewayType = Literal['Serial', 'TCP', 'MQTT']
 CONF_GATEWAY_TYPE_SERIAL: ConfGatewayType
 CONF_GATEWAY_TYPE_TCP: ConfGatewayType
 CONF_GATEWAY_TYPE_MQTT: ConfGatewayType
@@ -39,6 +40,10 @@ class NodeDiscoveryInfo(TypedDict):
     node_id: int
 
 SERVICE_SEND_IR_CODE: Final[str]
+type SensorType = str
+type ValueType = str
+type GatewayId = str
+type DevId = tuple[GatewayId, int, int, int]
 BINARY_SENSOR_TYPES: dict[SensorType, set[ValueType]]
 CLIMATE_TYPES: dict[SensorType, set[ValueType]]
 COVER_TYPES: dict[SensorType, set[ValueType]]

homeassistant-stubs/components/myuplink/__init__.pyi

@@ -9,6 +9,7 @@ from homeassistant.helpers import aiohttp_client as aiohttp_client, config_entry
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
 
 PLATFORMS: list[Platform]
+type MyUplinkConfigEntry = ConfigEntry[MyUplinkDataCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: MyUplinkConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/myuplink/config_flow.pyi

@@ -1,13 +1,12 @@
 import logging
 from .const import DOMAIN as DOMAIN, OAUTH2_SCOPES as OAUTH2_SCOPES
 from collections.abc import Mapping
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlowResult as ConfigFlowResult
+from homeassistant.config_entries import ConfigFlowResult as ConfigFlowResult, SOURCE_REAUTH as SOURCE_REAUTH
 from homeassistant.helpers import config_entry_oauth2_flow as config_entry_oauth2_flow
 from typing import Any
 
 class OAuth2FlowHandler(config_entry_oauth2_flow.AbstractOAuth2FlowHandler, domain=DOMAIN):
     DOMAIN = DOMAIN
-    config_entry_reauth: ConfigEntry | None
     @property
     def logger(self) -> logging.Logger: ...
     @property

homeassistant-stubs/components/myuplink/helpers.pyi

@@ -1,7 +1,13 @@
+from _typeshed import Incomplete
 from homeassistant.components.number import NumberEntityDescription as NumberEntityDescription
 from homeassistant.components.sensor import SensorEntityDescription as SensorEntityDescription
 from homeassistant.const import Platform as Platform
 from myuplink import DevicePoint as DevicePoint
 
 def find_matching_platform(device_point: DevicePoint, description: SensorEntityDescription | NumberEntityDescription | None = None) -> Platform: ...
+
+WEEKDAYS: Incomplete
+PARAMETER_ID_TO_EXCLUDE_F730: Incomplete
+PARAMETER_ID_TO_INCLUDE_SMO20: Incomplete
+
 def skip_entity(model: str, device_point: DevicePoint) -> bool: ...

homeassistant-stubs/components/nam/__init__.pyi

@@ -9,6 +9,7 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession as asyn
 
 _LOGGER: Incomplete
 PLATFORMS: Incomplete
+type NAMConfigEntry = ConfigEntry[NAMDataUpdateCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: NAMConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: NAMConfigEntry) -> bool: ...

homeassistant-stubs/components/nam/config_flow.pyi

@@ -3,7 +3,7 @@ from _typeshed import Incomplete
 from collections.abc import Mapping
 from dataclasses import dataclass
 from homeassistant.components import zeroconf as zeroconf
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
@@ -24,15 +24,12 @@ async def async_check_credentials(hass: HomeAssistant, host: str, data: dict[str
 
 class NAMFlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    host: Incomplete
-    entry: Incomplete
-    _config: Incomplete
-    def __init__(self) -> None: ...
+    _config: NamConfig
+    host: str
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_credentials(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_zeroconf(self, discovery_info: zeroconf.ZeroconfServiceInfo) -> ConfigFlowResult: ...
     async def async_step_confirm_discovery(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
     async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    async def async_step_reconfigure(self, _: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    async def async_step_reconfigure_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/nanoleaf/__init__.pyi

@@ -10,6 +10,7 @@ from homeassistant.helpers.dispatcher import async_dispatcher_send as async_disp
 
 _LOGGER: Incomplete
 PLATFORMS: Incomplete
+type NanoleafConfigEntry = ConfigEntry[NanoleafCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: NanoleafConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: NanoleafConfigEntry) -> bool: ...

homeassistant-stubs/components/nanoleaf/config_flow.pyi

@@ -3,7 +3,7 @@ from _typeshed import Incomplete
 from aionanoleaf import Nanoleaf
 from collections.abc import Mapping
 from homeassistant.components import ssdp as ssdp, zeroconf as zeroconf
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, SOURCE_REAUTH as SOURCE_REAUTH
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_TOKEN as CONF_TOKEN
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.json import save_json as save_json
@@ -15,7 +15,6 @@ CONFIG_FILE: Final[str]
 USER_SCHEMA: Final[Incomplete]
 
 class NanoleafConfigFlow(ConfigFlow, domain=DOMAIN):
-    reauth_entry: ConfigEntry | None
     nanoleaf: Nanoleaf
     discovery_conf: JsonObjectType
     device_id: str

homeassistant-stubs/components/nest/__init__.pyi

@@ -1,6 +1,6 @@
 import abc
 from . import api as api
-from .const import CONF_PROJECT_ID as CONF_PROJECT_ID, CONF_SUBSCRIBER_ID as CONF_SUBSCRIBER_ID, CONF_SUBSCRIBER_ID_IMPORTED as CONF_SUBSCRIBER_ID_IMPORTED, DATA_DEVICE_MANAGER as DATA_DEVICE_MANAGER, DATA_SDM as DATA_SDM, DATA_SUBSCRIBER as DATA_SUBSCRIBER, DOMAIN as DOMAIN
+from .const import CONF_PROJECT_ID as CONF_PROJECT_ID, CONF_SUBSCRIBER_ID as CONF_SUBSCRIBER_ID, CONF_SUBSCRIBER_ID_IMPORTED as CONF_SUBSCRIBER_ID_IMPORTED, CONF_SUBSCRIPTION_NAME as CONF_SUBSCRIPTION_NAME, DATA_DEVICE_MANAGER as DATA_DEVICE_MANAGER, DATA_SDM as DATA_SDM, DATA_SUBSCRIBER as DATA_SUBSCRIBER, DOMAIN as DOMAIN
 from .events import EVENT_NAME_MAP as EVENT_NAME_MAP, NEST_EVENT as NEST_EVENT
 from .media_source import EVENT_MEDIA_API_URL_FORMAT as EVENT_MEDIA_API_URL_FORMAT, EVENT_THUMBNAIL_URL_FORMAT as EVENT_THUMBNAIL_URL_FORMAT, async_get_media_event_store as async_get_media_event_store, async_get_media_source_devices as async_get_media_source_devices, async_get_transcoder as async_get_transcoder
 from _typeshed import Incomplete

homeassistant-stubs/components/nest/api.pyi

@@ -1,7 +1,8 @@
-from .const import API_URL as API_URL, CONF_PROJECT_ID as CONF_PROJECT_ID, CONF_SUBSCRIBER_ID as CONF_SUBSCRIBER_ID, OAUTH2_TOKEN as OAUTH2_TOKEN, SDM_SCOPES as SDM_SCOPES
+from .const import API_URL as API_URL, CONF_PROJECT_ID as CONF_PROJECT_ID, CONF_SUBSCRIBER_ID as CONF_SUBSCRIBER_ID, CONF_SUBSCRIPTION_NAME as CONF_SUBSCRIPTION_NAME, OAUTH2_TOKEN as OAUTH2_TOKEN, SDM_SCOPES as SDM_SCOPES
 from _typeshed import Incomplete
 from aiohttp import ClientSession as ClientSession
 from google.oauth2.credentials import Credentials
+from google_nest_sdm.admin_client import AdminClient
 from google_nest_sdm.auth import AbstractAuth
 from google_nest_sdm.google_nest_subscriber import GoogleNestSubscriber
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -20,9 +21,10 @@ class AsyncConfigEntryAuth(AbstractAuth):
 
 class AccessTokenAuthImpl(AbstractAuth):
     _access_token: Incomplete
-    def __init__(self, websession: ClientSession, access_token: str) -> None: ...
+    def __init__(self, websession: ClientSession, access_token: str, host: str) -> None: ...
     async def async_get_access_token(self) -> str: ...
     async def async_get_creds(self) -> Credentials: ...
 
 async def new_subscriber(hass: HomeAssistant, entry: ConfigEntry) -> GoogleNestSubscriber | None: ...
-def new_subscriber_with_token(hass: HomeAssistant, access_token: str, project_id: str, subscriber_id: str) -> GoogleNestSubscriber: ...
+def new_subscriber_with_token(hass: HomeAssistant, access_token: str, project_id: str, subscription_name: str) -> GoogleNestSubscriber: ...
+def new_pubsub_admin_client(hass: HomeAssistant, access_token: str, cloud_project_id: str) -> AdminClient: ...

homeassistant-stubs/components/nest/camera.pyi

@@ -3,13 +3,13 @@ from .const import DATA_DEVICE_MANAGER as DATA_DEVICE_MANAGER, DOMAIN as DOMAIN
 from .device_info import NestDeviceInfo as NestDeviceInfo
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
-from google_nest_sdm.camera_traits import RtspStream as RtspStream
+from google_nest_sdm.camera_traits import RtspStream as RtspStream, Stream as Stream, WebRtcStream as WebRtcStream
 from google_nest_sdm.device import Device as Device
 from google_nest_sdm.device_manager import DeviceManager as DeviceManager
-from homeassistant.components.camera import Camera as Camera, CameraEntityFeature as CameraEntityFeature, StreamType as StreamType
+from homeassistant.components.camera import Camera as Camera, CameraEntityFeature as CameraEntityFeature, StreamType as StreamType, WebRTCAnswer as WebRTCAnswer, WebRTCClientConfiguration as WebRTCClientConfiguration, WebRTCSendMessage as WebRTCSendMessage
 from homeassistant.components.stream import CONF_EXTRA_PART_WAIT_TIME as CONF_EXTRA_PART_WAIT_TIME
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.event import async_track_point_in_utc_time as async_track_point_in_utc_time
@@ -28,7 +28,8 @@ class NestCamera(Camera):
     _attr_device_info: Incomplete
     _attr_brand: Incomplete
     _attr_model: Incomplete
-    _stream: Incomplete
+    _rtsp_stream: Incomplete
+    _webrtc_sessions: Incomplete
     _create_stream_url_lock: Incomplete
     _stream_refresh_unsub: Incomplete
     _attr_is_streaming: bool
@@ -43,12 +44,17 @@ class NestCamera(Camera):
     @property
     def available(self) -> bool: ...
     async def stream_source(self) -> str | None: ...
+    def _all_streams(self) -> list[Stream]: ...
     def _schedule_stream_refresh(self) -> None: ...
+    async def _handle_stream_refresh(self, _: datetime.datetime) -> None: ...
     stream: Incomplete
-    async def _handle_stream_refresh(self, now: datetime.datetime) -> None: ...
+    async def _handle_rtsp_stream_refresh(self) -> None: ...
+    async def _handle_webrtc_stream_refresh(self) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     async def async_camera_image(self, width: int | None = None, height: int | None = None) -> bytes | None: ...
     @classmethod
     def placeholder_image(cls) -> bytes: ...
-    async def async_handle_web_rtc_offer(self, offer_sdp: str) -> str | None: ...
+    async def async_handle_async_webrtc_offer(self, offer_sdp: str, session_id: str, send_message: WebRTCSendMessage) -> None: ...
+    def close_webrtc_session(self, session_id: str) -> None: ...
+    def _async_get_webrtc_client_configuration(self) -> WebRTCClientConfiguration: ...