homeassistant-stubs 2025.1.1__py3-none-any.whl → 2025.1.3__py3-none-any.whl

homeassistant-stubs/components/mqtt/client.pyi

@@ -1,4 +1,5 @@
 import asyncio
+import contextlib
 import paho.mqtt.client as mqtt
 import socket
 import ssl
@@ -42,8 +43,11 @@ type SubscribePayloadType = str | bytes | bytearray
 
 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: ...
+@bind_hass
 async def async_subscribe(hass: HomeAssistant, topic: str, msg_callback: Callable[[ReceiveMessage], Coroutine[Any, Any, None] | None], qos: int = ..., encoding: str | None = ...) -> CALLBACK_TYPE: ...
+@callback
 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: ...
+@bind_hass
 def subscribe(hass: HomeAssistant, topic: str, msg_callback: MessageCallbackType, qos: int = ..., encoding: str = 'utf-8') -> Callable[[], None]: ...
 
 @dataclass(slots=True, frozen=True)
@@ -54,7 +58,6 @@ class Subscription:
     job: HassJob[[ReceiveMessage], Coroutine[Any, Any, None] | None]
     qos: int = ...
     encoding: str | None = ...
-    def __init__(self, topic, is_simple_match, complex_matcher, job, qos=..., encoding=...) -> None: ...
 
 class MqttClientSetup:
     _client: AsyncMQTTClient
@@ -91,50 +94,76 @@ class MQTT:
     _pending_unsubscribes: set[str]
     _socket_buffersize: int | None
     def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, conf: ConfigType) -> None: ...
+    @callback
     def _async_ha_started(self, _hass: HomeAssistant) -> None: ...
     async def _async_ha_stop(self, _event: Event) -> None: ...
     async def async_start(self, mqtt_data: MqttData) -> None: ...
     @property
     def subscriptions(self) -> set[Subscription]: ...
     def cleanup(self) -> None: ...
+    @contextlib.asynccontextmanager
     async def _async_connect_in_executor(self) -> AsyncGenerator[None]: ...
     async def async_init_client(self) -> None: ...
+    @callback
     def _async_reader_callback(self, client: mqtt.Client) -> None: ...
+    @callback
     def _async_start_misc_periodic(self) -> None: ...
     def _increase_socket_buffer_size(self, sock: SocketType) -> None: ...
     def _on_socket_open(self, client: mqtt.Client, userdata: Any, sock: SocketType) -> None: ...
+    @callback
     def _async_on_socket_open(self, client: mqtt.Client, userdata: Any, sock: SocketType) -> None: ...
+    @callback
     def _async_on_socket_close(self, client: mqtt.Client, userdata: Any, sock: SocketType) -> None: ...
+    @callback
     def _async_writer_callback(self, client: mqtt.Client) -> None: ...
     def _on_socket_register_write(self, client: mqtt.Client, userdata: Any, sock: SocketType) -> None: ...
+    @callback
     def _async_on_socket_register_write(self, client: mqtt.Client, userdata: Any, sock: SocketType) -> None: ...
+    @callback
     def _async_on_socket_unregister_write(self, client: mqtt.Client, userdata: Any, sock: SocketType) -> None: ...
     def _is_active_subscription(self, topic: str) -> bool: ...
     async def async_publish(self, topic: str, payload: PublishPayloadType, qos: int, retain: bool) -> None: ...
     async def async_connect(self, client_available: asyncio.Future[bool]) -> None: ...
+    @callback
     def _async_connection_result(self, connected: bool) -> None: ...
+    @callback
     def _async_cancel_reconnect(self) -> None: ...
     async def _reconnect_loop(self) -> None: ...
     async def async_disconnect(self, disconnect_paho_client: bool = False) -> None: ...
+    @callback
     def async_restore_tracked_subscriptions(self, subscriptions: set[Subscription]) -> None: ...
+    @callback
     def _async_track_subscription(self, subscription: Subscription) -> None: ...
