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

homeassistant-stubs/components/media_source/__init__.pyi

@@ -5,9 +5,13 @@ from collections.abc import Callable
 from homeassistant.components.media_player import BrowseMedia
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.typing import UndefinedType
+from typing import Protocol
 
 __all__ = ['DOMAIN', 'is_media_source_id', 'generate_media_source_id', 'async_browse_media', 'async_resolve_media', 'BrowseMediaSource', 'PlayMedia', 'MediaSourceItem', 'Unresolvable', 'MediaSource', 'MediaSourceError', 'MEDIA_CLASS_MAP', 'MEDIA_MIME_TYPES']
 
+class MediaSourceProtocol(Protocol):
+    async def async_get_media_source(self, hass: HomeAssistant) -> MediaSource: ...
+
 def is_media_source_id(media_content_id: str) -> bool: ...
 def generate_media_source_id(domain: str, identifier: str) -> str: ...
 async def async_browse_media(hass: HomeAssistant, media_content_id: str | None, *, content_filter: Callable[[BrowseMedia], bool] | None = None) -> BrowseMediaSource: ...

homeassistant-stubs/components/metoffice/weather.pyi

@@ -15,7 +15,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 def _build_forecast_data(timestep: Timestep) -> Forecast: ...
 def _calculate_unique_id(coordinates: str, use_3hourly: bool) -> str: ...
 
-class MetOfficeWeather(CoordinatorWeatherEntity[TimestampDataUpdateCoordinator[MetOfficeData], TimestampDataUpdateCoordinator[MetOfficeData], TimestampDataUpdateCoordinator[MetOfficeData], TimestampDataUpdateCoordinator[MetOfficeData]]):
+class MetOfficeWeather(CoordinatorWeatherEntity[TimestampDataUpdateCoordinator[MetOfficeData], TimestampDataUpdateCoordinator[MetOfficeData]]):
     _attr_attribution = ATTRIBUTION
     _attr_has_entity_name: bool
     _attr_native_temperature_unit: Incomplete

homeassistant-stubs/components/mikrotik/__init__.pyi

@@ -1,6 +1,6 @@
 from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, DOMAIN as DOMAIN
+from .coordinator import MikrotikDataUpdateCoordinator as MikrotikDataUpdateCoordinator, get_api as get_api
 from .errors import CannotConnect as CannotConnect, LoginError as LoginError
-from .hub import MikrotikDataUpdateCoordinator as MikrotikDataUpdateCoordinator, get_api as get_api
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform

homeassistant-stubs/components/mikrotik/config_flow.pyi

@@ -1,6 +1,6 @@
 from .const import CONF_ARP_PING as CONF_ARP_PING, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_FORCE_DHCP as CONF_FORCE_DHCP, DEFAULT_API_PORT as DEFAULT_API_PORT, DEFAULT_DETECTION_TIME as DEFAULT_DETECTION_TIME, DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN
+from .coordinator import get_api as get_api
 from .errors import CannotConnect as CannotConnect, LoginError as LoginError
-from .hub import get_api as get_api
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow

homeassistant-stubs/components/mikrotik/device_tracker.pyi

@@ -1,5 +1,5 @@
 from .const import DOMAIN as DOMAIN
-from .hub import Device as Device, MikrotikDataUpdateCoordinator as MikrotikDataUpdateCoordinator
+from .coordinator import Device as Device, MikrotikDataUpdateCoordinator as MikrotikDataUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.components.device_tracker import ScannerEntity as ScannerEntity, SourceType as SourceType
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/monzo/__init__.pyi

@@ -0,0 +1,13 @@
+from .api import AuthenticatedMonzoAPI as AuthenticatedMonzoAPI
+from .const import DOMAIN as DOMAIN
+from .coordinator import MonzoCoordinator as MonzoCoordinator
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.config_entry_oauth2_flow import OAuth2Session as OAuth2Session, async_get_config_entry_implementation as async_get_config_entry_implementation
+
+PLATFORMS: list[Platform]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/monzo/api.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aiohttp import ClientSession as ClientSession
+from homeassistant.helpers import config_entry_oauth2_flow as config_entry_oauth2_flow
+from monzopy import AbstractMonzoApi
+
+class AuthenticatedMonzoAPI(AbstractMonzoApi):
+    _oauth_session: Incomplete
+    def __init__(self, websession: ClientSession, oauth_session: config_entry_oauth2_flow.OAuth2Session) -> None: ...
+    async def async_get_access_token(self) -> str: ...

homeassistant-stubs/components/monzo/application_credentials.pyi

@@ -0,0 +1,7 @@
+from homeassistant.components.application_credentials import AuthorizationServer as AuthorizationServer
+from homeassistant.core import HomeAssistant as HomeAssistant
+
+OAUTH2_AUTHORIZE: str
+OAUTH2_TOKEN: str
+
+async def async_get_authorization_server(hass: HomeAssistant) -> AuthorizationServer: ...

homeassistant-stubs/components/monzo/config_flow.pyi

