From a99f36f519e4e1de1164b6846ef35d27dbd8cc4e Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Tue, 11 Oct 2016 09:56:57 +0200
Subject: [PATCH] Migrate to voluptuous (#3737)

---
 homeassistant/components/sensor/arduino.py | 36 +++++++++++++------
 homeassistant/components/switch/arduino.py | 41 ++++++++++++++++------
 2 files changed, 56 insertions(+), 21 deletions(-)

diff --git a/homeassistant/components/sensor/arduino.py b/homeassistant/components/sensor/arduino.py
index 203848fbe6e..03307a49768 100644
--- a/homeassistant/components/sensor/arduino.py
+++ b/homeassistant/components/sensor/arduino.py
@@ -8,28 +8,44 @@ https://home-assistant.io/components/sensor.arduino/
 """
 import logging
 
+import voluptuous as vol
+
+from homeassistant.components.sensor import PLATFORM_SCHEMA
 import homeassistant.components.arduino as arduino
-from homeassistant.const import DEVICE_DEFAULT_NAME
+from homeassistant.const import CONF_NAME
 from homeassistant.helpers.entity import Entity
+import homeassistant.helpers.config_validation as cv
+
+
+_LOGGER = logging.getLogger(__name__)
+
+CONF_PINS = 'pins'
+CONF_TYPE = 'analog'
 
 DEPENDENCIES = ['arduino']
-_LOGGER = logging.getLogger(__name__)
+
+PIN_SCHEMA = vol.Schema({
+    vol.Required(CONF_NAME): cv.string,
+})
+
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Required(CONF_PINS):
+        vol.Schema({cv.positive_int: PIN_SCHEMA}),
+})
 
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
-    """Setup the Arduino platform."""
+    """Set up the Arduino platform."""
     # Verify that the Arduino board is present
     if arduino.BOARD is None:
-        _LOGGER.error('A connection has not been made to the Arduino board.')
+        _LOGGER.error("A connection has not been made to the Arduino board")
         return False
 
+    pins = config.get(CONF_PINS)
+
     sensors = []
-    pins = config.get('pins')
     for pinnum, pin in pins.items():
-        if pin.get('name'):
-            sensors.append(ArduinoSensor(pin.get('name'),
-                                         pinnum,
-                                         'analog'))
+        sensors.append(ArduinoSensor(pin.get(CONF_NAME), pinnum, CONF_TYPE))
     add_devices(sensors)
 
 
@@ -39,7 +55,7 @@ class ArduinoSensor(Entity):
     def __init__(self, name, pin, pin_type):
         """Initialize the sensor."""
         self._pin = pin
-        self._name = name or DEVICE_DEFAULT_NAME
+        self._name = name
         self.pin_type = pin_type
         self.direction = 'in'
         self._value = None
diff --git a/homeassistant/components/switch/arduino.py b/homeassistant/components/switch/arduino.py
index 46e6baf8943..3aa61feffc8 100644
--- a/homeassistant/components/switch/arduino.py
+++ b/homeassistant/components/switch/arduino.py
@@ -8,27 +8,46 @@ https://home-assistant.io/components/switch.arduino/
 """
 import logging
 
+import voluptuous as vol
+
 import homeassistant.components.arduino as arduino
-from homeassistant.components.switch import SwitchDevice
-from homeassistant.const import DEVICE_DEFAULT_NAME
+from homeassistant.components.switch import (SwitchDevice, PLATFORM_SCHEMA)
+from homeassistant.const import CONF_NAME
+import homeassistant.helpers.config_validation as cv
 
 DEPENDENCIES = ['arduino']
 
 _LOGGER = logging.getLogger(__name__)
 
+CONF_PINS = 'pins'
+CONF_TYPE = 'digital'
+CONF_NEGATE = 'negate'
+CONF_INITIAL = 'initial'
+
+PIN_SCHEMA = vol.Schema({
+    vol.Required(CONF_NAME): cv.string,
+    vol.Optional(CONF_INITIAL, default=False): cv.boolean,
+    vol.Optional(CONF_NEGATE, default=False): cv.boolean,
+})
+
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Required(CONF_PINS, default={}):
+        vol.Schema({cv.positive_int: PIN_SCHEMA}),
+})
+
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
-    """Setup the Arduino platform."""
+    """Set up the Arduino platform."""
     # Verify that Arduino board is present
     if arduino.BOARD is None:
-        _LOGGER.error('A connection has not been made to the Arduino board.')
+        _LOGGER.error("A connection has not been made to the Arduino board")
         return False
 
+    pins = config.get(CONF_PINS)
+
     switches = []
-    pins = config.get('pins')
     for pinnum, pin in pins.items():
-        if pin.get('name'):
-            switches.append(ArduinoSwitch(pinnum, pin))
+        switches.append(ArduinoSwitch(pinnum, pin))
     add_devices(switches)
 
 
@@ -38,13 +57,13 @@ class ArduinoSwitch(SwitchDevice):
     def __init__(self, pin, options):
         """Initialize the Pin."""
         self._pin = pin
-        self._name = options.get('name') or DEVICE_DEFAULT_NAME
-        self.pin_type = options.get('type')
+        self._name = options.get(CONF_NAME)
+        self.pin_type = CONF_TYPE
         self.direction = 'out'
 
-        self._state = options.get('initial', False)
+        self._state = options.get(CONF_INITIAL)
 
-        if options.get('negate', False):
+        if options.get(CONF_NEGATE):
             self.turn_on_handler = arduino.BOARD.set_digital_out_low
             self.turn_off_handler = arduino.BOARD.set_digital_out_high
         else: