binary_sensor.py

Go to the documentation of this file.

"""Support for Freebox devices (Freebox v6 and Freebox mini 4K)."""
 
from __future__ import annotations
 
import logging
from typing import Any
 
from homeassistant.components.binary_sensor import (
    BinarySensorDeviceClass,
    BinarySensorEntity,
    BinarySensorEntityDescription,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .const import DOMAIN, FreeboxHomeCategory
from .entity import FreeboxHomeEntity
from .router import FreeboxRouter
 
_LOGGER = logging.getLogger(__name__)
 
 
RAID_SENSORS: tuple[BinarySensorEntityDescription, ...] = (
    BinarySensorEntityDescription(
        key="raid_degraded",
        name="degraded",
        device_class=BinarySensorDeviceClass.PROBLEM,
        entity_category=EntityCategory.DIAGNOSTIC,
    ),
)
 
 
async def async_setup_entry(
    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
    """Set up binary sensors."""
    router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
 
    _LOGGER.debug("%s - %s - %s raid(s)", router.name, router.mac, len(router.raids))
 
    binary_entities: list[BinarySensorEntity] = [
        FreeboxRaidDegradedSensor(router, raid, description)
        for raid in router.raids.values()
        for description in RAID_SENSORS
    ]
 
    for node in router.home_devices.values():
        if node["category"] == FreeboxHomeCategory.PIR:
            binary_entities.append(FreeboxPirSensor(hass, router, node))
        elif node["category"] == FreeboxHomeCategory.DWS:
            binary_entities.append(FreeboxDwsSensor(hass, router, node))
 
        binary_entities.extend(
            FreeboxCoverSensor(hass, router, node)
            for endpoint in node["show_endpoints"]
            if (
                endpoint["name"] == "cover"
                and endpoint["ep_type"] == "signal"
                and endpoint.get("value") is not None
            )
        )
 
    async_add_entities(binary_entities, True)
 
 
class FreeboxHomeBinarySensor(FreeboxHomeEntity, BinarySensorEntity):
    """Representation of a Freebox binary sensor."""
 
    _sensor_name = "trigger"
 
    def __init__(
        self,
        hass: HomeAssistant,
        router: FreeboxRouter,
        node: dict[str, Any],
        sub_node: dict[str, Any] | None = None,
    ) -> None:
        """Initialize a Freebox binary sensor."""
        super().__init__(hass, router, node, sub_node)
        self._command_id_command_id = self.get_command_idget_command_id(
            node["type"]["endpoints"], "signal", self._sensor_name_sensor_name_sensor_name
        )
        self._attr_is_on_attr_is_on = self._edit_state_edit_state(self.get_valueget_value("signal", self._sensor_name_sensor_name_sensor_name))
 
    async def async_update_signal(self) -> None:
        """Update name & state."""
        self._attr_is_on_attr_is_on = self._edit_state_edit_state(
            await self.get_home_endpoint_valueget_home_endpoint_value(self._command_id_command_id)
        )
        await FreeboxHomeEntity.async_update_signal(self)
 
    def _edit_state(self, state: bool | None) -> bool | None:
        """Edit state depending on sensor name."""
        if state is None:
            return None
        if self._sensor_name_sensor_name_sensor_name == "trigger":
            return not state
        return state
 
 
class FreeboxPirSensor(FreeboxHomeBinarySensor):
    """Representation of a Freebox motion binary sensor."""
 
    _attr_device_class = BinarySensorDeviceClass.MOTION
 
 
class FreeboxDwsSensor(FreeboxHomeBinarySensor):
    """Representation of a Freebox door opener binary sensor."""
 
    _attr_device_class = BinarySensorDeviceClass.DOOR
 
 
class FreeboxCoverSensor(FreeboxHomeBinarySensor):
    """Representation of a cover Freebox plastic removal cover binary sensor (for some sensors: motion detector, door opener detector...)."""
 
    _attr_device_class = BinarySensorDeviceClass.SAFETY
    _attr_entity_category = EntityCategory.DIAGNOSTIC
    _attr_entity_registry_enabled_default = False
 
    _sensor_name = "cover"
 
    def __init__(
        self, hass: HomeAssistant, router: FreeboxRouter, node: dict[str, Any]
    ) -> None:
        """Initialize a cover for another device."""
        cover_node = next(
            filter(
                lambda x: (x["name"] == self._sensor_name_sensor_name_sensor_name_sensor_name and x["ep_type"] == "signal"),
                node["type"]["endpoints"],
            ),
            None,
        )
        super().__init__(hass, router, node, cover_node)
 
 
class FreeboxRaidDegradedSensor(BinarySensorEntity):
    """Representation of a Freebox raid sensor."""
 
    _attr_should_poll = False
    _attr_has_entity_name = True
 
    def __init__(
        self,
        router: FreeboxRouter,
        raid: dict[str, Any],
        description: BinarySensorEntityDescription,
    ) -> None:
        """Initialize a Freebox raid degraded sensor."""
        self.entity_descriptionentity_description = description
        self._router_router = router
        self._attr_device_info_attr_device_info = router.device_info
        self._raid_raid = raid
        self._attr_name_attr_name = f"Raid array {raid['id']} {description.name}"
        self._attr_unique_id_attr_unique_id = (
            f"{router.mac} {description.key} {raid['name']} {raid['id']}"
        )
 
    @callback
    def async_update_state(self) -> None:
        """Update the Freebox Raid sensor."""
        self._raid_raid = self._router_router.raids[self._raid_raid["id"]]
 
    @property
    def is_on(self) -> bool:
        """Return true if degraded."""
        return self._raid_raid["degraded"]
 
    @callback
    def async_on_demand_update(self) -> None:
        """Update state."""
        self.async_update_stateasync_update_state()
        self.async_write_ha_stateasync_write_ha_state()
 
    async def async_added_to_hass(self) -> None:
        """Register state update callback."""
        self.async_update_stateasync_update_state()
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(
                self.hasshass,
                self._router_router.signal_sensor_update,
                self.async_on_demand_updateasync_on_demand_update,
            )
        )