jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-18 02:37:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add icons to scripts (#31899)

Browse Source
This commit is contained in:
Paulus Schoutsen 2020-02-17 08:44:36 -08:00 committed by GitHub
parent c788946430
commit 1b7dd6c603
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 37 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
homeassistant/components/script/__init__.py
View File

@ -9,6 +9,7 @@ from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
ATTR_NAME, ATTR_NAME,
CONF_ALIAS, CONF_ALIAS,
CONF_ICON,
EVENT_SCRIPT_STARTED, EVENT_SCRIPT_STARTED,
SERVICE_RELOAD, SERVICE_RELOAD,
SERVICE_TOGGLE, SERVICE_TOGGLE,
@ -42,7 +43,8 @@ ENTITY_ID_FORMAT = DOMAIN + ".{}"
SCRIPT_ENTRY_SCHEMA = vol.Schema( SCRIPT_ENTRY_SCHEMA = vol.Schema(
{ {
CONF_ALIAS: cv.string, vol.Optional(CONF_ALIAS): cv.string,
vol.Optional(CONF_ICON): cv.icon,
vol.Required(CONF_SEQUENCE): cv.SCRIPT_SCHEMA, vol.Required(CONF_SEQUENCE): cv.SCRIPT_SCHEMA,
vol.Optional(CONF_DESCRIPTION, default=""): cv.string, vol.Optional(CONF_DESCRIPTION, default=""): cv.string,
vol.Optional(CONF_FIELDS, default={}): { vol.Optional(CONF_FIELDS, default={}): {
@ -207,9 +209,15 @@ async def _async_process_config(hass, config, component):
scripts = [] scripts = []
for object_id, cfg in config.get(DOMAIN, {}).items(): for object_id, cfg in config.get(DOMAIN, {}).items():
alias = cfg.get(CONF_ALIAS, object_id) scripts.append(
script = ScriptEntity(hass, object_id, alias, cfg[CONF_SEQUENCE]) ScriptEntity(
scripts.append(script) hass,
object_id,
cfg.get(CONF_ALIAS, object_id),
cfg.get(CONF_ICON),
cfg[CONF_SEQUENCE],
)
)
hass.services.async_register( hass.services.async_register(
DOMAIN, object_id, service_handler, schema=SCRIPT_SERVICE_SCHEMA DOMAIN, object_id, service_handler, schema=SCRIPT_SERVICE_SCHEMA
) )
@ -227,9 +235,12 @@ async def _async_process_config(hass, config, component):
class ScriptEntity(ToggleEntity): class ScriptEntity(ToggleEntity):
"""Representation of a script entity.""" """Representation of a script entity."""
def __init__(self, hass, object_id, name, sequence): icon = None
def __init__(self, hass, object_id, name, icon, sequence):
"""Initialize the script.""" """Initialize the script."""
self.object_id = object_id self.object_id = object_id
self.icon = icon
self.entity_id = ENTITY_ID_FORMAT.format(object_id) self.entity_id = ENTITY_ID_FORMAT.format(object_id)
self.script = Script(hass, sequence, name, self.async_update_ha_state) self.script = Script(hass, sequence, name, self.async_update_ha_state)

21
tests/components/script/test_init.py
View File

@ -446,3 +446,24 @@ async def test_extraction_functions(hass):
"device-in-both", "device-in-both",
"device-in-last", "device-in-last",
} }
async def test_config(hass):
"""Test passing info in config."""
assert await async_setup_component(
hass,
"script",
{
"script": {
"test_script": {
"alias": "Script Name",
"icon": "mdi:party",
"sequence": [],
}
}
},
)
test_script = hass.states.get("script.test_script")
assert test_script.name == "Script Name"
assert test_script.attributes["icon"] == "mdi:party"