sensor.py

Go to the documentation of this file.

"""Initialization of ATAG One sensor platform."""
 
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import (
    PERCENTAGE,
    UnitOfPressure,
    UnitOfTemperature,
    UnitOfTime,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .coordinator import AtagConfigEntry, AtagDataUpdateCoordinator
from .entity import AtagEntity
 
SENSORS = {
    "Outside Temperature": "outside_temp",
    "Average Outside Temperature": "tout_avg",
    "Weather Status": "weather_status",
    "CH Water Pressure": "ch_water_pres",
    "CH Water Temperature": "ch_water_temp",
    "CH Return Temperature": "ch_return_temp",
    "Burning Hours": "burning_hours",
    "Flame": "rel_mod_level",
}
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: AtagConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Initialize sensor platform from config entry."""
    coordinator = config_entry.runtime_data
    async_add_entities([AtagSensor(coordinator, sensor) for sensor in SENSORS])
 
 
class AtagSensor(AtagEntity, SensorEntity):
    """Representation of a AtagOne Sensor."""
 
    def __init__(self, coordinator: AtagDataUpdateCoordinator, sensor: str) -> None:
        """Initialize Atag sensor."""
        super().__init__(coordinator, SENSORS[sensor])
        self._attr_name_attr_name_attr_name = sensor
        if coordinator.atag.report[self._id_id].sensorclass in (
            SensorDeviceClass.PRESSURE,
            SensorDeviceClass.TEMPERATURE,
        ):
            self._attr_device_class_attr_device_class = coordinator.atag.report[self._id_id].sensorclass
        if coordinator.atag.report[self._id_id].measure in (
            UnitOfPressure.BAR,
            UnitOfTemperature.CELSIUS,
            UnitOfTemperature.FAHRENHEIT,
            PERCENTAGE,
            UnitOfTime.HOURS,
        ):
            self._attr_native_unit_of_measurement_attr_native_unit_of_measurement = coordinator.atag.report[
                self._id_id
            ].measure
 
    @property
    def native_value(self):
        """Return the state of the sensor."""
        return self.coordinator.atag.report[self._id_id].state
 
    @property
    def icon(self):
        """Return icon."""
        return self.coordinator.atag.report[self._id_id].icon