evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add command_line icon_template (#58877)

Browse source 
Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>
This commit is contained in:
Marius 2021-12-03 19:06:32 +01:00 • committed by GitHub
parent e0562385f1
commit b61dede826
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
homeassistant/components/command_line/switch.py
View file

@ -13,6 +13,7 @@ from homeassistant.const import (
CONF_COMMAND_ON, CONF_COMMAND_ON,
CONF_COMMAND_STATE, CONF_COMMAND_STATE,
CONF_FRIENDLY_NAME, CONF_FRIENDLY_NAME,
CONF_ICON_TEMPLATE,
CONF_SWITCHES, CONF_SWITCHES,
CONF_VALUE_TEMPLATE, CONF_VALUE_TEMPLATE,
) )
@ -31,6 +32,7 @@ SWITCH_SCHEMA = vol.Schema(
vol.Optional(CONF_COMMAND_STATE): cv.string, vol.Optional(CONF_COMMAND_STATE): cv.string,
vol.Optional(CONF_FRIENDLY_NAME): cv.string, vol.Optional(CONF_FRIENDLY_NAME): cv.string,
vol.Optional(CONF_VALUE_TEMPLATE): cv.template, vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
vol.Optional(CONF_ICON_TEMPLATE): cv.template,
vol.Optional(CONF_COMMAND_TIMEOUT, default=DEFAULT_TIMEOUT): cv.positive_int, vol.Optional(CONF_COMMAND_TIMEOUT, default=DEFAULT_TIMEOUT): cv.positive_int,
} }
) )
@ -54,6 +56,10 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
if value_template is not None: if value_template is not None:
value_template.hass = hass value_template.hass = hass
icon_template = device_config.get(CONF_ICON_TEMPLATE)
if icon_template is not None:
icon_template.hass = hass
switches.append( switches.append(
CommandSwitch( CommandSwitch(
hass, hass,
@ -62,6 +68,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
device_config[CONF_COMMAND_ON], device_config[CONF_COMMAND_ON],
device_config[CONF_COMMAND_OFF], device_config[CONF_COMMAND_OFF],
device_config.get(CONF_COMMAND_STATE), device_config.get(CONF_COMMAND_STATE),
icon_template,
value_template, value_template,
device_config[CONF_COMMAND_TIMEOUT], device_config[CONF_COMMAND_TIMEOUT],
) )
@ -85,6 +92,7 @@ class CommandSwitch(SwitchEntity):
command_on, command_on,
command_off, command_off,
command_state, command_state,
icon_template,
value_template, value_template,
timeout, timeout,
): ):
@ -96,6 +104,7 @@ class CommandSwitch(SwitchEntity):
self._command_on = command_on self._command_on = command_on
self._command_off = command_off self._command_off = command_off
self._command_state = command_state self._command_state = command_state
self._icon_template = icon_template
self._value_template = value_template self._value_template = value_template
self._timeout = timeout self._timeout = timeout
@ -152,6 +161,10 @@ class CommandSwitch(SwitchEntity):
"""Update device state.""" """Update device state."""
if self._command_state: if self._command_state:
payload = str(self._query_state()) payload = str(self._query_state())
if self._icon_template:
self._attr_icon = self._icon_template.render_with_possible_json_value(
payload
)
if self._value_template: if self._value_template:
payload = self._value_template.render_with_possible_json_value(payload) payload = self._value_template.render_with_possible_json_value(payload)
self._state = payload.lower() == "true" self._state = payload.lower() == "true"

6
tests/components/command_line/test_switch.py
View file

@ -90,6 +90,7 @@ async def test_state_value(hass: HomeAssistant) -> None:
"command_on": f"echo 1 > {path}", "command_on": f"echo 1 > {path}",
"command_off": f"echo 0 > {path}", "command_off": f"echo 0 > {path}",
"value_template": '{{ value=="1" }}', "value_template": '{{ value=="1" }}',
"icon_template": '{% if value=="1" %} mdi:on {% else %} mdi:off {% endif %}',
} }
}, },
) )
@ -108,6 +109,7 @@ async def test_state_value(hass: HomeAssistant) -> None:
entity_state = hass.states.get("switch.test") entity_state = hass.states.get("switch.test")
assert entity_state assert entity_state
assert entity_state.state == STATE_ON assert entity_state.state == STATE_ON
assert entity_state.attributes.get("icon") == "mdi:on"
await hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
@ -119,6 +121,7 @@ async def test_state_value(hass: HomeAssistant) -> None:
entity_state = hass.states.get("switch.test") entity_state = hass.states.get("switch.test")
assert entity_state assert entity_state
assert entity_state.state == STATE_OFF assert entity_state.state == STATE_OFF
assert entity_state.attributes.get("icon") == "mdi:off"
async def test_state_json_value(hass: HomeAssistant) -> None: async def test_state_json_value(hass: HomeAssistant) -> None:
@ -136,6 +139,7 @@ async def test_state_json_value(hass: HomeAssistant) -> None:
"command_on": f"echo '{oncmd}' > {path}", "command_on": f"echo '{oncmd}' > {path}",
"command_off": f"echo '{offcmd}' > {path}", "command_off": f"echo '{offcmd}' > {path}",
"value_template": '{{ value_json.status=="ok" }}', "value_template": '{{ value_json.status=="ok" }}',
"icon_template": '{% if value_json.status=="ok" %} mdi:on {% else %} mdi:off {% endif %}',
} }
}, },
) )
@ -154,6 +158,7 @@ async def test_state_json_value(hass: HomeAssistant) -> None:
entity_state = hass.states.get("switch.test") entity_state = hass.states.get("switch.test")
assert entity_state assert entity_state
assert entity_state.state == STATE_ON assert entity_state.state == STATE_ON
assert entity_state.attributes.get("icon") == "mdi:on"
await hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
@ -165,6 +170,7 @@ async def test_state_json_value(hass: HomeAssistant) -> None:
entity_state = hass.states.get("switch.test") entity_state = hass.states.get("switch.test")
assert entity_state assert entity_state
assert entity_state.state == STATE_OFF assert entity_state.state == STATE_OFF
assert entity_state.attributes.get("icon") == "mdi:off"
async def test_state_code(hass: HomeAssistant) -> None: async def test_state_code(hass: HomeAssistant) -> None: