From 6b787ee01e88a55afa2c21195fdd7df9933a771d Mon Sep 17 00:00:00 2001 From: Fabian Affolter Date: Mon, 5 Sep 2016 18:05:27 +0200 Subject: [PATCH] Migrate to voluptuous (#3209) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🐬 --- homeassistant/components/media_player/roku.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/media_player/roku.py b/homeassistant/components/media_player/roku.py index 7951530e3e8..e7a87d2d773 100644 --- a/homeassistant/components/media_player/roku.py +++ b/homeassistant/components/media_player/roku.py @@ -6,13 +6,15 @@ https://home-assistant.io/components/media_player.roku/ """ import logging +import voluptuous as vol + from homeassistant.components.media_player import ( MEDIA_TYPE_VIDEO, SUPPORT_NEXT_TRACK, SUPPORT_PLAY_MEDIA, SUPPORT_PREVIOUS_TRACK, SUPPORT_VOLUME_MUTE, SUPPORT_VOLUME_SET, - SUPPORT_SELECT_SOURCE, MediaPlayerDevice) - + SUPPORT_SELECT_SOURCE, MediaPlayerDevice, PLATFORM_SCHEMA) from homeassistant.const import ( CONF_HOST, STATE_IDLE, STATE_PLAYING, STATE_UNKNOWN, STATE_HOME) +import homeassistant.helpers.config_validation as cv REQUIREMENTS = [ 'https://github.com/bah2830/python-roku/archive/3.1.2.zip' @@ -27,6 +29,10 @@ SUPPORT_ROKU = SUPPORT_PREVIOUS_TRACK | SUPPORT_NEXT_TRACK |\ SUPPORT_PLAY_MEDIA | SUPPORT_VOLUME_SET | SUPPORT_VOLUME_MUTE |\ SUPPORT_SELECT_SOURCE +PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ + vol.Optional(CONF_HOST): cv.string, +}) + # pylint: disable=abstract-method def setup_platform(hass, config, add_devices, discovery_info=None): @@ -41,7 +47,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None): hosts.append(discovery_info[0]) elif CONF_HOST in config: - hosts.append(config[CONF_HOST]) + hosts.append(config.get(CONF_HOST)) rokus = [] for host in hosts: