homeassistant-stubs 2024.5.5__py3-none-any.whl → 2024.6.0b0__py3-none-any.whl

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,54 @@
 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.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.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 +56,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 +70,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 +89,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
@@ -125,7 +130,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 +142,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
@@ -64,6 +76,7 @@ CONF_TEMP_MIN: str
 CONF_CERTIFICATE: str
 CONF_CLIENT_KEY: str
 CONF_CLIENT_CERT: str
+CONF_COMPONENTS: str
 CONF_TLS_INSECURE: str
 CONF_IDENTIFIERS: str
 CONF_CONNECTIONS: str
@@ -77,8 +90,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 +118,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

homeassistant-stubs/components/mqtt/cover.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_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_ENCODING as CONF_ENCODING, CONF_PAYLOAD_CLOSE as CONF_PAYLOAD_CLOSE, CONF_PAYLOAD_OPEN as CONF_PAYLOAD_OPEN, CONF_PAYLOAD_STOP as CONF_PAYLOAD_STOP, CONF_POSITION_CLOSED as CONF_POSITION_CLOSED, CONF_POSITION_OPEN as CONF_POSITION_OPEN, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_STATE_CLOSED as CONF_STATE_CLOSED, CONF_STATE_CLOSING as CONF_STATE_CLOSING, CONF_STATE_OPEN as CONF_STATE_OPEN, CONF_STATE_OPENING as CONF_STATE_OPENING, CONF_STATE_TOPIC as CONF_STATE_TOPIC, DEFAULT_OPTIMISTIC as DEFAULT_OPTIMISTIC, DEFAULT_PAYLOAD_CLOSE as DEFAULT_PAYLOAD_CLOSE, DEFAULT_PAYLOAD_OPEN as DEFAULT_PAYLOAD_OPEN, DEFAULT_POSITION_CLOSED as DEFAULT_POSITION_CLOSED, DEFAULT_POSITION_OPEN as DEFAULT_POSITION_OPEN, DEFAULT_RETAIN as DEFAULT_RETAIN
-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_TOPIC as CONF_COMMAND_TOPIC, CONF_PAYLOAD_CLOSE as CONF_PAYLOAD_CLOSE, CONF_PAYLOAD_OPEN as CONF_PAYLOAD_OPEN, CONF_PAYLOAD_STOP as CONF_PAYLOAD_STOP, CONF_POSITION_CLOSED as CONF_POSITION_CLOSED, CONF_POSITION_OPEN as CONF_POSITION_OPEN, CONF_RETAIN as CONF_RETAIN, CONF_STATE_CLOSED as CONF_STATE_CLOSED, CONF_STATE_CLOSING as CONF_STATE_CLOSING, CONF_STATE_OPEN as CONF_STATE_OPEN, CONF_STATE_OPENING as CONF_STATE_OPENING, CONF_STATE_TOPIC as CONF_STATE_TOPIC, DEFAULT_OPTIMISTIC as DEFAULT_OPTIMISTIC, DEFAULT_PAYLOAD_CLOSE as DEFAULT_PAYLOAD_CLOSE, DEFAULT_PAYLOAD_OPEN as DEFAULT_PAYLOAD_OPEN, DEFAULT_POSITION_CLOSED as DEFAULT_POSITION_CLOSED, DEFAULT_POSITION_OPEN as DEFAULT_POSITION_OPEN, DEFAULT_RETAIN as DEFAULT_RETAIN, 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 import cover as cover
@@ -79,9 +79,11 @@ class MqttCover(MqttEntity, CoverEntity):
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_is_opening: Incomplete
     _attr_is_closing: Incomplete
-    def _update_state(self, state: str) -> None: ...
+    def _update_state(self, state: str | None) -> None: ...
+    def _tilt_message_received(self, msg: ReceiveMessage) -> None: ...
+    def _state_message_received(self, msg: ReceiveMessage) -> None: ...
     _attr_current_cover_position: Incomplete
