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

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,12 +1,13 @@
 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_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 .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
@@ -16,21 +17,25 @@ 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]):
     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 _valid_origin_info(discovery_payload: MQTTDiscoveryPayload) -> bool: ...
 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

homeassistant-stubs/components/mqtt/humidifier.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_ACTION_TEMPLATE as CONF_ACTION_TEMPLATE, CONF_ACTION_TOPIC as CONF_ACTION_TOPIC, CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_CURRENT_HUMIDITY_TEMPLATE as CONF_CURRENT_HUMIDITY_TEMPLATE, CONF_CURRENT_HUMIDITY_TOPIC as CONF_CURRENT_HUMIDITY_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 .const import CONF_ACTION_TEMPLATE as CONF_ACTION_TEMPLATE, CONF_ACTION_TOPIC as CONF_ACTION_TOPIC, CONF_COMMAND_TEMPLATE as CONF_COMMAND_TEMPLATE, CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_CURRENT_HUMIDITY_TEMPLATE as CONF_CURRENT_HUMIDITY_TEMPLATE, CONF_CURRENT_HUMIDITY_TOPIC as CONF_CURRENT_HUMIDITY_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
@@ -70,12 +70,15 @@ class MqttHumidifier(MqttEntity, HumidifierEntity):
     _attr_supported_features: Incomplete
     _attr_assumed_state: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
-    def add_subscription(self, topics: dict[str, dict[str, Any]], topic: str, msg_callback: Callable[[ReceiveMessage], None]) -> None: ...
     _attr_is_on: bool
+    def _state_received(self, msg: ReceiveMessage) -> None: ...
     _attr_action: Incomplete
+    def _action_received(self, msg: ReceiveMessage) -> None: ...
     _attr_current_humidity: Incomplete
+    def _current_humidity_received(self, msg: ReceiveMessage) -> None: ...
     _attr_target_humidity: Incomplete
-    _sub_state: Incomplete
+    def _target_humidity_received(self, msg: ReceiveMessage) -> None: ...
+    def _mode_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/mqtt/image.pyi

@@ -2,11 +2,10 @@ import httpx
 import voluptuous as vol
 from . import subscription as subscription
 from .config import MQTT_BASE_SCHEMA as MQTT_BASE_SCHEMA
-from .const import CONF_ENCODING as CONF_ENCODING, CONF_QOS as CONF_QOS
-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 MessageCallbackType as MessageCallbackType, MqttValueTemplate as MqttValueTemplate, MqttValueTemplateException as MqttValueTemplateException, ReceiveMessage as ReceiveMessage
-from .util import get_mqtt_data as get_mqtt_data, valid_subscribe_topic as valid_subscribe_topic
+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, 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 collections.abc import Callable as Callable
 from homeassistant.components import image as image
@@ -51,8 +50,9 @@ class MqttImage(MqttEntity, ImageEntity):
     _attr_image_url: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_image_last_updated: Incomplete
+    def _image_data_received(self, msg: ReceiveMessage) -> None: ...
     _cached_image: Incomplete
-    _sub_state: Incomplete
+    def _image_from_url_request_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_image(self) -> bytes | None: ...