@@ -0,0 +1,14 @@
+import logging
+from .const import DOMAIN as DOMAIN
+from homeassistant.config_entries import ConfigFlowResult as ConfigFlowResult
+from homeassistant.const import CONF_TOKEN as CONF_TOKEN
+from homeassistant.helpers import config_entry_oauth2_flow as config_entry_oauth2_flow
+from typing import Any
+
+class MonzoFlowHandler(config_entry_oauth2_flow.AbstractOAuth2FlowHandler, domain=DOMAIN):
+    DOMAIN = DOMAIN
+    oauth_data: dict[str, Any]
+    @property
+    def logger(self) -> logging.Logger: ...
+    async def async_step_await_approval_confirmation(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_oauth_create_entry(self, data: dict[str, Any]) -> ConfigFlowResult: ...

homeassistant-stubs/components/monzo/const.pyi

@@ -0,0 +1 @@
+DOMAIN: str

homeassistant-stubs/components/monzo/coordinator.pyi

@@ -0,0 +1,20 @@
+from .api import AuthenticatedMonzoAPI as AuthenticatedMonzoAPI
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+from typing import Any
+
+_LOGGER: Incomplete
+
+@dataclass
+class MonzoData:
+    accounts: list[dict[str, Any]]
+    pots: list[dict[str, Any]]
+    def __init__(self, accounts, pots) -> None: ...
+
+class MonzoCoordinator(DataUpdateCoordinator[MonzoData]):
+    api: Incomplete
+    def __init__(self, hass: HomeAssistant, api: AuthenticatedMonzoAPI) -> None: ...
+    async def _async_update_data(self) -> MonzoData: ...

homeassistant-stubs/components/monzo/entity.pyi

@@ -0,0 +1,17 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import MonzoCoordinator as MonzoCoordinator, MonzoData as MonzoData
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+from typing import Any
+
+class MonzoBaseEntity(CoordinatorEntity[MonzoCoordinator]):
+    _attr_attribution: str
+    _attr_has_entity_name: bool
+    index: Incomplete
+    _data_accessor: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, coordinator: MonzoCoordinator, index: int, device_model: str, data_accessor: Callable[[MonzoData], list[dict[str, Any]]]) -> None: ...
+    @property
+    def data(self) -> dict[str, Any]: ...

homeassistant-stubs/components/monzo/sensor.pyi

@@ -0,0 +1,31 @@
+from . import MonzoCoordinator as MonzoCoordinator
+from .const import DOMAIN as DOMAIN
+from .coordinator import MonzoData as MonzoData
+from .entity import MonzoBaseEntity as MonzoBaseEntity
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import StateType as StateType
+from typing import Any
+
+@dataclass(frozen=True, kw_only=True)
+class MonzoSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[dict[str, Any]], StateType]
+    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, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+
+ACCOUNT_SENSORS: Incomplete
+POT_SENSORS: Incomplete
+MODEL_POT: str
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class MonzoSensor(MonzoBaseEntity, SensorEntity):
+    entity_description: MonzoSensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: MonzoCoordinator, entity_description: MonzoSensorEntityDescription, index: int, device_model: str, data_accessor: Callable[[MonzoData], list[dict[str, Any]]]) -> None: ...
+    @property
+    def native_value(self) -> StateType: ...

homeassistant-stubs/components/mopeka/sensor.pyi

@@ -8,13 +8,13 @@ from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info as sensor_device_info_to_hass_device_info
-from mopeka_iot_ble import SensorUpdate as SensorUpdate
+from mopeka_iot_ble import SensorUpdate
 
 SENSOR_DESCRIPTIONS: Incomplete
 
 def sensor_update_to_bluetooth_data_update(sensor_update: SensorUpdate) -> PassiveBluetoothDataUpdate: ...
 async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class MopekaBluetoothSensorEntity(PassiveBluetoothProcessorEntity[PassiveBluetoothDataProcessor[float | int | None]], SensorEntity):
+class MopekaBluetoothSensorEntity(PassiveBluetoothProcessorEntity[PassiveBluetoothDataProcessor[float | int | None, SensorUpdate]], SensorEntity):
     @property
     def native_value(self) -> int | float | None: ...

homeassistant-stubs/components/motionmount/entity.pyi

@@ -13,5 +13,8 @@ class MotionMountEntity(Entity):
     _base_unique_id: Incomplete
     _attr_device_info: Incomplete
     def __init__(self, mm: motionmount.MotionMount, config_entry: ConfigEntry) -> None: ...
+    @property
+    def available(self) -> bool: ...
+    def update_name(self) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...

homeassistant-stubs/components/mqtt/__init__.pyi

@@ -1,26 +1,28 @@
 from . import debug_info as debug_info, discovery as discovery
-from .client import MQTT as MQTT, async_publish as async_publish, async_subscribe as async_subscribe, publish as publish, subscribe as subscribe
+from .client import MQTT as MQTT, async_publish as async_publish, async_subscribe as async_subscribe, async_subscribe_internal as async_subscribe_internal, publish as publish, subscribe as subscribe
 from .config import MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA, MQTT_RO_SCHEMA as MQTT_RO_SCHEMA, MQTT_RW_SCHEMA as MQTT_RW_SCHEMA
 from .config_integration import CONFIG_SCHEMA_BASE as CONFIG_SCHEMA_BASE
