mirror of
https://github.com/home-assistant/core.git
synced 2026-05-30 04:05:01 +01:00
Franck Nijhof
118 lines
3.5 KiB
Python
118 lines
3.5 KiB
Python
"""Support for Genius Hub sensor devices."""
|
|
|
|
from datetime import timedelta
|
|
from typing import Any
|
|
|
|
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
|
|
from homeassistant.const import PERCENTAGE
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
from . import GeniusHubConfigEntry
|
|
from .entity import GeniusDevice, GeniusEntity
|
|
|
|
GH_STATE_ATTR = "batteryLevel"
|
|
|
|
GH_LEVEL_MAPPING = {
|
|
"error": "Errors",
|
|
"warning": "Warnings",
|
|
"information": "Information",
|
|
}
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: GeniusHubConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up the Genius Hub sensor entities."""
|
|
|
|
broker = entry.runtime_data
|
|
|
|
entities: list[GeniusBattery | GeniusIssue] = [
|
|
GeniusBattery(broker, d, GH_STATE_ATTR)
|
|
for d in broker.client.device_objs
|
|
if GH_STATE_ATTR in d.data["state"]
|
|
]
|
|
entities.extend([GeniusIssue(broker, i) for i in list(GH_LEVEL_MAPPING)])
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
class GeniusBattery(GeniusDevice, SensorEntity):
|
|
"""Representation of a Genius Hub sensor."""
|
|
|
|
_attr_device_class = SensorDeviceClass.BATTERY
|
|
_attr_native_unit_of_measurement = PERCENTAGE
|
|
|
|
def __init__(self, broker, device, state_attr) -> None:
|
|
"""Initialize the sensor."""
|
|
super().__init__(broker, device)
|
|
|
|
self._state_attr = state_attr
|
|
|
|
self._attr_name = f"{device.type} {device.id}"
|
|
|
|
@property
|
|
def icon(self) -> str:
|
|
"""Return the icon of the sensor."""
|
|
if "_state" in self._device.data: # only for v3 API
|
|
interval = timedelta(
|
|
seconds=self._device.data["_state"].get("wakeupInterval", 30 * 60)
|
|
)
|
|
if (
|
|
not self._last_comms
|
|
or self._last_comms < dt_util.utcnow() - interval * 3
|
|
):
|
|
return "mdi:battery-unknown"
|
|
|
|
battery_level = self._device.data["state"][self._state_attr]
|
|
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"-{round(battery_level / 10 - 0.01) * 10}"
|
|
|
|
return icon
|
|
|
|
@property
|
|
def native_value(self) -> int:
|
|
"""Return the state of the sensor."""
|
|
level = self._device.data["state"][self._state_attr]
|
|
return level if level != 255 else 0
|
|
|
|
|
|
class GeniusIssue(GeniusEntity, SensorEntity):
|
|
"""Representation of a Genius Hub sensor."""
|
|
|
|
def __init__(self, broker, level) -> None:
|
|
"""Initialize the sensor."""
|
|
super().__init__()
|
|
|
|
self._hub = broker.client
|
|
self._unique_id = f"{broker.hub_uid}_{GH_LEVEL_MAPPING[level]}"
|
|
|
|
self._attr_name = f"GeniusHub {GH_LEVEL_MAPPING[level]}"
|
|
self._level = level
|
|
self._issues: list = []
|
|
|
|
@property
|
|
def native_value(self) -> int:
|
|
"""Return the number of issues."""
|
|
return len(self._issues)
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, Any]:
|
|
"""Return the device state attributes."""
|
|
return {f"{self._level}_list": self._issues}
|
|
|
|
async def async_update(self) -> None:
|
|
"""Process the sensor's state data."""
|
|
self._issues = [
|
|
i["description"] for i in self._hub.issues if i["level"] == self._level
|
|
]
|