c43bdbf7c8
* Add state_class and unique_id properties * Update homeassistant/components/lightwave/sensor.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * fix isort * set class via attribute Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
"""Support for LightwaveRF TRV - Associated Battery."""
|
|
from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT, SensorEntity
|
|
from homeassistant.const import CONF_NAME, DEVICE_CLASS_BATTERY, PERCENTAGE
|
|
|
|
from . import CONF_SERIAL, LIGHTWAVE_LINK
|
|
|
|
|
|
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
|
|
"""Find and return battery."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
batteries = []
|
|
|
|
lwlink = hass.data[LIGHTWAVE_LINK]
|
|
|
|
for device_config in discovery_info.values():
|
|
name = device_config[CONF_NAME]
|
|
serial = device_config[CONF_SERIAL]
|
|
batteries.append(LightwaveBattery(name, lwlink, serial))
|
|
|
|
async_add_entities(batteries)
|
|
|
|
|
|
class LightwaveBattery(SensorEntity):
|
|
"""Lightwave TRV Battery."""
|
|
|
|
_attr_device_class = DEVICE_CLASS_BATTERY
|
|
_attr_unit_of_measurement = PERCENTAGE
|
|
_attr_state_class = STATE_CLASS_MEASUREMENT
|
|
|
|
def __init__(self, name, lwlink, serial):
|
|
"""Initialize the Lightwave Trv battery sensor."""
|
|
self._name = name
|
|
self._state = None
|
|
self._lwlink = lwlink
|
|
self._serial = serial
|
|
self._attr_unique_id = f"{serial}-trv-battery"
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return self._name
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the sensor."""
|
|
return self._state
|
|
|
|
def update(self):
|
|
"""Communicate with a Lightwave RTF Proxy to get state."""
|
|
(dummy_temp, dummy_targ, battery, dummy_output) = self._lwlink.read_trv_status(
|
|
self._serial
|
|
)
|
|
self._state = battery
|