-from .const import ATTR_PAYLOAD as ATTR_PAYLOAD, ATTR_QOS as ATTR_QOS, ATTR_RETAIN as ATTR_RETAIN, ATTR_TOPIC as ATTR_TOPIC, CONF_BIRTH_MESSAGE as CONF_BIRTH_MESSAGE, CONF_BROKER as CONF_BROKER, CONF_CERTIFICATE as CONF_CERTIFICATE, CONF_CLIENT_CERT as CONF_CLIENT_CERT, CONF_CLIENT_KEY as CONF_CLIENT_KEY, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_DISCOVERY_PREFIX as CONF_DISCOVERY_PREFIX, CONF_KEEPALIVE as CONF_KEEPALIVE, CONF_QOS as CONF_QOS, CONF_STATE_TOPIC as CONF_STATE_TOPIC, CONF_TLS_INSECURE as CONF_TLS_INSECURE, CONF_TOPIC as CONF_TOPIC, CONF_TRANSPORT as CONF_TRANSPORT, CONF_WILL_MESSAGE as CONF_WILL_MESSAGE, CONF_WS_HEADERS as CONF_WS_HEADERS, CONF_WS_PATH as CONF_WS_PATH, DATA_MQTT as DATA_MQTT, DATA_MQTT_AVAILABLE as DATA_MQTT_AVAILABLE, DEFAULT_DISCOVERY as DEFAULT_DISCOVERY, DEFAULT_ENCODING as DEFAULT_ENCODING, DEFAULT_PREFIX as DEFAULT_PREFIX, DEFAULT_QOS as DEFAULT_QOS, DEFAULT_RETAIN as DEFAULT_RETAIN, DOMAIN as DOMAIN, MQTT_CONNECTED as MQTT_CONNECTED, MQTT_DISCONNECTED as MQTT_DISCONNECTED, RELOADABLE_PLATFORMS as RELOADABLE_PLATFORMS, TEMPLATE_ERRORS as TEMPLATE_ERRORS
-from .models import MqttCommandTemplate as MqttCommandTemplate, MqttData as MqttData, MqttValueTemplate as MqttValueTemplate, PayloadSentinel as PayloadSentinel, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
-from .util import async_create_certificate_temp_files as async_create_certificate_temp_files, async_forward_entry_setup_and_setup_discovery as async_forward_entry_setup_and_setup_discovery, async_wait_for_mqtt_client as async_wait_for_mqtt_client, get_mqtt_data as get_mqtt_data, mqtt_config_entry_enabled as mqtt_config_entry_enabled, platforms_from_config as platforms_from_config, valid_publish_topic as valid_publish_topic, valid_qos_schema as valid_qos_schema, valid_subscribe_topic as valid_subscribe_topic
+from .const import ATTR_PAYLOAD as ATTR_PAYLOAD, ATTR_QOS as ATTR_QOS, ATTR_RETAIN as ATTR_RETAIN, ATTR_TOPIC as ATTR_TOPIC, CONF_BIRTH_MESSAGE as CONF_BIRTH_MESSAGE, CONF_BROKER as CONF_BROKER, CONF_CERTIFICATE as CONF_CERTIFICATE, CONF_CLIENT_CERT as CONF_CLIENT_CERT, CONF_CLIENT_KEY as CONF_CLIENT_KEY, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_DISCOVERY_PREFIX as CONF_DISCOVERY_PREFIX, CONF_KEEPALIVE as CONF_KEEPALIVE, CONF_QOS as CONF_QOS, CONF_STATE_TOPIC as CONF_STATE_TOPIC, CONF_TLS_INSECURE as CONF_TLS_INSECURE, CONF_TOPIC as CONF_TOPIC, CONF_TRANSPORT as CONF_TRANSPORT, CONF_WILL_MESSAGE as CONF_WILL_MESSAGE, CONF_WS_HEADERS as CONF_WS_HEADERS, CONF_WS_PATH as CONF_WS_PATH, DEFAULT_DISCOVERY as DEFAULT_DISCOVERY, DEFAULT_ENCODING as DEFAULT_ENCODING, DEFAULT_PREFIX as DEFAULT_PREFIX, DEFAULT_QOS as DEFAULT_QOS, DEFAULT_RETAIN as DEFAULT_RETAIN, DOMAIN as DOMAIN, MQTT_CONNECTION_STATE as MQTT_CONNECTION_STATE, RELOADABLE_PLATFORMS as RELOADABLE_PLATFORMS, TEMPLATE_ERRORS as TEMPLATE_ERRORS
+from .models import DATA_MQTT as DATA_MQTT, DATA_MQTT_AVAILABLE as DATA_MQTT_AVAILABLE, MqttCommandTemplate as MqttCommandTemplate, MqttData as MqttData, MqttValueTemplate as MqttValueTemplate, PayloadSentinel as PayloadSentinel, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
+from .subscription import EntitySubscription as EntitySubscription, async_prepare_subscribe_topics as async_prepare_subscribe_topics, async_subscribe_topics as async_subscribe_topics, async_unsubscribe_topics as async_unsubscribe_topics
+from .util import async_create_certificate_temp_files as async_create_certificate_temp_files, async_forward_entry_setup_and_setup_discovery as async_forward_entry_setup_and_setup_discovery, async_wait_for_mqtt_client as async_wait_for_mqtt_client, mqtt_config_entry_enabled as mqtt_config_entry_enabled, platforms_from_config as platforms_from_config, 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 collections.abc import Callable
 from homeassistant.components import websocket_api as websocket_api
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_CLIENT_ID as CONF_CLIENT_ID, CONF_DISCOVERY as CONF_DISCOVERY, CONF_PASSWORD as CONF_PASSWORD, CONF_PAYLOAD as CONF_PAYLOAD, CONF_PORT as CONF_PORT, CONF_PROTOCOL as CONF_PROTOCOL, CONF_USERNAME as CONF_USERNAME, SERVICE_RELOAD as SERVICE_RELOAD
-from homeassistant.core import HassJob as HassJob, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
+from homeassistant.const import CONF_DISCOVERY as CONF_DISCOVERY, CONF_PAYLOAD as CONF_PAYLOAD, SERVICE_RELOAD as SERVICE_RELOAD
+from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.exceptions import ConfigValidationError as ConfigValidationError, ServiceValidationError as ServiceValidationError, Unauthorized as Unauthorized
 from homeassistant.helpers import template as template
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import async_get_platforms as async_get_platforms
-from homeassistant.helpers.issue_registry import async_delete_issue as async_delete_issue
 from homeassistant.helpers.reload import async_integration_yaml_config as async_integration_yaml_config
 from homeassistant.helpers.service import async_register_admin_service as async_register_admin_service
 from homeassistant.helpers.typing import ConfigType as ConfigType
