Add unique_id to MQTT switch (#13719)

This commit is contained in:
thepotoo 2018-05-06 02:21:02 -04:00 committed by Sebastian Muszynski
parent f3411f8db2
commit 2bb1a95098
2 changed files with 35 additions and 1 deletions

View file

@ -5,6 +5,7 @@ For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/switch.mqtt/
"""
import logging
from typing import Optional
import voluptuous as vol
@ -29,12 +30,14 @@ DEFAULT_NAME = 'MQTT Switch'
DEFAULT_PAYLOAD_ON = 'ON'
DEFAULT_PAYLOAD_OFF = 'OFF'
DEFAULT_OPTIMISTIC = False
CONF_UNIQUE_ID = 'unique_id'
PLATFORM_SCHEMA = mqtt.MQTT_RW_PLATFORM_SCHEMA.extend({
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_ICON): cv.icon,
vol.Optional(CONF_PAYLOAD_ON, default=DEFAULT_PAYLOAD_ON): cv.string,
vol.Optional(CONF_PAYLOAD_OFF, default=DEFAULT_PAYLOAD_OFF): cv.string,
vol.Optional(CONF_UNIQUE_ID): cv.string,
vol.Optional(CONF_OPTIMISTIC, default=DEFAULT_OPTIMISTIC): cv.boolean,
}).extend(mqtt.MQTT_AVAILABILITY_SCHEMA.schema)
@ -62,6 +65,7 @@ async def async_setup_platform(hass, config, async_add_devices,
config.get(CONF_OPTIMISTIC),
config.get(CONF_PAYLOAD_AVAILABLE),
config.get(CONF_PAYLOAD_NOT_AVAILABLE),
config.get(CONF_UNIQUE_ID),
value_template,
)])
@ -72,7 +76,8 @@ class MqttSwitch(MqttAvailability, SwitchDevice):
def __init__(self, name, icon,
state_topic, command_topic, availability_topic,
qos, retain, payload_on, payload_off, optimistic,
payload_available, payload_not_available, value_template):
payload_available, payload_not_available,
unique_id: Optional[str], value_template):
"""Initialize the MQTT switch."""
super().__init__(availability_topic, qos, payload_available,
payload_not_available)
@ -87,6 +92,7 @@ class MqttSwitch(MqttAvailability, SwitchDevice):
self._payload_off = payload_off
self._optimistic = optimistic
self._template = value_template
self._unique_id = unique_id
async def async_added_to_hass(self):
"""Subscribe to MQTT events."""
@ -139,6 +145,11 @@ class MqttSwitch(MqttAvailability, SwitchDevice):
"""Return true if we do optimistic updates."""
return self._optimistic
@property
def unique_id(self):
"""Return a unique ID."""
return self._unique_id
@property
def icon(self):
"""Return the icon."""