+    @callback
     def _async_untrack_subscription(self, subscription: Subscription) -> None: ...
+    @callback
     def _async_queue_subscriptions(self, subscriptions: Iterable[tuple[str, int]], queue_only: bool = False) -> None: ...
     def _exception_message(self, msg_callback: Callable[[ReceiveMessage], Coroutine[Any, Any, None] | None], msg: ReceiveMessage) -> str: ...
+    @callback
     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]: ...
+    @callback
     def _async_remove(self, subscription: Subscription) -> None: ...
+    @callback
     def _async_unsubscribe(self, topic: str) -> None: ...
     async def _async_perform_subscriptions(self) -> None: ...
     async def _async_perform_unsubscribes(self) -> None: ...
     async def _async_resubscribe_and_publish_birth_message(self, birth_message: PublishMessage) -> None: ...
+    @callback
     def _async_mqtt_on_connect(self, _mqttc: mqtt.Client, _userdata: None, _flags: dict[str, int], result_code: int, properties: mqtt.Properties | None = None) -> None: ...
+    @callback
     def _async_queue_resubscribe(self) -> None: ...
     def _matching_subscriptions(self, topic: str) -> list[Subscription]: ...
+    @callback
     def _async_mqtt_on_message(self, _mqttc: mqtt.Client, _userdata: None, msg: mqtt.MQTTMessage) -> None: ...
+    @callback
     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: ...
+    @callback
     def _async_get_mid_future(self, mid: int) -> asyncio.Future[None]: ...
+    @callback
     def _async_mqtt_on_disconnect(self, _mqttc: mqtt.Client, _userdata: None, result_code: int, properties: mqtt.Properties | None = None) -> None: ...
+    @callback
     def _async_on_disconnect(self, result_code: int) -> None: ...
+    @callback
     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: ...

homeassistant-stubs/components/mqtt/climate.pyi

@@ -80,7 +80,9 @@ class MqttTemperatureControlEntity(MqttEntity, ABC, metaclass=abc.ABCMeta):
     _command_templates: dict[str, Callable[[PublishPayloadType], PublishPayloadType]]
     _value_templates: dict[str, Callable[[ReceivePayloadType], ReceivePayloadType]]
     def render_template(self, msg: ReceiveMessage, template_name: str) -> ReceivePayloadType: ...
+    @callback
     def handle_climate_attribute_received(self, template_name: str, attr: str, msg: ReceiveMessage) -> None: ...
+    @callback
     def prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def _publish(self, topic: str, payload: PublishPayloadType) -> None: ...
@@ -121,9 +123,13 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_hvac_action: Incomplete
+    @callback
     def _handle_action_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _handle_mode_received(self, template_name: str, attr: str, mode_list: str, msg: ReceiveMessage) -> None: ...
+    @callback
     def _handle_preset_mode_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     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

@@ -46,6 +46,7 @@ KEY_UPLOAD_SELECTOR: Incomplete
 REAUTH_SCHEMA: Incomplete
 PWD_NOT_CHANGED: str
 
+@callback
 def update_password_from_user_input(entry_password: str | None, user_input: dict[str, Any]) -> dict[str, Any]: ...
 
 class FlowHandler(ConfigFlow, domain=DOMAIN):
@@ -56,6 +57,7 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
     start_task: asyncio.Task | None
     def __init__(self) -> None: ...
     @staticmethod