-from homeassistant.loader import async_get_integration as async_get_integration
+from homeassistant.loader import async_get_integration as async_get_integration, async_get_loaded_integration as async_get_loaded_integration
+from homeassistant.setup import SetupPhases as SetupPhases, async_pause_setup as async_pause_setup
+from homeassistant.util.async_ import create_eager_task as create_eager_task
 from typing import Any
 
 _LOGGER: Incomplete
@@ -32,8 +34,6 @@ MAX_RECONNECT_WAIT: int
 CONNECTION_SUCCESS: str
 CONNECTION_FAILED: str
 CONNECTION_FAILED_RECOVERABLE: str
-CONFIG_ENTRY_CONFIG_KEYS: Incomplete
-REMOVED_OPTIONS: Incomplete
 CONFIG_SCHEMA: Incomplete
 MQTT_PUBLISH_SCHEMA: Incomplete
 

homeassistant-stubs/components/mqtt/alarm_control_panel.pyi

@@ -2,10 +2,10 @@ import homeassistant.components.alarm_control_panel as alarm
 import voluptuous as vol
 from . import subscription as subscription
 from .config import DEFAULT_RETAIN as DEFAULT_RETAIN, MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA
-from .const import CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_ENCODING as CONF_ENCODING, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_STATE_TOPIC as CONF_STATE_TOPIC, CONF_SUPPORTED_FEATURES as CONF_SUPPORTED_FEATURES
-from .debug_info import log_messages as log_messages
-from .mixins import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper, write_state_on_attr_change as write_state_on_attr_change
+from .const import CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_RETAIN as CONF_RETAIN, CONF_STATE_TOPIC as CONF_STATE_TOPIC, CONF_SUPPORTED_FEATURES as CONF_SUPPORTED_FEATURES, PAYLOAD_NONE as PAYLOAD_NONE
+from .mixins import MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
 from .models import MqttCommandTemplate as MqttCommandTemplate, MqttValueTemplate as MqttValueTemplate, ReceiveMessage as ReceiveMessage
+from .schemas import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA
 from .util import valid_publish_topic as valid_publish_topic, valid_subscribe_topic as valid_subscribe_topic
 from _typeshed import Incomplete
 from homeassistant.components.alarm_control_panel import AlarmControlPanelEntityFeature as AlarmControlPanelEntityFeature
@@ -56,7 +56,7 @@ class MqttAlarm(MqttEntity, alarm.AlarmControlPanelEntity):
     _attr_code_arm_required: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_state: Incomplete
-    _sub_state: Incomplete
+    def _state_message_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_alarm_disarm(self, code: str | None = None) -> None: ...

homeassistant-stubs/components/mqtt/binary_sensor.pyi

@@ -1,11 +1,12 @@
 import voluptuous as vol
 from . import subscription as subscription
 from .config import MQTT_RO_SCHEMA as MQTT_RO_SCHEMA
-from .const import CONF_ENCODING as CONF_ENCODING, CONF_QOS as CONF_QOS, CONF_STATE_TOPIC as CONF_STATE_TOPIC, PAYLOAD_NONE as PAYLOAD_NONE
-from .debug_info import log_messages as log_messages
-from .mixins import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA, MqttAvailability as MqttAvailability, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper, write_state_on_attr_change as write_state_on_attr_change
+from .const import CONF_STATE_TOPIC as CONF_STATE_TOPIC, PAYLOAD_NONE as PAYLOAD_NONE
+from .mixins import MqttAvailabilityMixin as MqttAvailabilityMixin, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
 from .models import MqttValueTemplate as MqttValueTemplate, ReceiveMessage as ReceiveMessage
+from .schemas import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA
 from _typeshed import Incomplete
+from datetime import datetime
 from homeassistant.components import binary_sensor as binary_sensor
 from homeassistant.components.binary_sensor import BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -45,7 +46,8 @@ class MqttBinarySensor(MqttEntity, BinarySensorEntity, RestoreEntity):
     _attr_device_class: Incomplete
     _value_template: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
-    _sub_state: Incomplete
+    def _off_delay_listener(self, now: datetime) -> None: ...
+    def _state_message_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     def _value_is_expired(self, *_: Any) -> None: ...

homeassistant-stubs/components/mqtt/button.pyi

@@ -1,15 +1,16 @@
 import voluptuous as vol
 from .config import DEFAULT_RETAIN as DEFAULT_RETAIN, MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA
-from .const import CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_ENCODING as CONF_ENCODING, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN
-from .mixins import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
+from .const import CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_RETAIN as CONF_RETAIN
+from .mixins import MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
 from .models import MqttCommandTemplate as MqttCommandTemplate
+from .schemas import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA
 from .util import valid_publish_topic as valid_publish_topic
 from _typeshed import Incomplete
 from homeassistant.components import button as button
 from homeassistant.components.button import ButtonEntity as ButtonEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_NAME as CONF_NAME
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import ConfigType as ConfigType
 

homeassistant-stubs/components/mqtt/camera.pyi

@@ -1,10 +1,10 @@
 import voluptuous as vol
 from . import subscription as subscription
 from .config import MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA
-from .const import CONF_QOS as CONF_QOS, CONF_TOPIC as CONF_TOPIC
-from .debug_info import log_messages as log_messages
-from .mixins import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
+from .const import CONF_TOPIC as CONF_TOPIC
+from .mixins import MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
 from .models import ReceiveMessage as ReceiveMessage
+from .schemas import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA
 from .util import valid_subscribe_topic as valid_subscribe_topic
 from _typeshed import Incomplete
 from homeassistant.components import camera as camera
@@ -33,7 +33,7 @@ class MqttCamera(MqttEntity, Camera):
     def __init__(self, hass: HomeAssistant, config: ConfigType, config_entry: ConfigEntry, discovery_data: DiscoveryInfoType | None) -> None: ...
     @staticmethod
     def config_schema() -> vol.Schema: ...