homeassistant-stubs/components/mqtt/lawn_mower.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_ENCODING as CONF_ENCODING, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, DEFAULT_OPTIMISTIC as DEFAULT_OPTIMISTIC, 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_RETAIN as CONF_RETAIN, DEFAULT_OPTIMISTIC as DEFAULT_OPTIMISTIC, DEFAULT_RETAIN as DEFAULT_RETAIN
+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
@@ -50,7 +50,7 @@ class MqttLawnMower(MqttEntity, LawnMowerEntity, RestoreEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_activity: Incomplete
-    _sub_state: Incomplete
+    def _message_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def _async_operate(self, option: str, activity: LawnMowerActivity) -> None: ...

homeassistant-stubs/components/mqtt/light/schema_basic.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_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, write_state_on_attr_change as write_state_on_attr_change
-from ..models import MessageCallbackType as MessageCallbackType, MqttCommandTemplate as MqttCommandTemplate, MqttValueTemplate as MqttValueTemplate, PayloadSentinel as PayloadSentinel, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType, TemplateVarsType as TemplateVarsType
+from ..const import 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
+from ..models import MqttCommandTemplate as MqttCommandTemplate, MqttValueTemplate as MqttValueTemplate, PayloadSentinel as PayloadSentinel, PublishPayloadType as PublishPayloadType, ReceiveMessage as ReceiveMessage, ReceivePayloadType as ReceivePayloadType, TemplateVarsType as TemplateVarsType
+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 .schema import MQTT_LIGHT_SCHEMA_SCHEMA as MQTT_LIGHT_SCHEMA_SCHEMA
 from _typeshed import Incomplete
@@ -101,15 +101,25 @@ class MqttLight(MqttEntity, LightEntity, RestoreEntity):
     def _setup_from_config(self, config: ConfigType) -> None: ...
     def _is_optimistic(self, attribute: str) -> bool: ...
     _attr_is_on: bool
+    def _state_received(self, msg: ReceiveMessage) -> None: ...
     _attr_brightness: Incomplete
+    def _brightness_received(self, msg: ReceiveMessage) -> None: ...
+    def _rgbx_received(self, msg: ReceiveMessage, template: str, color_mode: ColorMode, convert_color: Callable[..., tuple[int, ...]]) -> tuple[int, ...] | None: ...
     _attr_rgb_color: Incomplete
+    def _rgb_received(self, msg: ReceiveMessage) -> None: ...
     _attr_rgbw_color: Incomplete
+    def _rgbw_received(self, msg: ReceiveMessage) -> None: ...
     _attr_rgbww_color: Incomplete
+    def _rgbww_received(self, msg: ReceiveMessage) -> None: ...
+    def _color_mode_received(self, msg: ReceiveMessage) -> None: ...
     _attr_color_temp: Incomplete
+    def _color_temp_received(self, msg: ReceiveMessage) -> None: ...
     _attr_effect: Incomplete
+    def _effect_received(self, msg: ReceiveMessage) -> None: ...
     _attr_hs_color: Incomplete
+    def _hs_received(self, msg: ReceiveMessage) -> None: ...
     _attr_xy_color: Incomplete
-    _sub_state: Incomplete
+    def _xy_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/mqtt/light/schema_json.pyi

@@ -1,10 +1,10 @@
 import voluptuous as vol
 from .. import subscription as subscription
 from ..config import DEFAULT_QOS as DEFAULT_QOS, DEFAULT_RETAIN as DEFAULT_RETAIN, MQTT_RW_SCHEMA as MQTT_RW_SCHEMA
-from ..const import 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
-from ..debug_info import log_messages as log_messages
-from ..mixins import MQTT_ENTITY_COMMON_SCHEMA as MQTT_ENTITY_COMMON_SCHEMA, MqttEntity as MqttEntity, write_state_on_attr_change as write_state_on_attr_change
+from ..const import CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_STATE_TOPIC as CONF_STATE_TOPIC
+from ..mixins import MqttEntity as MqttEntity
 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 .schema import MQTT_LIGHT_SCHEMA_SCHEMA as MQTT_LIGHT_SCHEMA_SCHEMA
 from .schema_basic import CONF_BRIGHTNESS_SCALE as CONF_BRIGHTNESS_SCALE, CONF_WHITE_SCALE as CONF_WHITE_SCALE, MQTT_LIGHT_ATTRIBUTES_BLOCKED as MQTT_LIGHT_ATTRIBUTES_BLOCKED
@@ -77,7 +77,7 @@ class MqttLightJson(MqttEntity, LightEntity, RestoreEntity):
     _attr_is_on: bool
     _attr_brightness: Incomplete
     _attr_effect: Incomplete
-    _sub_state: Incomplete
+    def _state_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     @property

homeassistant-stubs/components/mqtt/light/schema_template.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_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, 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, write_state_on_attr_change as write_state_on_attr_change
+from ..const import CONF_COMMAND_TOPIC as CONF_COMMAND_TOPIC, CONF_STATE_TOPIC as CONF_STATE_TOPIC, PAYLOAD_NONE as PAYLOAD_NONE
+from ..mixins import MqttEntity as MqttEntity
 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 .schema import MQTT_LIGHT_SCHEMA_SCHEMA as MQTT_LIGHT_SCHEMA_SCHEMA
 from .schema_basic import MQTT_LIGHT_ATTRIBUTES_BLOCKED as MQTT_LIGHT_ATTRIBUTES_BLOCKED
 from _typeshed import Incomplete
@@ -60,7 +60,7 @@ class MqttLightTemplate(MqttEntity, LightEntity, RestoreEntity):
     _attr_color_temp: Incomplete
     _attr_hs_color: Incomplete
     _attr_effect: Incomplete
-    _sub_state: Incomplete
+    def _state_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/mqtt/lock.pyi

@@ -2,10 +2,10 @@ import re
 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_PAYLOAD_RESET as CONF_PAYLOAD_RESET, CONF_QOS as CONF_QOS, CONF_RETAIN as CONF_RETAIN, CONF_STATE_TOPIC as CONF_STATE_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, 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_PAYLOAD_RESET as CONF_PAYLOAD_RESET, CONF_STATE_OPEN as CONF_STATE_OPEN, CONF_STATE_OPENING as CONF_STATE_OPENING, CONF_STATE_TOPIC as CONF_STATE_TOPIC
+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 _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from homeassistant.components import lock as lock
@@ -17,6 +17,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 from homeassistant.helpers.typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType
 from typing import Any
 
+_LOGGER: Incomplete
 CONF_CODE_FORMAT: str
 CONF_PAYLOAD_LOCK: str
 CONF_PAYLOAD_UNLOCK: str
@@ -33,6 +34,8 @@ DEFAULT_PAYLOAD_OPEN: str
 DEFAULT_PAYLOAD_RESET: str
 DEFAULT_STATE_LOCKED: str
 DEFAULT_STATE_LOCKING: str
+DEFAULT_STATE_OPEN: str
+DEFAULT_STATE_OPENING: str
 DEFAULT_STATE_UNLOCKED: str
 DEFAULT_STATE_UNLOCKING: str
 DEFAULT_STATE_JAMMED: str
@@ -60,9 +63,11 @@ class MqttLock(MqttEntity, LockEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_is_locking: Incomplete
+    _attr_is_open: Incomplete
+    _attr_is_opening: Incomplete
     _attr_is_unlocking: Incomplete
     _attr_is_jammed: Incomplete
-    _sub_state: Incomplete
+    def _message_received(self, msg: ReceiveMessage) -> None: ...
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_lock(self, **kwargs: Any) -> None: ...