+    @callback
     def async_get_options_flow(config_entry: ConfigEntry) -> MQTTOptionsFlowHandler: ...
     async def _async_install_addon(self) -> None: ...
     async def async_step_install_failed(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/mqtt/cover.pyi

@@ -79,11 +79,16 @@ class MqttCover(MqttEntity, CoverEntity):
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_is_opening: Incomplete
     _attr_is_closing: Incomplete
+    @callback
     def _update_state(self, state: str | None) -> None: ...
+    @callback
     def _tilt_message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _state_message_received(self, msg: ReceiveMessage) -> None: ...
     _attr_current_cover_position: Incomplete
+    @callback
     def _position_message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_open_cover(self, **kwargs: Any) -> None: ...
@@ -95,4 +100,5 @@ class MqttCover(MqttEntity, CoverEntity):
     async def async_set_cover_tilt_position(self, **kwargs: Any) -> None: ...
     async def async_set_cover_position(self, **kwargs: Any) -> None: ...
     async def async_toggle_tilt(self, **kwargs: Any) -> None: ...
+    @callback
     def tilt_payload_received(self, _payload: Any) -> None: ...

homeassistant-stubs/components/mqtt/debug_info.pyi

@@ -14,7 +14,6 @@ class TimestampedPublishMessage:
     qos: int
     retain: bool
     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, subscription: str, entity_id: str | None) -> None: ...

homeassistant-stubs/components/mqtt/device_tracker.pyi

@@ -40,7 +40,9 @@ class MqttDeviceTracker(MqttEntity, TrackerEntity):
     @staticmethod
     def config_schema() -> VolSchemaType: ...
     def _setup_from_config(self, config: ConfigType) -> None: ...
+    @callback
     def _tracker_message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     @property
     def force_update(self) -> bool: ...

homeassistant-stubs/components/mqtt/device_trigger.pyi

@@ -7,7 +7,7 @@ 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 dataclasses import dataclass, field
 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
@@ -34,7 +34,6 @@ class TriggerInstance:
     trigger: Trigger
     remove: CALLBACK_TYPE | None = ...
     async def async_attach_trigger(self) -> None: ...
-    def __init__(self, action, trigger_info, trigger, remove=...) -> None: ...
 
 @dataclass(slots=True, kw_only=True)
 class Trigger:
@@ -48,11 +47,10 @@ class Trigger:
     topic: str | None
     type: str
     value_template: str | None
-    trigger_instances: list[TriggerInstance] = ...
+    trigger_instances: list[TriggerInstance] = field(default_factory=list)
     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: ...
 
 class MqttDeviceTrigger(MqttDiscoveryDeviceUpdateMixin):
     _config: Incomplete

homeassistant-stubs/components/mqtt/diagnostics.pyi

@@ -14,5 +14,7 @@ REDACT_STATE_DEVICE_TRACKER: Incomplete
 
 async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
 async def async_get_device_diagnostics(hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry) -> dict[str, Any]: ...
+@callback
 def _async_get_diagnostics(hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry | None = None) -> dict[str, Any]: ...
