From 7e0d9bc7095d9ba47def6f54f2970b35e7bf30ec Mon Sep 17 00:00:00 2001 From: Jan Harkes Date: Tue, 12 Apr 2016 23:04:06 -0400 Subject: [PATCH] Service validation for input_boolean component. --- homeassistant/components/input_boolean.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/input_boolean.py b/homeassistant/components/input_boolean.py index 75d5d940f47..8702fd6eae2 100644 --- a/homeassistant/components/input_boolean.py +++ b/homeassistant/components/input_boolean.py @@ -6,8 +6,11 @@ at https://home-assistant.io/components/input_boolean/ """ import logging +import voluptuous as vol + from homeassistant.const import ( ATTR_ENTITY_ID, SERVICE_TURN_OFF, SERVICE_TURN_ON, STATE_ON) +import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity import ToggleEntity from homeassistant.helpers.entity_component import EntityComponent from homeassistant.util import slugify @@ -22,6 +25,10 @@ CONF_NAME = "name" CONF_INITIAL = "initial" CONF_ICON = "icon" +TOGGLE_SERVICE_SCHEMA = vol.Schema({ + vol.Optional(ATTR_ENTITY_ID): cv.entity_ids, +}) + def is_on(hass, entity_id): """Test if input_boolean is True.""" @@ -75,8 +82,10 @@ def setup(hass, config): else: input_b.turn_off() - hass.services.register(DOMAIN, SERVICE_TURN_OFF, toggle_service) - hass.services.register(DOMAIN, SERVICE_TURN_ON, toggle_service) + hass.services.register(DOMAIN, SERVICE_TURN_OFF, toggle_service, + schema=TOGGLE_SERVICE_SCHEMA) + hass.services.register(DOMAIN, SERVICE_TURN_ON, toggle_service, + schema=TOGGLE_SERVICE_SCHEMA) component.add_entities(entities)