mirror of https://github.com/home-assistant/core
122 lines
3.7 KiB
Python
122 lines
3.7 KiB
Python
"""Support for Genius Hub sensor devices."""
|
|
from datetime import timedelta
|
|
from typing import Any, Awaitable, Dict
|
|
|
|
from homeassistant.const import DEVICE_CLASS_BATTERY
|
|
from homeassistant.util.dt import utc_from_timestamp, utcnow
|
|
|
|
from . import DOMAIN, GeniusEntity
|
|
|
|
GH_HAS_BATTERY = ["Room Thermostat", "Genius Valve", "Room Sensor", "Radiator Valve"]
|
|
|
|
GH_LEVEL_MAPPING = {
|
|
"error": "Errors",
|
|
"warning": "Warnings",
|
|
"information": "Information",
|
|
}
|
|
|
|
|
|
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
|
|
"""Set up the Genius Hub sensor entities."""
|
|
client = hass.data[DOMAIN]["client"]
|
|
|
|
sensors = [GeniusBattery(d) for d in client.device_objs if d.type in GH_HAS_BATTERY]
|
|
issues = [GeniusIssue(client, i) for i in list(GH_LEVEL_MAPPING)]
|
|
|
|
async_add_entities(sensors + issues, update_before_add=True)
|
|
|
|
|
|
class GeniusBattery(GeniusEntity):
|
|
"""Representation of a Genius Hub sensor."""
|
|
|
|
def __init__(self, device) -> None:
|
|
"""Initialize the sensor."""
|
|
super().__init__()
|
|
|
|
self._device = device
|
|
self._name = f"{device.type} {device.id}"
|
|
|
|
@property
|
|
def icon(self) -> str:
|
|
"""Return the icon of the sensor."""
|
|
|
|
values = self._device._raw["childValues"] # pylint: disable=protected-access
|
|
|
|
last_comms = utc_from_timestamp(values["lastComms"]["val"])
|
|
if "WakeUp_Interval" in values:
|
|
interval = timedelta(seconds=values["WakeUp_Interval"]["val"])
|
|
else:
|
|
interval = timedelta(minutes=20)
|
|
|
|
if last_comms < utcnow() - interval * 3:
|
|
return "mdi:battery-unknown"
|
|
|
|
battery_level = self._device.data["state"]["batteryLevel"]
|
|
if battery_level == 255:
|
|
return "mdi:battery-unknown"
|
|
if battery_level < 40:
|
|
return "mdi:battery-alert"
|
|
|
|
icon = "mdi:battery"
|
|
if battery_level <= 95:
|
|
icon += f"-{int(round(battery_level / 10 - 0.01)) * 10}"
|
|
|
|
return icon
|
|
|
|
@property
|
|
def device_class(self) -> str:
|
|
"""Return the device class of the sensor."""
|
|
return DEVICE_CLASS_BATTERY
|
|
|
|
@property
|
|
def unit_of_measurement(self) -> str:
|
|
"""Return the unit of measurement of the sensor."""
|
|
return "%"
|
|
|
|
@property
|
|
def state(self) -> str:
|
|
"""Return the state of the sensor."""
|
|
level = self._device.data["state"].get("batteryLevel", 255)
|
|
return level if level != 255 else 0
|
|
|
|
@property
|
|
def device_state_attributes(self) -> Dict[str, Any]:
|
|
"""Return the device state attributes."""
|
|
attrs = {}
|
|
attrs["assigned_zone"] = self._device.data["assignedZones"][0]["name"]
|
|
|
|
# pylint: disable=protected-access
|
|
last_comms = self._device._raw["childValues"]["lastComms"]["val"]
|
|
attrs["last_comms"] = utc_from_timestamp(last_comms).isoformat()
|
|
|
|
return {**attrs}
|
|
|
|
|
|
class GeniusIssue(GeniusEntity):
|
|
"""Representation of a Genius Hub sensor."""
|
|
|
|
def __init__(self, hub, level) -> None:
|
|
"""Initialize the sensor."""
|
|
super().__init__()
|
|
|
|
self._hub = hub
|
|
self._name = GH_LEVEL_MAPPING[level]
|
|
self._level = level
|
|
self._issues = []
|
|
|
|
@property
|
|
def state(self) -> str:
|
|
"""Return the number of issues."""
|
|
return len(self._issues)
|
|
|
|
@property
|
|
def device_state_attributes(self) -> Dict[str, Any]:
|
|
"""Return the device state attributes."""
|
|
return {f"{self._level}_list": self._issues}
|
|
|
|
async def async_update(self) -> Awaitable[None]:
|
|
"""Process the sensor's state data."""
|
|
self._issues = [
|
|
i["description"] for i in self._hub.issues if i["level"] == self._level
|
|
]
|