87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
|
"""Support for hunterdouglass_powerview sensors."""
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from aiopvapi.resources.shade import factory as PvShade
|
||
|
|
|
||
|
|
from homeassistant.const import DEVICE_CLASS_BATTERY, UNIT_PERCENTAGE
|
||
|
|
from homeassistant.core import callback
|
||
|
|
|
||
|
|
from .const import (
|
||
|
|
COORDINATOR,
|
||
|
|
DEVICE_INFO,
|
||
|
|
DOMAIN,
|
||
|
|
PV_API,
|
||
|
|
PV_SHADE_DATA,
|
||
|
|
SHADE_BATTERY_LEVEL,
|
||
|
|
SHADE_BATTERY_LEVEL_MAX,
|
||
|
|
)
|
||
|
|
from .entity import ShadeEntity
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_setup_entry(hass, entry, async_add_entities):
|
||
|
|
"""Set up the hunter douglas shades sensors."""
|
||
|
|
|
||
|
|
pv_data = hass.data[DOMAIN][entry.entry_id]
|
||
|
|
shade_data = pv_data[PV_SHADE_DATA]
|
||
|
|
pv_request = pv_data[PV_API]
|
||
|
|
coordinator = pv_data[COORDINATOR]
|
||
|
|
device_info = pv_data[DEVICE_INFO]
|
||
|
|
|
||
|
|
entities = []
|
||
|
|
for raw_shade in shade_data.values():
|
||
|
|
shade = PvShade(raw_shade, pv_request)
|
||
|
|
if SHADE_BATTERY_LEVEL not in shade.raw_data:
|
||
|
|
continue
|
||
|
|
name_before_refresh = shade.name
|
||
|
|
entities.append(
|
||
|
|
PowerViewShadeBatterySensor(
|
||
|
|
coordinator, device_info, shade, name_before_refresh
|
||
|
|
)
|
||
|
|
)
|
||
|
|
async_add_entities(entities)
|
||
|
|
|
||
|
|
|
||
|
|
class PowerViewShadeBatterySensor(ShadeEntity):
|
||
|
|
"""Representation of an shade battery charge sensor."""
|
||
|
|
|
||
|
|
@property
|
||
|
|
def unit_of_measurement(self):
|
||
|
|
"""Return the unit of measurement."""
|
||
|
|
return UNIT_PERCENTAGE
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self):
|
||
|
|
"""Name of the shade battery."""
|
||
|
|
return f"{self._shade_name} Battery"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def device_class(self):
|
||
|
|
"""Shade battery Class."""
|
||
|
|
return DEVICE_CLASS_BATTERY
|
||
|
|
|
||
|
|
@property
|
||
|
|
def unique_id(self):
|
||
|
|
"""Shade battery Uniqueid."""
|
||
|
|
return f"{self._unique_id}_charge"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def state(self):
|
||
|
|
"""Get the current value in percentage."""
|
||
|
|
return round(
|
||
|
|
self._shade.raw_data[SHADE_BATTERY_LEVEL] / SHADE_BATTERY_LEVEL_MAX * 100
|
||
|
|
)
|
||
|
|
|
||
|
|
async def async_added_to_hass(self):
|
||
|
|
"""When entity is added to hass."""
|
||
|
|
self.async_on_remove(
|
||
|
|
self._coordinator.async_add_listener(self._async_update_shade_from_group)
|
||
|
|
)
|
||
|
|
|
||
|
|
@callback
|
||
|
|
def _async_update_shade_from_group(self):
|
||
|
|
"""Update with new data from the coordinator."""
|
||
|
|
self._shade.raw_data = self._coordinator.data[self._shade.id]
|
||
|
|
self.async_write_ha_state()
|