core/homeassistant/components/venstar/binary_sensor.py

51 lines
1.5 KiB
Python

"""Alarm sensors for the Venstar Thermostat."""
from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN
from .entity import VenstarEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up Vensar device binary_sensors based on a config entry."""
coordinator = hass.data[DOMAIN][config_entry.entry_id]
if coordinator.client.alerts is None:
return
async_add_entities(
VenstarBinarySensor(coordinator, config_entry, alert["name"])
for alert in coordinator.client.alerts
)
class VenstarBinarySensor(VenstarEntity, BinarySensorEntity):
"""Represent a Venstar alert."""
_attr_device_class = BinarySensorDeviceClass.PROBLEM
def __init__(self, coordinator, config, alert):
"""Initialize the alert."""
super().__init__(coordinator, config)
self.alert = alert
self._attr_unique_id = f"{config.entry_id}_{alert.replace(' ', '_')}"
self._attr_name = alert
@property
def is_on(self):
"""Return true if the binary sensor is on."""
for alert in self._client.alerts:
if alert["name"] == self.alert:
return alert["active"]
return None