-    _sub_state: Incomplete
+    def _image_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_camera_image(self, width: int | None = None, height: int | None = None) -> bytes | None: ...

homeassistant-stubs/components/mqtt/client.pyi

@@ -1,48 +1,56 @@
 import asyncio
 import paho.mqtt.client as mqtt
-from .const import CONF_BIRTH_MESSAGE as CONF_BIRTH_MESSAGE, CONF_BROKER as CONF_BROKER, CONF_CERTIFICATE as CONF_CERTIFICATE, CONF_CLIENT_CERT as CONF_CLIENT_CERT, CONF_CLIENT_KEY as CONF_CLIENT_KEY, CONF_KEEPALIVE as CONF_KEEPALIVE, CONF_TLS_INSECURE as CONF_TLS_INSECURE, CONF_TRANSPORT as CONF_TRANSPORT, CONF_WILL_MESSAGE as CONF_WILL_MESSAGE, CONF_WS_HEADERS as CONF_WS_HEADERS, CONF_WS_PATH as CONF_WS_PATH, DEFAULT_BIRTH as DEFAULT_BIRTH, DEFAULT_ENCODING as DEFAULT_ENCODING, DEFAULT_KEEPALIVE as DEFAULT_KEEPALIVE, DEFAULT_PORT as DEFAULT_PORT, DEFAULT_PROTOCOL as DEFAULT_PROTOCOL, DEFAULT_QOS as DEFAULT_QOS, DEFAULT_TRANSPORT as DEFAULT_TRANSPORT, DEFAULT_WILL as DEFAULT_WILL, DEFAULT_WS_HEADERS as DEFAULT_WS_HEADERS, DEFAULT_WS_PATH as DEFAULT_WS_PATH, DOMAIN as DOMAIN, MQTT_CONNECTED as MQTT_CONNECTED, MQTT_DISCONNECTED as MQTT_DISCONNECTED, PROTOCOL_31 as PROTOCOL_31, PROTOCOL_5 as PROTOCOL_5, TRANSPORT_WEBSOCKETS as TRANSPORT_WEBSOCKETS
-from .models import AsyncMessageCallbackType as AsyncMessageCallbackType, MessageCallbackType as MessageCallbackType, MqttData as MqttData, PublishMessage as PublishMessage, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage
-from .util import get_file_path as get_file_path, get_mqtt_data as get_mqtt_data, mqtt_config_entry_enabled as mqtt_config_entry_enabled
+from .const import CONF_BIRTH_MESSAGE as CONF_BIRTH_MESSAGE, CONF_BROKER as CONF_BROKER, CONF_CERTIFICATE as CONF_CERTIFICATE, CONF_CLIENT_CERT as CONF_CLIENT_CERT, CONF_CLIENT_KEY as CONF_CLIENT_KEY, CONF_KEEPALIVE as CONF_KEEPALIVE, CONF_TLS_INSECURE as CONF_TLS_INSECURE, CONF_TRANSPORT as CONF_TRANSPORT, CONF_WILL_MESSAGE as CONF_WILL_MESSAGE, CONF_WS_HEADERS as CONF_WS_HEADERS, CONF_WS_PATH as CONF_WS_PATH, DEFAULT_BIRTH as DEFAULT_BIRTH, DEFAULT_ENCODING as DEFAULT_ENCODING, DEFAULT_KEEPALIVE as DEFAULT_KEEPALIVE, DEFAULT_PORT as DEFAULT_PORT, DEFAULT_PROTOCOL as DEFAULT_PROTOCOL, DEFAULT_QOS as DEFAULT_QOS, DEFAULT_TRANSPORT as DEFAULT_TRANSPORT, DEFAULT_WILL as DEFAULT_WILL, DEFAULT_WS_HEADERS as DEFAULT_WS_HEADERS, DEFAULT_WS_PATH as DEFAULT_WS_PATH, DOMAIN as DOMAIN, MQTT_CONNECTION_STATE as MQTT_CONNECTION_STATE, PROTOCOL_31 as PROTOCOL_31, PROTOCOL_5 as PROTOCOL_5, TRANSPORT_WEBSOCKETS as TRANSPORT_WEBSOCKETS
+from .models import DATA_MQTT as DATA_MQTT, MessageCallbackType as MessageCallbackType, MqttData as MqttData, PublishMessage as PublishMessage, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage
+from .util import get_file_path as get_file_path, mqtt_config_entry_enabled as mqtt_config_entry_enabled
 from _typeshed import Incomplete
 from collections.abc import AsyncGenerator, Callable as Callable, Coroutine, Iterable
 from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_CLIENT_ID as CONF_CLIENT_ID, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_PROTOCOL as CONF_PROTOCOL, CONF_USERNAME as CONF_USERNAME, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HassJob as HassJob, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HassJob as HassJob, HassJobType as HassJobType, HomeAssistant as HomeAssistant, callback as callback, get_hassjob_callable_job_type as get_hassjob_callable_job_type
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
+from homeassistant.helpers.importlib import async_import_module as async_import_module
 from homeassistant.helpers.start import async_at_started as async_at_started
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.setup import SetupPhases as SetupPhases, async_pause_setup as async_pause_setup
 from homeassistant.util.async_ import create_eager_task as create_eager_task
-from homeassistant.util.logging import catch_log_exception as catch_log_exception
+from homeassistant.util.collection import chunked_or_all as chunked_or_all
+from homeassistant.util.logging import catch_log_exception as catch_log_exception, log_exception as log_exception
 from typing import Any
 
 _LOGGER: Incomplete
 MIN_BUFFER_SIZE: int
