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

Add node sensor status icons (#56137)

Browse Source 
Co-authored-by: kpine <keith.pine@gmail.com>
Co-authored-by: Raman Gupta <7243222+raman325@users.noreply.github.com>
This commit is contained in:
Marius 2021-09-27 12:50:14 +02:00 committed by GitHub
parent 83b1b3e92c
commit 4ce7166afd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/zwave_js/sensor.py
View File

@ -8,7 +8,7 @@ from typing import cast
import voluptuous as vol import voluptuous as vol
from zwave_js_server.client import Client as ZwaveClient from zwave_js_server.client import Client as ZwaveClient
from zwave_js_server.const import CommandClass, ConfigurationValueType from zwave_js_server.const import CommandClass, ConfigurationValueType, NodeStatus
from zwave_js_server.const.command_class.meter import ( from zwave_js_server.const.command_class.meter import (
RESET_METER_OPTION_TARGET_VALUE, RESET_METER_OPTION_TARGET_VALUE,
RESET_METER_OPTION_TYPE, RESET_METER_OPTION_TYPE,
@ -80,6 +80,14 @@ from .helpers import get_device_id
LOGGER = logging.getLogger(__name__) LOGGER = logging.getLogger(__name__)
STATUS_ICON: dict[NodeStatus, str] = {
NodeStatus.ALIVE: "mdi:heart-pulse",
NodeStatus.ASLEEP: "mdi:sleep",
NodeStatus.AWAKE: "mdi:eye",
NodeStatus.DEAD: "mdi:robot-dead",
NodeStatus.UNKNOWN: "mdi:help-rhombus",
}
@dataclass @dataclass
class ZwaveSensorEntityDescription(SensorEntityDescription): class ZwaveSensorEntityDescription(SensorEntityDescription):
@ -480,6 +488,11 @@ class ZWaveNodeStatusSensor(SensorEntity):
self._attr_native_value = self.node.status.name.lower() self._attr_native_value = self.node.status.name.lower()
self.async_write_ha_state() self.async_write_ha_state()
@property
def icon(self) -> str | None:
"""Icon of the entity."""
return STATUS_ICON[self.node.status]
async def async_added_to_hass(self) -> None: async def async_added_to_hass(self) -> None:
"""Call when entity is added.""" """Call when entity is added."""
# Add value_changed callbacks. # Add value_changed callbacks.

7
tests/components/zwave_js/test_sensor.py
View File

@ -20,6 +20,7 @@ from homeassistant.components.zwave_js.const import (
from homeassistant.const import ( from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
ATTR_ICON,
DEVICE_CLASS_CURRENT, DEVICE_CLASS_CURRENT,
DEVICE_CLASS_ENERGY, DEVICE_CLASS_ENERGY,
DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_HUMIDITY,
@ -168,24 +169,30 @@ async def test_node_status_sensor(hass, client, lock_id_lock_as_id150, integrati
) )
node.receive_event(event) node.receive_event(event)
assert hass.states.get(NODE_STATUS_ENTITY).state == "dead" assert hass.states.get(NODE_STATUS_ENTITY).state == "dead"
assert hass.states.get(NODE_STATUS_ENTITY).attributes[ATTR_ICON] == "mdi:robot-dead"
event = Event( event = Event(
"wake up", data={"source": "node", "event": "wake up", "nodeId": node.node_id} "wake up", data={"source": "node", "event": "wake up", "nodeId": node.node_id}
) )
node.receive_event(event) node.receive_event(event)
assert hass.states.get(NODE_STATUS_ENTITY).state == "awake" assert hass.states.get(NODE_STATUS_ENTITY).state == "awake"
assert hass.states.get(NODE_STATUS_ENTITY).attributes[ATTR_ICON] == "mdi:eye"
event = Event( event = Event(
"sleep", data={"source": "node", "event": "sleep", "nodeId": node.node_id} "sleep", data={"source": "node", "event": "sleep", "nodeId": node.node_id}
) )
node.receive_event(event) node.receive_event(event)
assert hass.states.get(NODE_STATUS_ENTITY).state == "asleep" assert hass.states.get(NODE_STATUS_ENTITY).state == "asleep"
assert hass.states.get(NODE_STATUS_ENTITY).attributes[ATTR_ICON] == "mdi:sleep"
event = Event( event = Event(
"alive", data={"source": "node", "event": "alive", "nodeId": node.node_id} "alive", data={"source": "node", "event": "alive", "nodeId": node.node_id}
) )
node.receive_event(event) node.receive_event(event)
assert hass.states.get(NODE_STATUS_ENTITY).state == "alive" assert hass.states.get(NODE_STATUS_ENTITY).state == "alive"
assert (
hass.states.get(NODE_STATUS_ENTITY).attributes[ATTR_ICON] == "mdi:heart-pulse"
)
# Disconnect the client and make sure the entity is still available # Disconnect the client and make sure the entity is still available
await client.disconnect() await client.disconnect()