ad86eb4be3
So far the Flo integration only supports shutoff valves. Add support for Flo leak detector pucks, which measure temperature and humidity in addition to providing leak alerts.
82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
"""Support for Flo Water Monitor binary sensors."""
|
|
|
|
from typing import List
|
|
|
|
from homeassistant.components.binary_sensor import (
|
|
DEVICE_CLASS_PROBLEM,
|
|
BinarySensorEntity,
|
|
)
|
|
|
|
from .const import DOMAIN as FLO_DOMAIN
|
|
from .device import FloDeviceDataUpdateCoordinator
|
|
from .entity import FloEntity
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up the Flo sensors from config entry."""
|
|
devices: List[FloDeviceDataUpdateCoordinator] = hass.data[FLO_DOMAIN][
|
|
config_entry.entry_id
|
|
]["devices"]
|
|
entities = []
|
|
for device in devices:
|
|
if device.device_type == "puck_oem":
|
|
# Flo "pucks" (leak detectors) *do* support pending alerts.
|
|
# However these pending alerts mix unrelated issues like
|
|
# low-battery alerts, humidity alerts, & temperature alerts
|
|
# in addition to the critical "water detected" alert.
|
|
#
|
|
# Since there are non-binary sensors for battery, humidity,
|
|
# and temperature, the binary sensor should only cover
|
|
# water detection. If this sensor trips, you really have
|
|
# a problem - vs. battery/temp/humidity which are warnings.
|
|
entities.append(FloWaterDetectedBinarySensor(device))
|
|
else:
|
|
entities.append(FloPendingAlertsBinarySensor(device))
|
|
async_add_entities(entities)
|
|
|
|
|
|
class FloPendingAlertsBinarySensor(FloEntity, BinarySensorEntity):
|
|
"""Binary sensor that reports on if there are any pending system alerts."""
|
|
|
|
def __init__(self, device):
|
|
"""Initialize the pending alerts binary sensor."""
|
|
super().__init__("pending_system_alerts", "Pending System Alerts", device)
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if the Flo device has pending alerts."""
|
|
return self._device.has_alerts
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes."""
|
|
if not self._device.has_alerts:
|
|
return {}
|
|
return {
|
|
"info": self._device.pending_info_alerts_count,
|
|
"warning": self._device.pending_warning_alerts_count,
|
|
"critical": self._device.pending_critical_alerts_count,
|
|
}
|
|
|
|
@property
|
|
def device_class(self):
|
|
"""Return the device class for the binary sensor."""
|
|
return DEVICE_CLASS_PROBLEM
|
|
|
|
|
|
class FloWaterDetectedBinarySensor(FloEntity, BinarySensorEntity):
|
|
"""Binary sensor that reports if water is detected (for leak detectors)."""
|
|
|
|
def __init__(self, device):
|
|
"""Initialize the pending alerts binary sensor."""
|
|
super().__init__("water_detected", "Water Detected", device)
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if the Flo device is detecting water."""
|
|
return self._device.water_detected
|
|
|
|
@property
|
|
def device_class(self):
|
|
"""Return the device class for the binary sensor."""
|
|
return DEVICE_CLASS_PROBLEM
|