-PREFERRED_BUFFER_SIZE: int
+PREFERRED_BUFFER_SIZE: Incomplete
 DISCOVERY_COOLDOWN: int
 INITIAL_SUBSCRIBE_COOLDOWN: float
 SUBSCRIBE_COOLDOWN: float
 UNSUBSCRIBE_COOLDOWN: float
 TIMEOUT_ACK: int
 RECONNECT_INTERVAL_SECONDS: int
+MAX_SUBSCRIBES_PER_CALL: int
+MAX_UNSUBSCRIBES_PER_CALL: int
+MAX_PACKETS_TO_READ: int
 SocketType: Incomplete
 SubscribePayloadType: Incomplete
 
 def publish(hass: HomeAssistant, topic: str, payload: PublishPayloadType, qos: int | None = 0, retain: bool | None = False, encoding: str | None = ...) -> None: ...
 async def async_publish(hass: HomeAssistant, topic: str, payload: PublishPayloadType, qos: int | None = 0, retain: bool | None = False, encoding: str | None = ...) -> None: ...
-async def async_subscribe(hass: HomeAssistant, topic: str, msg_callback: AsyncMessageCallbackType | MessageCallbackType, qos: int = ..., encoding: str | None = ...) -> CALLBACK_TYPE: ...
+async def async_subscribe(hass: HomeAssistant, topic: str, msg_callback: Callable[[ReceiveMessage], Coroutine[Any, Any, None] | None], qos: int = ..., encoding: str | None = ...) -> CALLBACK_TYPE: ...
+def async_subscribe_internal(hass: HomeAssistant, topic: str, msg_callback: Callable[[ReceiveMessage], Coroutine[Any, Any, None] | None], qos: int = ..., encoding: str | None = ..., job_type: HassJobType | None = None) -> CALLBACK_TYPE: ...
 def subscribe(hass: HomeAssistant, topic: str, msg_callback: MessageCallbackType, qos: int = ..., encoding: str = 'utf-8') -> Callable[[], None]: ...
 
-@dataclass(frozen=True)
+@dataclass(slots=True, frozen=True)
 class Subscription:
     topic: str
-    matcher: Any
+    is_simple_match: bool
+    complex_matcher: Callable[[str], bool] | None
     job: HassJob[[ReceiveMessage], Coroutine[Any, Any, None] | None]
     qos: int = ...
     encoding: str | None = ...
-    def __init__(self, topic, matcher, job, qos, encoding) -> None: ...
+    def __init__(self, topic, is_simple_match, complex_matcher, job, qos, encoding) -> None: ...
 
 class MqttClientSetup:
     _client: Incomplete
@@ -50,14 +58,13 @@ class MqttClientSetup:
     @property
     def client(self) -> mqtt.Client: ...
 
-def _is_simple_match(topic: str) -> bool: ...
-
 class EnsureJobAfterCooldown:
     _loop: Incomplete
     _timeout: Incomplete
     _callback: Incomplete
     _task: Incomplete
     _timer: Incomplete
+    _next_execute_time: float
     def __init__(self, timeout: float, callback_job: Callable[[], Coroutine[Any, None, None]]) -> None: ...
     def set_timeout(self, timeout: float) -> None: ...
     async def _async_job(self) -> None: ...
@@ -65,6 +72,7 @@ class EnsureJobAfterCooldown:
     def async_execute(self) -> asyncio.Task: ...
     def _async_cancel_timer(self) -> None: ...
     def async_schedule(self) -> None: ...
+    def _async_timer_reached(self) -> None: ...
     async def async_cleanup(self) -> None: ...
 
 class MQTT:
@@ -83,7 +91,6 @@ class MQTT:
     _cleanup_on_unload: Incomplete
     _connection_lock: Incomplete
     _pending_operations: Incomplete
-    _pending_operations_condition: Incomplete
     _subscribe_debouncer: Incomplete
     _misc_task: Incomplete
     _reconnect_task: Incomplete
@@ -97,12 +104,12 @@ class MQTT:
     def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, conf: ConfigType) -> None: ...
     def _async_ha_started(self, _hass: HomeAssistant) -> None: ...
     async def _async_ha_stop(self, _event: Event) -> None: ...
-    def start(self, mqtt_data: MqttData) -> None: ...
+    async def async_start(self, mqtt_data: MqttData) -> None: ...
     @property
     def subscriptions(self) -> list[Subscription]: ...
     def cleanup(self) -> None: ...
     async def _async_connect_in_executor(self) -> AsyncGenerator[None, None]: ...
-    def init_client(self) -> None: ...
+    async def async_init_client(self) -> None: ...
     async def _misc_loop(self) -> None: ...
     def _async_reader_callback(self, client: mqtt.Client) -> None: ...
     def _async_start_misc_loop(self) -> None: ...
@@ -125,7 +132,9 @@ class MQTT:
     def _async_track_subscription(self, subscription: Subscription) -> None: ...
     def _async_untrack_subscription(self, subscription: Subscription) -> None: ...
     def _async_queue_subscriptions(self, subscriptions: Iterable[tuple[str, int]], queue_only: bool = False) -> None: ...
-    async def async_subscribe(self, topic: str, msg_callback: AsyncMessageCallbackType | MessageCallbackType, qos: int, encoding: str | None = None) -> Callable[[], None]: ...
+    def _exception_message(self, msg_callback: Callable[[ReceiveMessage], Coroutine[Any, Any, None] | None], msg: ReceiveMessage) -> str: ...
+    def async_subscribe(self, topic: str, msg_callback: Callable[[ReceiveMessage], Coroutine[Any, Any, None] | None], qos: int, encoding: str | None = None, job_type: HassJobType | None = None) -> Callable[[], None]: ...
+    def _async_remove(self, subscription: Subscription) -> None: ...
     def _async_unsubscribe(self, topic: str) -> None: ...
     async def _async_perform_subscriptions(self) -> None: ...
     async def _async_perform_unsubscribes(self) -> None: ...
