From ddc260b6289c5482a7f8802d4cc3d5120c05b5e7 Mon Sep 17 00:00:00 2001
From: Robbie Trencheny <me@robbiet.us>
Date: Wed, 8 Mar 2017 23:57:54 -0800
Subject: [PATCH] Add required changes to support MQTT discovery that I
 previously missed. See #6481.

---
 homeassistant/components/light/mqtt.py          | 3 +++
 homeassistant/components/light/mqtt_json.py     | 2 ++
 homeassistant/components/light/mqtt_template.py | 3 +++
 3 files changed, 8 insertions(+)

diff --git a/homeassistant/components/light/mqtt.py b/homeassistant/components/light/mqtt.py
index 4740858b5fb..018d7a40639 100644
--- a/homeassistant/components/light/mqtt.py
+++ b/homeassistant/components/light/mqtt.py
@@ -93,6 +93,9 @@ PLATFORM_SCHEMA = mqtt.MQTT_RW_PLATFORM_SCHEMA.extend({
 @asyncio.coroutine
 def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
     """Add MQTT Light."""
+    if discovery_info is not None:
+        config = PLATFORM_SCHEMA(discovery_info)
+
     config.setdefault(
         CONF_STATE_VALUE_TEMPLATE, config.get(CONF_VALUE_TEMPLATE))
 
diff --git a/homeassistant/components/light/mqtt_json.py b/homeassistant/components/light/mqtt_json.py
index 3ec73db87aa..969ac1ccdda 100755
--- a/homeassistant/components/light/mqtt_json.py
+++ b/homeassistant/components/light/mqtt_json.py
@@ -72,6 +72,8 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
 @asyncio.coroutine
 def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
     """Setup a MQTT JSON Light."""
+    if discovery_info is not None:
+        config = PLATFORM_SCHEMA(discovery_info)
     async_add_devices([MqttJson(
         config.get(CONF_NAME),
         config.get(CONF_EFFECT_LIST),
diff --git a/homeassistant/components/light/mqtt_template.py b/homeassistant/components/light/mqtt_template.py
index d2d9c243c03..fa0d71f46ef 100755
--- a/homeassistant/components/light/mqtt_template.py
+++ b/homeassistant/components/light/mqtt_template.py
@@ -66,6 +66,9 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
 @asyncio.coroutine
 def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
     """Setup a MQTT Template light."""
+    if discovery_info is not None:
+        config = PLATFORM_SCHEMA(discovery_info)
+
     async_add_devices([MqttTemplate(
         hass,
         config.get(CONF_NAME),