+@callback
 def _async_device_as_dict(hass: HomeAssistant, device: DeviceEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/mqtt/discovery.pyi

@@ -39,20 +39,30 @@ class MQTTDiscoveryPayload(dict[str, Any]):
 class MQTTIntegrationDiscoveryConfig:
     integration: str
     msg: ReceiveMessage
-    def __init__(self, integration, msg) -> None: ...
 
+@callback
 def _async_process_discovery_migration(payload: MQTTDiscoveryPayload) -> bool: ...
 def clear_discovery_hash(hass: HomeAssistant, discovery_hash: tuple[str, str]) -> None: ...
 def set_discovery_hash(hass: HomeAssistant, discovery_hash: tuple[str, str]) -> None: ...
+@callback
 def get_origin_log_string(discovery_payload: MQTTDiscoveryPayload, *, include_url: bool) -> str: ...
+@callback
 def get_origin_support_url(discovery_payload: MQTTDiscoveryPayload) -> str | None: ...
+@callback
 def async_log_discovery_origin_info(message: str, discovery_payload: MQTTDiscoveryPayload, level: int = ...) -> None: ...
+@callback
 def _replace_abbreviations(payload: dict[str, Any] | str, abbreviations: dict[str, str], abbreviations_set: set[str]) -> None: ...
+@callback
 def _replace_all_abbreviations(discovery_payload: dict[str, Any], component_only: bool = False) -> None: ...
+@callback
 def _replace_topic_base(discovery_payload: MQTTDiscoveryPayload) -> None: ...
+@callback
 def _generate_device_config(hass: HomeAssistant, object_id: str, node_id: str | None, migrate_discovery: bool = False) -> MQTTDiscoveryPayload: ...
+@callback
 def _parse_device_payload(hass: HomeAssistant, payload: ReceivePayloadType, object_id: str, node_id: str | None) -> MQTTDiscoveryPayload: ...
+@callback
 def _valid_origin_info(discovery_payload: MQTTDiscoveryPayload) -> bool: ...
+@callback
 def _merge_common_device_options(component_config: MQTTDiscoveryPayload, device_config: dict[str, Any]) -> None: ...
 async def async_start(hass: HomeAssistant, discovery_topic: str, config_entry: ConfigEntry) -> None: ...
 async def async_stop(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/mqtt/entity.pyi

@@ -24,11 +24,12 @@ from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity,
 from homeassistant.helpers.service_info.mqtt import ReceivePayloadType as ReceivePayloadType
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, UNDEFINED as UNDEFINED, UndefinedType as UndefinedType, VolSchemaType as VolSchemaType
 from homeassistant.util.json import json_loads as json_loads
-from typing import Any, Protocol
+from typing import Any, Protocol, final
 
 _LOGGER: Incomplete
 MQTT_ATTRIBUTES_BLOCKED: Incomplete
 
+@callback
 def async_handle_schema_error(discovery_payload: MQTTDiscoveryPayload, err: vol.Invalid) -> None: ...
 def _handle_discovery_failure(hass: HomeAssistant, discovery_payload: MQTTDiscoveryPayload) -> None: ...
 def _verify_mqtt_config_entry_enabled_for_discovery(hass: HomeAssistant, domain: str, discovery_payload: MQTTDiscoveryPayload) -> bool: ...
@@ -36,7 +37,9 @@ def _verify_mqtt_config_entry_enabled_for_discovery(hass: HomeAssistant, domain:
 class _SetupNonEntityHelperCallbackProtocol(Protocol):
     async def __call__(self, config: ConfigType, discovery_data: DiscoveryInfoType) -> None: ...
 
+@callback
 def async_setup_non_entity_entry_helper(hass: HomeAssistant, domain: str, async_setup: _SetupNonEntityHelperCallbackProtocol, discovery_schema: vol.Schema) -> None: ...
+@callback
 def async_setup_entity_entry_helper(hass: HomeAssistant, entry: ConfigEntry, entity_class: type[MqttEntity] | None, domain: str, async_add_entities: AddEntitiesCallback, discovery_schema: VolSchemaType, platform_schema_modern: VolSchemaType, schema_class_mapping: dict[str, type[MqttEntity]] | None = None) -> None: ...
 def init_entity_id_from_config(hass: HomeAssistant, entity: Entity, config: ConfigType, entity_id_format: str) -> None: ...
 
@@ -50,9 +53,11 @@ class MqttAttributesMixin(Entity):
     def attributes_prepare_discovery_update(self, config: DiscoveryInfoType) -> None: ...
     async def attributes_discovery_update(self, config: DiscoveryInfoType) -> None: ...
     def _attributes_prepare_subscribe_topics(self) -> None: ...
+    @callback
     def _attributes_subscribe_topics(self) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...
     _attr_extra_state_attributes: Incomplete
+    @callback
     def _attributes_message_received(self, msg: ReceiveMessage) -> None: ...
 
 class MqttAvailabilityMixin(Entity):
@@ -67,8 +72,11 @@ class MqttAvailabilityMixin(Entity):
     _avail_config: Incomplete
     def _availability_setup_from_config(self, config: ConfigType) -> None: ...
     def _availability_prepare_subscribe_topics(self) -> None: ...
+    @callback
     def _availability_message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _availability_subscribe_topics(self) -> None: ...
+    @callback
     def async_mqtt_connection_state_changed(self, state: bool) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...
     @property
@@ -93,6 +101,7 @@ class MqttDiscoveryDeviceUpdateMixin(ABC, metaclass=abc.ABCMeta):
     _remove_discovery_updated: Incomplete
     _remove_device_updated: Incomplete
     def __init__(self, hass: HomeAssistant, discovery_data: DiscoveryInfoType, device_id: str | None, config_entry: ConfigEntry, log_name: str) -> None: ...
+    @callback
     def _entry_unload(self, *_: Any) -> None: ...
     async def async_discovery_update(self, discovery_payload: MQTTDiscoveryPayload) -> None: ...
     async def _async_device_removed(self, event: Event[EventDeviceRegistryUpdatedData]) -> None: ...
@@ -114,14 +123,18 @@ class MqttDiscoveryUpdateMixin(Entity):
     async def _async_remove_state_and_registry_entry(self) -> None: ...
     async def _async_process_discovery_update(self, payload: MQTTDiscoveryPayload, discovery_update: Callable[[MQTTDiscoveryPayload], Coroutine[Any, Any, None]], discovery_data: DiscoveryInfoType) -> None: ...
     async def _async_process_discovery_update_and_remove(self) -> None: ...
+    @callback
     def _async_discovery_callback(self, payload: MQTTDiscoveryPayload) -> None: ...
     async def async_removed_from_registry(self) -> None: ...
+    @final
     async def add_to_platform_finish(self) -> None: ...
+    @callback
     def add_to_platform_abort(self) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...
     def _cleanup_discovery_on_remove(self) -> None: ...
 
 def device_info_from_specifications(specifications: dict[str, Any] | None) -> DeviceInfo | None: ...
+@callback
 def ensure_via_device_exists(hass: HomeAssistant, device_info: DeviceInfo | None, config_entry: ConfigEntry) -> None: ...
 
 class MqttEntityDeviceInfo(Entity):
@@ -146,6 +159,7 @@ class MqttEntity(MqttAttributesMixin, MqttAvailabilityMixin, MqttDiscoveryUpdate
     _subscriptions: dict[str, dict[str, Any]]
     def __init__(self, hass: HomeAssistant, config: ConfigType, config_entry: ConfigEntry, discovery_data: DiscoveryInfoType | None) -> None: ...
     def _init_entity_id(self) -> None: ...
+    @final
     async def async_added_to_hass(self) -> None: ...
     async def mqtt_async_added_to_hass(self) -> None: ...
     async def discovery_update(self, discovery_payload: MQTTDiscoveryPayload) -> None: ...
@@ -164,12 +178,16 @@ class MqttEntity(MqttAttributesMixin, MqttAvailabilityMixin, MqttDiscoveryUpdate
     def _setup_common_attributes_from_config(self, config: ConfigType) -> None: ...
     def _setup_from_config(self, config: ConfigType) -> None: ...
     @abstractmethod
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     @abstractmethod
     async def _subscribe_topics(self) -> None: ...
+    @callback
     def _attrs_have_changed(self, attrs_snapshot: tuple[tuple[str, Any | UndefinedType], ...]) -> bool: ...
+    @callback
     def _message_callback(self, msg_callback: MessageCallbackType, attributes: set[str] | None, msg: ReceiveMessage) -> None: ...
     def add_subscription(self, state_topic_config_key: str, msg_callback: Callable[[ReceiveMessage], None], tracked_attributes: set[str] | None, disable_encoding: bool = False) -> bool: ...
 
 def update_device(hass: HomeAssistant, config_entry: ConfigEntry, config: ConfigType) -> str | None: ...
+@callback
 def async_removed_from_device(hass: HomeAssistant, event: Event[EventDeviceRegistryUpdatedData], mqtt_device_id: str, config_entry_id: str) -> bool: ...

homeassistant-stubs/components/mqtt/event.pyi

@@ -39,6 +39,8 @@ class MqttEvent(MqttEntity, EventEntity):
     _attr_device_class: Incomplete
     _attr_event_types: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
+    @callback
     def _event_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...

homeassistant-stubs/components/mqtt/fan.pyi

@@ -90,13 +90,19 @@ class MqttFan(MqttEntity, FanEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_is_on: bool
+    @callback
     def _state_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _percentage_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _preset_mode_received(self, msg: ReceiveMessage) -> None: ...
     _attr_oscillating: bool
+    @callback
     def _oscillation_received(self, msg: ReceiveMessage) -> None: ...
     _attr_current_direction: Incomplete
+    @callback
     def _direction_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     @property

homeassistant-stubs/components/mqtt/humidifier.pyi

@@ -72,14 +72,20 @@ class MqttHumidifier(MqttEntity, HumidifierEntity):
     _attr_assumed_state: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_is_on: bool
+    @callback
     def _state_received(self, msg: ReceiveMessage) -> None: ...
     _attr_action: Incomplete
+    @callback
     def _action_received(self, msg: ReceiveMessage) -> None: ...
     _attr_current_humidity: Incomplete
+    @callback
     def _current_humidity_received(self, msg: ReceiveMessage) -> None: ...
     _attr_target_humidity: Incomplete
+    @callback
     def _target_humidity_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _mode_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     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

@@ -50,9 +50,12 @@ class MqttImage(MqttEntity, ImageEntity):
     _attr_image_url: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_image_last_updated: Incomplete
+    @callback
     def _image_data_received(self, msg: ReceiveMessage) -> None: ...
     _cached_image: Incomplete
+    @callback
     def _image_from_url_request_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     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

@@ -51,7 +51,9 @@ class MqttLawnMower(MqttEntity, LawnMowerEntity, RestoreEntity):
     _attr_supported_features: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_activity: Incomplete
+    @callback
     def _message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     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

@@ -101,25 +101,37 @@ class MqttLight(MqttEntity, LightEntity, RestoreEntity):
     def _setup_from_config(self, config: ConfigType) -> None: ...
     def _is_optimistic(self, attribute: str) -> bool: ...
     _attr_is_on: bool
+    @callback
     def _state_received(self, msg: ReceiveMessage) -> None: ...
     _attr_brightness: Incomplete
+    @callback
     def _brightness_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _rgbx_received(self, msg: ReceiveMessage, template: str, color_mode: ColorMode, convert_color: Callable[..., tuple[int, ...]]) -> tuple[int, ...] | None: ...
     _attr_rgb_color: Incomplete
+    @callback
     def _rgb_received(self, msg: ReceiveMessage) -> None: ...
     _attr_rgbw_color: Incomplete
+    @callback
     def _rgbw_received(self, msg: ReceiveMessage) -> None: ...
     _attr_rgbww_color: Incomplete
+    @callback
     def _rgbww_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _color_mode_received(self, msg: ReceiveMessage) -> None: ...
     _attr_color_temp_kelvin: Incomplete
+    @callback
     def _color_temp_received(self, msg: ReceiveMessage) -> None: ...
     _attr_effect: Incomplete
+    @callback
     def _effect_received(self, msg: ReceiveMessage) -> None: ...
     _attr_hs_color: Incomplete
+    @callback
     def _hs_received(self, msg: ReceiveMessage) -> None: ...
     _attr_xy_color: Incomplete
+    @callback
     def _xy_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     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

@@ -76,7 +76,9 @@ class MqttLightJson(MqttEntity, LightEntity, RestoreEntity):
     _attr_is_on: bool
     _attr_brightness: Incomplete
     _attr_effect: Incomplete
+    @callback
     def _state_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     @property

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

@@ -60,7 +60,9 @@ class MqttLightTemplate(MqttEntity, LightEntity, RestoreEntity):
     _attr_color_temp_kelvin: Incomplete
     _attr_hs_color: Incomplete
     _attr_effect: Incomplete
+    @callback
     def _state_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     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

@@ -69,7 +69,9 @@ class MqttLock(MqttEntity, LockEntity):
     _attr_is_opening: Incomplete
     _attr_is_unlocking: Incomplete
     _attr_is_jammed: Incomplete
+    @callback
     def _message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_lock(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/mqtt/models.pyi

@@ -8,7 +8,7 @@ from .tag import MQTTTagScanner as MQTTTagScanner
 from _typeshed import Incomplete
 from collections import deque
 from collections.abc import Callable
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from enum import StrEnum
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_NAME as ATTR_NAME, Platform as Platform
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, callback as callback
@@ -37,7 +37,6 @@ class PublishMessage:
     payload: PublishPayloadType
     qos: int
     retain: bool
-    def __init__(self, topic, payload, qos, retain) -> None: ...
 
 @dataclass(slots=True, frozen=True, eq=False)
 class ReceiveMessage:
@@ -47,7 +46,6 @@ class ReceiveMessage:
     retain: bool
     subscribed_topic: str
     timestamp: float
-    def __init__(self, topic, payload, qos, retain, subscribed_topic, timestamp) -> None: ...
 type MessageCallbackType = Callable[[ReceiveMessage], None]
 
 class SubscriptionDebugInfo(TypedDict):
@@ -87,6 +85,7 @@ class MqttCommandTemplate:
     _command_template: Incomplete
     _entity: Incomplete
     def __init__(self, command_template: template.Template | None, *, entity: Entity | None = None) -> None: ...
+    @callback
     def async_render(self, value: PublishPayloadType = None, variables: TemplateVarsType = None) -> PublishPayloadType: ...
 
 class MqttValueTemplateException(TemplateError):
@@ -100,36 +99,38 @@ class MqttValueTemplate:
     _config_attributes: Incomplete
     _entity: Incomplete
     def __init__(self, value_template: template.Template | None, *, entity: Entity | None = None, config_attributes: TemplateVarsType = None) -> None: ...
+    @callback
     def async_render_with_possible_json_value(self, payload: ReceivePayloadType, default: ReceivePayloadType | PayloadSentinel = ..., variables: TemplateVarsType = None) -> ReceivePayloadType: ...
 
 class EntityTopicState:
     subscribe_calls: dict[str, Entity]
     def __init__(self) -> None: ...
+    @callback
     def process_write_state_requests(self, msg: MQTTMessage) -> None: ...
+    @callback
     def write_state_request(self, entity: Entity) -> None: ...
 
 @dataclass
 class MqttData:
     client: MQTT
     config: list[ConfigType]
-    debug_info_entities: dict[str, EntityDebugInfo] = ...
-    debug_info_triggers: dict[tuple[str, str], TriggerDebugInfo] = ...
-    device_triggers: dict[str, Trigger] = ...
-    data_config_flow_lock: asyncio.Lock = ...
-    discovery_already_discovered: set[tuple[str, str]] = ...
-    discovery_pending_discovered: dict[tuple[str, str], PendingDiscovered] = ...
-    discovery_registry_hooks: dict[tuple[str, str], CALLBACK_TYPE] = ...
-    discovery_unsubscribe: list[CALLBACK_TYPE] = ...
-    integration_unsubscribe: dict[str, CALLBACK_TYPE] = ...
+    debug_info_entities: dict[str, EntityDebugInfo] = field(default_factory=dict)
+    debug_info_triggers: dict[tuple[str, str], TriggerDebugInfo] = field(default_factory=dict)
+    device_triggers: dict[str, Trigger] = field(default_factory=dict)
+    data_config_flow_lock: asyncio.Lock = field(default_factory=asyncio.Lock)
+    discovery_already_discovered: set[tuple[str, str]] = field(default_factory=set)
+    discovery_pending_discovered: dict[tuple[str, str], PendingDiscovered] = field(default_factory=dict)
+    discovery_registry_hooks: dict[tuple[str, str], CALLBACK_TYPE] = field(default_factory=dict)
+    discovery_unsubscribe: list[CALLBACK_TYPE] = field(default_factory=list)
+    integration_unsubscribe: dict[str, CALLBACK_TYPE] = field(default_factory=dict)
     last_discovery: float = ...
-    platforms_loaded: set[Platform | str] = ...
-    reload_dispatchers: list[CALLBACK_TYPE] = ...
-    reload_handlers: dict[str, CALLBACK_TYPE] = ...
-    reload_schema: dict[str, VolSchemaType] = ...
-    state_write_requests: EntityTopicState = ...
-    subscriptions_to_restore: set[Subscription] = ...
-    tags: dict[str, dict[str, MQTTTagScanner]] = ...
-    def __init__(self, client, config, debug_info_entities=..., debug_info_triggers=..., device_triggers=..., data_config_flow_lock=..., discovery_already_discovered=..., discovery_pending_discovered=..., discovery_registry_hooks=..., discovery_unsubscribe=..., integration_unsubscribe=..., last_discovery=..., platforms_loaded=..., reload_dispatchers=..., reload_handlers=..., reload_schema=..., state_write_requests=..., subscriptions_to_restore=..., tags=...) -> None: ...
+    platforms_loaded: set[Platform | str] = field(default_factory=set)
+    reload_dispatchers: list[CALLBACK_TYPE] = field(default_factory=list)
+    reload_handlers: dict[str, CALLBACK_TYPE] = field(default_factory=dict)
+    reload_schema: dict[str, VolSchemaType] = field(default_factory=dict)
+    state_write_requests: EntityTopicState = field(default_factory=EntityTopicState)
+    subscriptions_to_restore: set[Subscription] = field(default_factory=set)
+    tags: dict[str, dict[str, MQTTTagScanner]] = field(default_factory=dict)
 
 @dataclass(slots=True)
 class MqttComponentConfig:
@@ -137,7 +138,6 @@ class MqttComponentConfig:
     object_id: str
     node_id: str | None
     discovery_payload: MQTTDiscoveryPayload
-    def __init__(self, component, object_id, node_id, discovery_payload) -> None: ...
 
 DATA_MQTT: HassKey[MqttData]
 DATA_MQTT_AVAILABLE: HassKey[asyncio.Future[bool]]

homeassistant-stubs/components/mqtt/notify.pyi

@@ -28,6 +28,7 @@ class MqttNotify(MqttEntity, NotifyEntity):
     def config_schema() -> vol.Schema: ...
     _command_template: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_send_message(self, message: str, title: str | None = None) -> None: ...

homeassistant-stubs/components/mqtt/number.pyi

@@ -51,7 +51,9 @@ class MqttNumber(MqttEntity, RestoreNumber):
     _attr_native_unit_of_measurement: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
     _attr_native_value: Incomplete
+    @callback
     def _message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_set_native_value(self, value: float) -> None: ...

homeassistant-stubs/components/mqtt/scene.pyi

@@ -29,6 +29,7 @@ class MqttScene(MqttEntity, Scene):
     @staticmethod
     def config_schema() -> vol.Schema: ...
     def _setup_from_config(self, config: ConfigType) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_activate(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/mqtt/select.pyi

@@ -38,7 +38,9 @@ class MqttSelect(MqttEntity, SelectEntity, RestoreEntity):
     _attr_assumed_state: Incomplete
     _attr_options: Incomplete
     def _setup_from_config(self, config: ConfigType) -> None: ...
+    @callback
     def _message_received(self, msg: ReceiveMessage) -> None: ...
+    @callback
     def _prepare_subscribe_topics(self) -> None: ...
     async def _subscribe_topics(self) -> None: ...
     async def async_select_option(self, option: str) -> None: ...