@@ -135,12 +144,11 @@ class MQTT:
     def _matching_subscriptions(self, topic: str) -> list[Subscription]: ...
     def _async_mqtt_on_message(self, _mqttc: mqtt.Client, _userdata: None, msg: mqtt.MQTTMessage) -> None: ...
     def _async_mqtt_on_callback(self, _mqttc: mqtt.Client, _userdata: None, mid: int, _granted_qos_reason: tuple[int, ...] | mqtt.ReasonCodes | None = None, _properties_reason: mqtt.ReasonCodes | None = None) -> None: ...
-    async def _mqtt_handle_mid(self, mid: int) -> None: ...
-    async def _register_mid(self, mid: int) -> None: ...
+    def _async_get_mid_future(self, mid: int) -> asyncio.Future[None]: ...
     def _async_mqtt_on_disconnect(self, _mqttc: mqtt.Client, _userdata: None, result_code: int, properties: mqtt.Properties | None = None) -> None: ...
     def _async_on_disconnect(self, result_code: int) -> None: ...
-    async def _wait_for_mid(self, mid: int) -> None: ...
+    def _async_timeout_mid(self, future: asyncio.Future[None]) -> None: ...
+    async def _async_wait_for_mid_or_raise(self, mid: int, result_code: int) -> None: ...
     async def _discovery_cooldown(self) -> None: ...
 
-def _raise_on_error(result_code: int) -> None: ...
-def _matcher_for_topic(subscription: str) -> Any: ...
+def _matcher_for_topic(subscription: str) -> Callable[[str], bool]: ...

homeassistant-stubs/components/mqtt/climate.pyi

@@ -2,10 +2,10 @@ import abc
 import voluptuous as vol
 from . import subscription as subscription
 from .config import DEFAULT_RETAIN as DEFAULT_RETAIN, MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA
-from .const import CONF_ACTION_TEMPLATE as CONF_ACTION_TEMPLATE, CONF_ACTION_TOPIC as CONF_ACTION_TOPIC, CONF_CURRENT_HUMIDITY_TEMPLATE as CONF_CURRENT_HUMIDITY_TEMPLATE, CONF_CURRENT_HUMIDITY_TOPIC as CONF_CURRENT_HUMIDITY_TOPIC, CONF_CURRENT_TEMP_TEMPLATE as CONF_CURRENT_TEMP_TEMPLATE, CONF_CURRENT_TEMP_TOPIC as CONF_CURRENT_TEMP_TOPIC, CONF_ENCODING as CONF_ENCODING, CONF_MODE_COMMAND_TEMPLATE as CONF_MODE_COMMAND_TEMPLATE, CONF_MODE_COMMAND_TOPIC as CONF_MODE_COMMAND_TOPIC, CONF_MODE_LIST as CONF_MODE_LIST, CONF_MODE_STATE_TEMPLATE as CONF_MODE_STATE_TEMPLATE, CONF_MODE_STATE_TOPIC as CONF_MODE_STATE_TOPIC, CONF_POWER_COMMAND_TEMPLATE as CONF_POWER_COMMAND_TEMPLATE, CONF_POWER_COMMAND_TOPIC as CONF_POWER_COMMAND_TOPIC, CONF_PRECISION as CONF_PRECISION, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_TEMP_COMMAND_TEMPLATE as CONF_TEMP_COMMAND_TEMPLATE, CONF_TEMP_COMMAND_TOPIC as CONF_TEMP_COMMAND_TOPIC, CONF_TEMP_INITIAL as CONF_TEMP_INITIAL, CONF_TEMP_MAX as CONF_TEMP_MAX, CONF_TEMP_MIN as CONF_TEMP_MIN, CONF_TEMP_STATE_TEMPLATE as CONF_TEMP_STATE_TEMPLATE, CONF_TEMP_STATE_TOPIC as CONF_TEMP_STATE_TOPIC, DEFAULT_OPTIMISTIC as DEFAULT_OPTIMISTIC, PAYLOAD_NONE as PAYLOAD_NONE
-from .debug_info import log_messages as log_messages
-from .mixins import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper, write_state_on_attr_change as write_state_on_attr_change
+from .const import CONF_ACTION_TEMPLATE as CONF_ACTION_TEMPLATE, CONF_ACTION_TOPIC as CONF_ACTION_TOPIC, CONF_CURRENT_HUMIDITY_TEMPLATE as CONF_CURRENT_HUMIDITY_TEMPLATE, CONF_CURRENT_HUMIDITY_TOPIC as CONF_CURRENT_HUMIDITY_TOPIC, CONF_CURRENT_TEMP_TEMPLATE as CONF_CURRENT_TEMP_TEMPLATE, CONF_CURRENT_TEMP_TOPIC as CONF_CURRENT_TEMP_TOPIC, CONF_MODE_COMMAND_TEMPLATE as CONF_MODE_COMMAND_TEMPLATE, CONF_MODE_COMMAND_TOPIC as CONF_MODE_COMMAND_TOPIC, CONF_MODE_LIST as CONF_MODE_LIST, CONF_MODE_STATE_TEMPLATE as CONF_MODE_STATE_TEMPLATE, CONF_MODE_STATE_TOPIC as CONF_MODE_STATE_TOPIC, CONF_POWER_COMMAND_TEMPLATE as CONF_POWER_COMMAND_TEMPLATE, CONF_POWER_COMMAND_TOPIC as CONF_POWER_COMMAND_TOPIC, CONF_PRECISION as CONF_PRECISION, CONF_RETAIN as CONF_RETAIN, CONF_TEMP_COMMAND_TEMPLATE as CONF_TEMP_COMMAND_TEMPLATE, CONF_TEMP_COMMAND_TOPIC as CONF_TEMP_COMMAND_TOPIC, CONF_TEMP_INITIAL as CONF_TEMP_INITIAL, CONF_TEMP_MAX as CONF_TEMP_MAX, CONF_TEMP_MIN as CONF_TEMP_MIN, CONF_TEMP_STATE_TEMPLATE as CONF_TEMP_STATE_TEMPLATE, CONF_TEMP_STATE_TOPIC as CONF_TEMP_STATE_TOPIC, DEFAULT_OPTIMISTIC as DEFAULT_OPTIMISTIC, PAYLOAD_NONE as PAYLOAD_NONE
+from .mixins import MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
 from .models import MqttCommandTemplate as MqttCommandTemplate, MqttValueTemplate as MqttValueTemplate, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
