Add MQTT notify platform

homeassistant/components/mqtt/__init__.py

@@ -154,6 +154,7 @@
     Platform.HUMIDIFIER,
     Platform.LIGHT,
     Platform.LOCK,
+    Platform.NOTIFY,
     Platform.NUMBER,
     Platform.SELECT,
     Platform.SCENE,

homeassistant/components/mqtt/abbreviations.py

@@ -183,6 +183,8 @@
     "set_fan_spd_t": "set_fan_speed_topic",
     "set_pos_tpl": "set_position_template",
     "set_pos_t": "set_position_topic",
+    "title": "title",
+    "trgts": "targets",
     "pos_t": "position_topic",
     "pos_tpl": "position_template",
     "spd_cmd_t": "speed_command_topic",

homeassistant/components/mqtt/const.py

@@ -1,4 +1,6 @@
 """Constants used by multiple MQTT modules."""
+from typing import Final
+
 from homeassistant.const import CONF_PAYLOAD
 
 ATTR_DISCOVERY_HASH = "discovery_hash"
@@ -12,11 +14,11 @@
 CONF_AVAILABILITY = "availability"
 CONF_BROKER = "broker"
 CONF_BIRTH_MESSAGE = "birth_message"
-CONF_COMMAND_TEMPLATE = "command_template"
-CONF_COMMAND_TOPIC = "command_topic"
-CONF_ENCODING = "encoding"
-CONF_QOS = ATTR_QOS
-CONF_RETAIN = ATTR_RETAIN
+CONF_COMMAND_TEMPLATE: Final = "command_template"
+CONF_COMMAND_TOPIC: Final = "command_topic"
+CONF_ENCODING: Final = "encoding"
+CONF_QOS: Final = "qos"
+CONF_RETAIN: Final = "retain"
 CONF_STATE_TOPIC = "state_topic"
 CONF_STATE_VALUE_TEMPLATE = "state_value_template"
 CONF_TOPIC = "topic"

homeassistant/components/mqtt/discovery.py

@@ -15,6 +15,7 @@
     async_dispatcher_connect,
     async_dispatcher_send,
 )
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.loader import async_get_mqtt
 
 from .. import mqtt
@@ -48,6 +49,7 @@
     "humidifier",
     "light",
     "lock",
+    "notify",
     "number",
     "scene",
     "siren",
@@ -232,7 +234,15 @@ async def discovery_done(_):
                         from . import device_automation
 
                         await device_automation.async_setup_entry(hass, config_entry)
-                    elif component == "tag":
+                    elif component in "notify":
+                        # Local import to avoid circular dependencies
+                        # pylint: disable=import-outside-toplevel
+                        from . import notify
+
+                        await notify.async_setup_entry(
+                            hass, config_entry, AddEntitiesCallback
+                        )
+                    elif component in "tag":
                         # Local import to avoid circular dependencies
                         # pylint: disable-next=import-outside-toplevel
                         from . import tag

homeassistant/components/mqtt/mixins.py

@@ -5,7 +5,7 @@
 from collections.abc import Callable
 import json
 import logging
-from typing import Any, Protocol
+from typing import Any, Protocol, cast
 
 import voluptuous as vol
 
@@ -237,10 +237,10 @@ async def __call__(
 
 
 async def async_setup_entry_helper(hass, domain, async_setup, schema):
-    """Set up entity, automation or tag creation dynamically through MQTT discovery."""
+    """Set up entity, automation, notify service or tag creation dynamically through MQTT discovery."""
 
     async def async_discover(discovery_payload):
-        """Discover and add an MQTT entity, automation or tag."""
+        """Discover and add an MQTT entity, automation, notify service or tag."""
         discovery_data = discovery_payload.discovery_data
         try:
             config = schema(discovery_payload)
@@ -496,11 +496,13 @@ def available(self) -> bool:
         return self._available_latest
 
 
-async def cleanup_device_registry(hass, device_id, config_entry_id):
-    """Remove device registry entry if there are no remaining entities or triggers."""
+async def cleanup_device_registry(
+    hass: HomeAssistant, device_id: str | None, config_entry_id: str | None
+) -> None:
+    """Remove device registry entry if there are no remaining entities, triggers or notify services."""
     # Local import to avoid circular dependencies
-    # pylint: disable-next=import-outside-toplevel
-    from . import device_trigger, tag
+    # pylint: disable=import-outside-toplevel
+    from . import device_trigger, notify, tag
 
     device_registry = dr.async_get(hass)
     entity_registry = er.async_get(hass)
@@ -511,9 +513,10 @@ async def cleanup_device_registry(hass, device_id, config_entry_id):
         )
         and not await device_trigger.async_get_triggers(hass, device_id)
         and not tag.async_has_tags(hass, device_id)
+        and not notify.device_has_notify_services(hass, device_id)
     ):
         device_registry.async_update_device(
-            device_id, remove_config_entry_id=config_entry_id
+            device_id, remove_config_entry_id=cast(str, config_entry_id)
         )