-    _sub_state: Incomplete
+    def _position_message_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_open_cover(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/mqtt/debug_info.pyi

@@ -1,8 +1,5 @@
-import datetime as dt
 from .const import ATTR_DISCOVERY_PAYLOAD as ATTR_DISCOVERY_PAYLOAD, ATTR_DISCOVERY_TOPIC as ATTR_DISCOVERY_TOPIC
-from .models import MessageCallbackType as MessageCallbackType, PublishPayloadType as PublishPayloadType
-from .util import get_mqtt_data as get_mqtt_data
-from collections.abc import Callable as Callable
+from .models import DATA_MQTT as DATA_MQTT, PublishPayloadType as PublishPayloadType
 from dataclasses import dataclass
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
@@ -10,20 +7,18 @@ from typing import Any
 
 STORED_MESSAGES: int
 
-def log_messages(hass: HomeAssistant, entity_id: str) -> Callable[[MessageCallbackType], MessageCallbackType]: ...
-
 @dataclass
 class TimestampedPublishMessage:
     topic: str
     payload: PublishPayloadType
     qos: int
     retain: bool
-    timestamp: dt.datetime
+    timestamp: float
     def __init__(self, topic, payload, qos, retain, timestamp) -> None: ...
 
 def log_message(hass: HomeAssistant, entity_id: str, topic: str, payload: PublishPayloadType, qos: int, retain: bool) -> None: ...
-def add_subscription(hass: HomeAssistant, message_callback: MessageCallbackType, subscription: str) -> None: ...
-def remove_subscription(hass: HomeAssistant, message_callback: MessageCallbackType, subscription: str) -> None: ...
+def add_subscription(hass: HomeAssistant, subscription: str, entity_id: str | None) -> None: ...
+def remove_subscription(hass: HomeAssistant, subscription: str, entity_id: str | None) -> None: ...
 def add_entity_discovery_data(hass: HomeAssistant, discovery_data: DiscoveryInfoType, entity_id: str) -> None: ...
 def update_entity_discovery_data(hass: HomeAssistant, discovery_payload: DiscoveryInfoType, entity_id: str) -> None: ...
 def remove_entity_data(hass: HomeAssistant, entity_id: str) -> None: ...

homeassistant-stubs/components/mqtt/device_tracker.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_PAYLOAD_RESET as CONF_PAYLOAD_RESET, CONF_QOS as CONF_QOS, CONF_STATE_TOPIC as CONF_STATE_TOPIC
-from .debug_info import log_messages as log_messages
-from .mixins import CONF_JSON_ATTRS_TOPIC as CONF_JSON_ATTRS_TOPIC, 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_PAYLOAD_RESET as CONF_PAYLOAD_RESET, CONF_STATE_TOPIC as CONF_STATE_TOPIC
+from .mixins import CONF_JSON_ATTRS_TOPIC as CONF_JSON_ATTRS_TOPIC, MqttEntity as MqttEntity, async_setup_entity_entry_helper as async_setup_entity_entry_helper
 from .models import MqttValueTemplate as MqttValueTemplate, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
+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 collections.abc import Callable as Callable
@@ -16,6 +16,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import ConfigType as ConfigType
 
+_LOGGER: Incomplete
 CONF_PAYLOAD_HOME: str
 CONF_PAYLOAD_NOT_HOME: str
 CONF_SOURCE_TYPE: str
@@ -34,11 +35,11 @@ class MqttDeviceTracker(MqttEntity, TrackerEntity):
     _default_name: Incomplete
     _entity_id_format: Incomplete
     _location_name: str | None
-    _value_template: Callable[..., ReceivePayloadType]
+    _value_template: Callable[[ReceivePayloadType], ReceivePayloadType]
     @staticmethod
     def config_schema() -> vol.Schema: ...
     def _setup_from_config(self, config: ConfigType) -> None: ...
-    _sub_state: Incomplete
+    def _tracker_message_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     @property
     def force_update(self) -> bool: ...

homeassistant-stubs/components/mqtt/device_trigger.pyi

@@ -2,10 +2,12 @@ from . import debug_info as debug_info
 from .config import MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA
 from .const import ATTR_DISCOVERY_HASH as ATTR_DISCOVERY_HASH, CONF_ENCODING as CONF_ENCODING, CONF_PAYLOAD as CONF_PAYLOAD, CONF_QOS as CONF_QOS, CONF_TOPIC as CONF_TOPIC, DOMAIN as DOMAIN
 from .discovery import MQTTDiscoveryPayload as MQTTDiscoveryPayload, clear_discovery_hash as clear_discovery_hash
-from .mixins import MQTT_ENTITY_DEVICE_INFO_SCHEMA as MQTT_ENTITY_DEVICE_INFO_SCHEMA, MqttDiscoveryDeviceUpdate as MqttDiscoveryDeviceUpdate, send_discovery_done as send_discovery_done, update_device as update_device
-from .util import get_mqtt_data as get_mqtt_data
+from .mixins import MqttDiscoveryDeviceUpdateMixin as MqttDiscoveryDeviceUpdateMixin, send_discovery_done as send_discovery_done, update_device as update_device
+from .models import DATA_MQTT as DATA_MQTT
+from .schemas import MQTT_ENTITY_DEVICE_INFO_SCHEMA as MQTT_ENTITY_DEVICE_INFO_SCHEMA
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
+from dataclasses import dataclass
 from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA as DEVICE_TRIGGER_BASE_SCHEMA
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_DEVICE as CONF_DEVICE, CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_DOMAIN as CONF_DOMAIN, CONF_PLATFORM as CONF_PLATFORM, CONF_TYPE as CONF_TYPE, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE
@@ -25,22 +27,20 @@ TRIGGER_SCHEMA: Incomplete
 TRIGGER_DISCOVERY_SCHEMA: Incomplete
 LOG_NAME: str
 
+@dataclass(slots=True)
 class TriggerInstance:
     action: TriggerActionType
     trigger_info: TriggerInfo
     trigger: Trigger
-    remove: CALLBACK_TYPE | None
+    remove: CALLBACK_TYPE | None = ...
     async def async_attach_trigger(self) -> None: ...
     def __init__(self, action, trigger_info, trigger, remove) -> None: ...
-    def __lt__(self, other): ...
-    def __le__(self, other): ...
-    def __gt__(self, other): ...
-    def __ge__(self, other): ...
 
+@dataclass(slots=True, kw_only=True)
 class Trigger:
     device_id: str
-    discovery_data: DiscoveryInfoType | None
-    discovery_id: str | None
+    discovery_data: DiscoveryInfoType | None = ...
+    discovery_id: str | None = ...
     hass: HomeAssistant
     payload: str | None
     qos: int | None
@@ -48,17 +48,13 @@ class Trigger:
     topic: str | None
     type: str
     value_template: str | None
-    trigger_instances: list[TriggerInstance]
+    trigger_instances: list[TriggerInstance] = ...
     async def add_trigger(self, action: TriggerActionType, trigger_info: TriggerInfo) -> Callable[[], None]: ...
     async def update_trigger(self, config: ConfigType) -> None: ...
     def detach_trigger(self) -> None: ...
-    def __init__(self, device_id, discovery_data, discovery_id, hass, payload, qos, subtype, topic, type, value_template, trigger_instances) -> None: ...
-    def __lt__(self, other): ...
-    def __le__(self, other): ...
-    def __gt__(self, other): ...
-    def __ge__(self, other): ...
+    def __init__(self, *, device_id, discovery_data, discovery_id, hass, payload, qos, subtype, topic, type, value_template, trigger_instances) -> None: ...
 
-class MqttDeviceTrigger(MqttDiscoveryDeviceUpdate):
+class MqttDeviceTrigger(MqttDiscoveryDeviceUpdateMixin):
     _config: Incomplete
     _config_entry: Incomplete
     device_id: Incomplete

homeassistant-stubs/components/mqtt/diagnostics.pyi

@@ -1,5 +1,5 @@
 from . import debug_info as debug_info, is_connected as is_connected
-from .util import get_mqtt_data as get_mqtt_data
+from .models import DATA_MQTT as DATA_MQTT
 from _typeshed import Incomplete
 from homeassistant.components import device_tracker as device_tracker
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data

homeassistant-stubs/components/mqtt/discovery.pyi

@@ -1,36 +1,45 @@
 from .. import mqtt as mqtt
 from .abbreviations import ABBREVIATIONS as ABBREVIATIONS, DEVICE_ABBREVIATIONS as DEVICE_ABBREVIATIONS, ORIGIN_ABBREVIATIONS as ORIGIN_ABBREVIATIONS
-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_SUPPORT_URL as CONF_SUPPORT_URL, CONF_SW_VERSION as CONF_SW_VERSION, CONF_TOPIC as CONF_TOPIC, DOMAIN as DOMAIN
-from .models import MqttOriginInfo as MqttOriginInfo, ReceiveMessage as ReceiveMessage
-from .util import async_forward_entry_setup_and_setup_discovery as async_forward_entry_setup_and_setup_discovery, get_mqtt_data as get_mqtt_data
+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 homeassistant.const import CONF_DEVICE as CONF_DEVICE, CONF_NAME as CONF_NAME, CONF_PLATFORM as CONF_PLATFORM
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+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.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
 from homeassistant.util.signal_type import SignalTypeFormat as SignalTypeFormat
 from typing import Any
 
+ABBREVIATIONS_SET: Incomplete
+DEVICE_ABBREVIATIONS_SET: Incomplete
+ORIGIN_ABBREVIATIONS_SET: Incomplete
 _LOGGER: Incomplete
 TOPIC_MATCHER: Incomplete
-SUPPORTED_COMPONENTS: Incomplete
 MQTT_DISCOVERY_UPDATED: SignalTypeFormat[MQTTDiscoveryPayload]
 MQTT_DISCOVERY_NEW: SignalTypeFormat[MQTTDiscoveryPayload]
-MQTT_DISCOVERY_NEW_COMPONENT: str
 MQTT_DISCOVERY_DONE: SignalTypeFormat[Any]
 TOPIC_BASE: str
-MQTT_ORIGIN_INFO_SCHEMA: Incomplete
 
 class MQTTDiscoveryPayload(dict[str, Any]):
+    device_discovery: bool
     discovery_data: DiscoveryInfoType
 
 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 async_log_discovery_origin_info(message: str, discovery_payload: MQTTDiscoveryPayload) -> 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 _generate_device_cleanup_config(hass: HomeAssistant, object_id: str, node_id: str | None) -> dict[str, Any]: ...
+def _parse_device_payload(hass: HomeAssistant, payload: ReceivePayloadType, object_id: str, node_id: str | None) -> dict[str, Any]: ...
+def _valid_origin_info(discovery_payload: MQTTDiscoveryPayload) -> bool: ...
+def _merge_common_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/event.pyi

@@ -1,11 +1,10 @@
 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_EMPTY_JSON as PAYLOAD_EMPTY_JSON, 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
-from .models import MqttValueTemplate as MqttValueTemplate, MqttValueTemplateException as MqttValueTemplateException, PayloadSentinel as PayloadSentinel, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
-from .util import get_mqtt_data as get_mqtt_data
+from .const import CONF_STATE_TOPIC as CONF_STATE_TOPIC, PAYLOAD_EMPTY_JSON as PAYLOAD_EMPTY_JSON, 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 DATA_MQTT as DATA_MQTT, MqttValueTemplate as MqttValueTemplate, MqttValueTemplateException as MqttValueTemplateException, PayloadSentinel as PayloadSentinel, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
+from .schemas import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from homeassistant.components import event as event
@@ -39,6 +38,6 @@ class MqttEvent(MqttEntity, EventEntity):
     _attr_device_class: Incomplete
     _attr_event_types: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
-    _sub_state: Incomplete
+    def _event_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...

homeassistant-stubs/components/mqtt/fan.pyi

@@ -1,10 +1,10 @@
 import voluptuous as vol
 from . import subscription as subscription
 from .config import MQTT_RW_SCHEMA as MQTT_RW_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_STATE_VALUE_TEMPLATE as CONF_STATE_VALUE_TEMPLATE, 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 .models import MessageCallbackType as MessageCallbackType, MqttCommandTemplate as MqttCommandTemplate, MqttValueTemplate as MqttValueTemplate, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType
+from .const import CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_STATE_TOPIC as CONF_STATE_TOPIC, CONF_STATE_VALUE_TEMPLATE as CONF_STATE_VALUE_TEMPLATE, 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 collections.abc import Callable as Callable
@@ -89,9 +89,13 @@ class MqttFan(MqttEntity, FanEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_is_on: bool
+    def _state_received(self, msg: ReceiveMessage) -> None: ...
+    def _percentage_received(self, msg: ReceiveMessage) -> None: ...
+    def _preset_mode_received(self, msg: ReceiveMessage) -> None: ...
     _attr_oscillating: bool
+    def _oscillation_received(self, msg: ReceiveMessage) -> None: ...
     _attr_current_direction: Incomplete
-    _sub_state: Incomplete
+    def _direction_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     @property