binary_sensor.py

"""Binary sensors for nibe uplink."""

import logging

from homeassistant.components.binary_sensor import ENTITY_ID_FORMAT, BinarySensorEntity
from homeassistant.exceptions import PlatformNotReady

from .const import CONF_BINARY_SENSORS, DATA_NIBE
from .entity import NibeParameterEntity

PARALLEL_UPDATES = 0
_LOGGER = logging.getLogger(__name__)


async def async_setup_entry(hass, entry, async_add_entities):
    """Set up the device based on a config entry."""
    if DATA_NIBE not in hass.data:
        raise PlatformNotReady

    uplink = hass.data[DATA_NIBE].uplink
    systems = hass.data[DATA_NIBE].systems

    entities = []
    for system in systems.values():
        for parameter_id in system.config[CONF_BINARY_SENSORS]:
            entities.append(
                NibeBinarySensor(uplink, system.system_id, parameter_id, entry)
            )

    async_add_entities(entities, True)


class NibeBinarySensor(NibeParameterEntity, BinarySensorEntity):
    """Binary sensor."""

    def __init__(self, uplink, system_id, parameter_id, entry):
        """Init."""
        super(NibeBinarySensor, self).__init__(
            uplink, system_id, parameter_id, None, ENTITY_ID_FORMAT
        )

    @property
    def is_on(self):
        """Return if sensor is on."""
        data = self._parameters[self._parameter_id]
        if data:
            return data["rawValue"] == "1"
        else:
            return None