mirror of https://github.com/home-assistant/core
135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
"""Support for KEBA charging station sensors."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.components.sensor import (
|
|
SensorDeviceClass,
|
|
SensorEntity,
|
|
SensorEntityDescription,
|
|
SensorStateClass,
|
|
)
|
|
from homeassistant.const import UnitOfElectricCurrent, UnitOfEnergy, UnitOfPower
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from . import DOMAIN, KebaHandler
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
async_add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the KEBA charging station platform."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
keba = hass.data[DOMAIN]
|
|
|
|
sensors = [
|
|
KebaSensor(
|
|
keba,
|
|
"max_current",
|
|
SensorEntityDescription(
|
|
key="Curr user",
|
|
name="Max Current",
|
|
native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
|
|
device_class=SensorDeviceClass.CURRENT,
|
|
),
|
|
),
|
|
KebaSensor(
|
|
keba,
|
|
"energy_target",
|
|
SensorEntityDescription(
|
|
key="Setenergy",
|
|
name="Energy Target",
|
|
native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
|
|
device_class=SensorDeviceClass.ENERGY,
|
|
),
|
|
),
|
|
KebaSensor(
|
|
keba,
|
|
"charging_power",
|
|
SensorEntityDescription(
|
|
key="P",
|
|
name="Charging Power",
|
|
native_unit_of_measurement=UnitOfPower.KILO_WATT,
|
|
device_class=SensorDeviceClass.POWER,
|
|
state_class=SensorStateClass.MEASUREMENT,
|
|
),
|
|
),
|
|
KebaSensor(
|
|
keba,
|
|
"session_energy",
|
|
SensorEntityDescription(
|
|
key="E pres", # codespell:ignore pres
|
|
name="Session Energy",
|
|
native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
|
|
device_class=SensorDeviceClass.ENERGY,
|
|
),
|
|
),
|
|
KebaSensor(
|
|
keba,
|
|
"total_energy",
|
|
SensorEntityDescription(
|
|
key="E total",
|
|
name="Total Energy",
|
|
native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
|
|
device_class=SensorDeviceClass.ENERGY,
|
|
state_class=SensorStateClass.TOTAL_INCREASING,
|
|
),
|
|
),
|
|
]
|
|
async_add_entities(sensors)
|
|
|
|
|
|
class KebaSensor(SensorEntity):
|
|
"""The entity class for KEBA charging stations sensors."""
|
|
|
|
_attr_should_poll = False
|
|
|
|
def __init__(
|
|
self,
|
|
keba: KebaHandler,
|
|
entity_type: str,
|
|
description: SensorEntityDescription,
|
|
) -> None:
|
|
"""Initialize the KEBA Sensor."""
|
|
self._keba = keba
|
|
self.entity_description = description
|
|
|
|
self._attr_name = f"{keba.device_name} {description.name}"
|
|
self._attr_unique_id = f"{keba.device_id}_{entity_type}"
|
|
|
|
self._attributes: dict[str, str] = {}
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, str]:
|
|
"""Return the state attributes of the binary sensor."""
|
|
return self._attributes
|
|
|
|
async def async_update(self) -> None:
|
|
"""Get latest cached states from the device."""
|
|
self._attr_native_value = self._keba.get_value(self.entity_description.key)
|
|
|
|
if self.entity_description.key == "P":
|
|
self._attributes["power_factor"] = self._keba.get_value("PF")
|
|
self._attributes["voltage_u1"] = str(self._keba.get_value("U1"))
|
|
self._attributes["voltage_u2"] = str(self._keba.get_value("U2"))
|
|
self._attributes["voltage_u3"] = str(self._keba.get_value("U3"))
|
|
self._attributes["current_i1"] = str(self._keba.get_value("I1"))
|
|
self._attributes["current_i2"] = str(self._keba.get_value("I2"))
|
|
self._attributes["current_i3"] = str(self._keba.get_value("I3"))
|
|
elif self.entity_description.key == "Curr user":
|
|
self._attributes["max_current_hardware"] = self._keba.get_value("Curr HW")
|
|
|
|
def update_callback(self) -> None:
|
|
"""Schedule a state update."""
|
|
self.async_schedule_update_ha_state(True)
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Add update callback after being added to hass."""
|
|
self._keba.add_update_listener(self.update_callback)
|