Added custom validator function for speed list

homeassistant/components/mqtt/fan.py

@@ -1,4 +1,5 @@
 """Support for MQTT fans."""
+import copy
 import logging
 
 import voluptuous as vol
@@ -69,7 +70,19 @@ OSCILLATE_OFF_PAYLOAD = "oscillate_off"
 
 OSCILLATION = "oscillation"
 
-PLATFORM_SCHEMA = (
+
+def _speed_list_validator(config):
+    """Validate the speed list."""
+    config = copy.deepcopy(config)
+
+    if CONF_SPEED_COMMAND_TOPIC not in config or CONF_SPEED_STATE_TOPIC not in config:
+        config[CONF_SPEED_LIST] = None
+
+    return config
+
+
+PLATFORM_SCHEMA = vol.Schema(
+    vol.All(
         mqtt.MQTT_RW_PLATFORM_SCHEMA.extend(
             {
                 vol.Optional(CONF_DEVICE): mqtt.MQTT_ENTITY_DEVICE_INFO_SCHEMA,
@@ -80,7 +93,9 @@ PLATFORM_SCHEMA = (
                 vol.Optional(CONF_OSCILLATION_VALUE_TEMPLATE): cv.template,
                 vol.Optional(CONF_PAYLOAD_HIGH_SPEED, default=SPEED_HIGH): cv.string,
                 vol.Optional(CONF_PAYLOAD_LOW_SPEED, default=SPEED_LOW): cv.string,
-            vol.Optional(CONF_PAYLOAD_MEDIUM_SPEED, default=SPEED_MEDIUM): cv.string,
+                vol.Optional(
+                    CONF_PAYLOAD_MEDIUM_SPEED, default=SPEED_MEDIUM
+                ): cv.string,
                 vol.Optional(CONF_PAYLOAD_OFF_SPEED, default=SPEED_OFF): cv.string,
                 vol.Optional(CONF_PAYLOAD_OFF, default=DEFAULT_PAYLOAD_OFF): cv.string,
                 vol.Optional(CONF_PAYLOAD_ON, default=DEFAULT_PAYLOAD_ON): cv.string,
@@ -102,7 +117,9 @@ PLATFORM_SCHEMA = (
             }
         )
         .extend(mqtt.MQTT_AVAILABILITY_SCHEMA.schema)
-    .extend(mqtt.MQTT_JSON_ATTRS_SCHEMA.schema)
+        .extend(mqtt.MQTT_JSON_ATTRS_SCHEMA.schema),
+        _speed_list_validator,
+    )
 )