sensor.py

Go to the documentation of this file.

"""Platform for beewi_smartclim integration."""
 
from __future__ import annotations
 
from beewi_smartclim import BeewiSmartClimPoller
import voluptuous as vol
 
from homeassistant.components.sensor import (
    PLATFORM_SCHEMA as SENSOR_PLATFORM_SCHEMA,
    SensorDeviceClass,
    SensorEntity,
)
from homeassistant.const import CONF_MAC, CONF_NAME, PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
# Default values
DEFAULT_NAME = "BeeWi SmartClim"
 
# Sensor config
SENSOR_TYPES = [
    [SensorDeviceClass.TEMPERATURE, "Temperature", UnitOfTemperature.CELSIUS],
    [SensorDeviceClass.HUMIDITY, "Humidity", PERCENTAGE],
    [SensorDeviceClass.BATTERY, "Battery", PERCENTAGE],
]
 
PLATFORM_SCHEMA = SENSOR_PLATFORM_SCHEMA.extend(
    {
        vol.Required(CONF_MAC): cv.string,
        vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
    }
)
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up the beewi_smartclim platform."""
 
    mac = config[CONF_MAC]
    prefix = config[CONF_NAME]
    poller = BeewiSmartClimPoller(mac)
 
    sensors = []
 
    for sensor_type in SENSOR_TYPES:
        device = sensor_type[0]
        name = sensor_type[1]
        unit = sensor_type[2]
        # `prefix` is the name configured by the user for the sensor, we're appending
        #  the device type at the end of the name (garden -> garden temperature)
        if prefix:
            name = f"{prefix} {name}"
 
        sensors.append(BeewiSmartclimSensor(poller, name, mac, device, unit))
 
    add_entities(sensors)
 
 
class BeewiSmartclimSensor(SensorEntity):
    """Representation of a Sensor."""
 
    def __init__(self, poller, name, mac, device, unit):
        """Initialize the sensor."""
        self._poller_poller = poller
        self._attr_name_attr_name = name
        self._device_device = device
        self._attr_native_unit_of_measurement_attr_native_unit_of_measurement = unit
        self._attr_device_class_attr_device_class = self._device_device
        self._attr_unique_id_attr_unique_id = f"{mac}_{device}"
 
    def update(self) -> None:
        """Fetch new state data from the poller."""
        self._poller_poller.update_sensor()
        self._attr_native_value_attr_native_value = None
        if self._device_device == SensorDeviceClass.TEMPERATURE:
            self._attr_native_value_attr_native_value = self._poller_poller.get_temperature()
        if self._device_device == SensorDeviceClass.HUMIDITY:
            self._attr_native_value_attr_native_value = self._poller_poller.get_humidity()
        if self._device_device == SensorDeviceClass.BATTERY:
            self._attr_native_value_attr_native_value = self._poller_poller.get_battery()