+from .schemas import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA
 from .util import valid_publish_topic as valid_publish_topic, valid_subscribe_topic as valid_subscribe_topic
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
@@ -83,11 +83,9 @@ class MqttTemperatureControlEntity(MqttEntity, ABC, metaclass=abc.ABCMeta):
     _topic: dict[str, Any]
     _command_templates: dict[str, Callable[[PublishPayloadType], PublishPayloadType]]
     _value_templates: dict[str, Callable[[ReceivePayloadType], ReceivePayloadType]]
-    def add_subscription(self, topics: dict[str, dict[str, Any]], topic: str, msg_callback: Callable[[ReceiveMessage], None]) -> None: ...
     def render_template(self, msg: ReceiveMessage, template_name: str) -> ReceivePayloadType: ...
-    def handle_climate_attribute_received(self, msg: ReceiveMessage, template_name: str, attr: str) -> None: ...
-    _sub_state: Incomplete
-    def prepare_subscribe_topics(self, topics: dict[str, dict[str, Any]]) -> None: ...
+    def handle_climate_attribute_received(self, template_name: str, attr: str, msg: ReceiveMessage) -> None: ...
+    def prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def _publish(self, topic: str, payload: PublishPayloadType) -> None: ...
     async def _set_climate_attribute(self, temp: float | None, cmnd_topic: str, cmnd_template: str, state_topic: str, attr: str) -> bool: ...
@@ -128,6 +126,9 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_hvac_action: Incomplete
+    def _handle_action_received(self, msg: ReceiveMessage) -> None: ...
+    def _handle_mode_received(self, template_name: str, attr: str, mode_list: str, msg: ReceiveMessage) -> None: ...
+    def _handle_preset_mode_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def async_set_temperature(self, **kwargs: Any) -> None: ...
     async def async_set_humidity(self, humidity: float) -> None: ...

homeassistant-stubs/components/mqtt/config_flow.pyi

@@ -44,7 +44,6 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
     entry: ConfigEntry | None
     _hassio_discovery: dict[str, Any] | None
-    _reauth_config_entry: ConfigEntry | None
     @staticmethod
     def async_get_options_flow(config_entry: ConfigEntry) -> MQTTOptionsFlowHandler: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/mqtt/const.pyi

@@ -10,13 +10,24 @@ ATTR_QOS: str
 ATTR_RETAIN: str
 ATTR_SERIAL_NUMBER: str
 ATTR_TOPIC: str
+AVAILABILITY_ALL: str
+AVAILABILITY_ANY: str
+AVAILABILITY_LATEST: str
+AVAILABILITY_MODES: Incomplete
+CONF_PAYLOAD_AVAILABLE: str
+CONF_PAYLOAD_NOT_AVAILABLE: str
 CONF_AVAILABILITY: str
+CONF_AVAILABILITY_MODE: str
+CONF_AVAILABILITY_TEMPLATE: str
+CONF_AVAILABILITY_TOPIC: str
 CONF_BROKER: str
 CONF_BIRTH_MESSAGE: str
 CONF_COMMAND_TEMPLATE: str
 CONF_COMMAND_TOPIC: str
 CONF_DISCOVERY_PREFIX: str
 CONF_ENCODING: str
+CONF_JSON_ATTRS_TOPIC: str
+CONF_JSON_ATTRS_TEMPLATE: str
 CONF_KEEPALIVE: str
 CONF_ORIGIN: str
 CONF_QOS = ATTR_QOS
@@ -37,6 +48,7 @@ CONF_CURRENT_HUMIDITY_TEMPLATE: str
 CONF_CURRENT_HUMIDITY_TOPIC: str
 CONF_CURRENT_TEMP_TEMPLATE: str
 CONF_CURRENT_TEMP_TOPIC: str
+CONF_ENABLED_BY_DEFAULT: str
 CONF_MODE_COMMAND_TEMPLATE: str
 CONF_MODE_COMMAND_TOPIC: str
 CONF_MODE_LIST: str
@@ -77,8 +89,6 @@ CONF_SUGGESTED_AREA: str
 CONF_CONFIGURATION_URL: str
 CONF_OBJECT_ID: str
 CONF_SUPPORT_URL: str
-DATA_MQTT: str
-DATA_MQTT_AVAILABLE: str
 DEFAULT_PREFIX: str
 DEFAULT_BIRTH_WILL_TOPIC: Incomplete
 DEFAULT_DISCOVERY: bool
@@ -107,9 +117,9 @@ DEFAULT_TRANSPORT = TRANSPORT_TCP
 DEFAULT_BIRTH: Incomplete
 DEFAULT_WILL: Incomplete
 DOMAIN: str
-MQTT_CONNECTED: str
-MQTT_DISCONNECTED: str
+MQTT_CONNECTION_STATE: str
 PAYLOAD_EMPTY_JSON: str
 PAYLOAD_NONE: str
 RELOADABLE_PLATFORMS: Incomplete
 TEMPLATE_ERRORS: Incomplete
+SUPPORTED_COMPONENTS: Incomplete