Migrate to voluptuous (#3234)

homeassistant/components/switch/zigbee.py

@@ -4,18 +4,29 @@ Contains functionality to use a ZigBee device as a switch.
 For more details about this platform, please refer to the documentation at
 https://home-assistant.io/components/switch.zigbee/
 """
+import voluptuous as vol
+
 from homeassistant.components.switch import SwitchDevice
 from homeassistant.components.zigbee import (
-    ZigBeeDigitalOut, ZigBeeDigitalOutConfig)
+    ZigBeeDigitalOut, ZigBeeDigitalOutConfig, PLATFORM_SCHEMA)
 
-DEPENDENCIES = ["zigbee"]
+DEPENDENCIES = ['zigbee']
+
+CONF_ON_STATE = 'on_state'
+
+DEFAULT_ON_STATE = 'high'
+DEPENDENCIES = ['zigbee']
+
+STATES = ['high', 'low']
+
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Optional(CONF_ON_STATE): vol.In(STATES),
+})
 
 
-def setup_platform(hass, config, add_entities, discovery_info=None):
+def setup_platform(hass, config, add_devices, discovery_info=None):
     """Setup the ZigBee switch platform."""
-    add_entities([
-        ZigBeeSwitch(hass, ZigBeeDigitalOutConfig(config))
-    ])
+    add_devices([ZigBeeSwitch(hass, ZigBeeDigitalOutConfig(config))])
 
 
 class ZigBeeSwitch(ZigBeeDigitalOut, SwitchDevice):