63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""Support for Abode Security System sensors."""
|
|
import abodepy.helpers.constants as CONST
|
|
|
|
from homeassistant.components.sensor import SensorEntity
|
|
from homeassistant.const import (
|
|
DEVICE_CLASS_HUMIDITY,
|
|
DEVICE_CLASS_ILLUMINANCE,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
)
|
|
|
|
from . import AbodeDevice
|
|
from .const import DOMAIN
|
|
|
|
# Sensor types: Name, icon
|
|
SENSOR_TYPES = {
|
|
CONST.TEMP_STATUS_KEY: ["Temperature", DEVICE_CLASS_TEMPERATURE],
|
|
CONST.HUMI_STATUS_KEY: ["Humidity", DEVICE_CLASS_HUMIDITY],
|
|
CONST.LUX_STATUS_KEY: ["Lux", DEVICE_CLASS_ILLUMINANCE],
|
|
}
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up Abode sensor devices."""
|
|
data = hass.data[DOMAIN]
|
|
|
|
entities = []
|
|
|
|
for device in data.abode.get_devices(generic_type=CONST.TYPE_SENSOR):
|
|
for sensor_type in SENSOR_TYPES:
|
|
if sensor_type not in device.get_value(CONST.STATUSES_KEY):
|
|
continue
|
|
entities.append(AbodeSensor(data, device, sensor_type))
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
class AbodeSensor(AbodeDevice, SensorEntity):
|
|
"""A sensor implementation for Abode devices."""
|
|
|
|
def __init__(self, data, device, sensor_type):
|
|
"""Initialize a sensor for an Abode device."""
|
|
super().__init__(data, device)
|
|
self._sensor_type = sensor_type
|
|
self._attr_name = f"{device.name} {SENSOR_TYPES[sensor_type][0]}"
|
|
self._attr_device_class = SENSOR_TYPES[self._sensor_type][1]
|
|
self._attr_unique_id = f"{device.device_uuid}-{sensor_type}"
|
|
if self._sensor_type == CONST.TEMP_STATUS_KEY:
|
|
self._attr_unit_of_measurement = device.temp_unit
|
|
elif self._sensor_type == CONST.HUMI_STATUS_KEY:
|
|
self._attr_unit_of_measurement = device.humidity_unit
|
|
elif self._sensor_type == CONST.LUX_STATUS_KEY:
|
|
self._attr_unit_of_measurement = device.lux_unit
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the sensor."""
|
|
if self._sensor_type == CONST.TEMP_STATUS_KEY:
|
|
return self._device.temp
|
|
if self._sensor_type == CONST.HUMI_STATUS_KEY:
|
|
return self._device.humidity
|
|
if self._sensor_type == CONST.LUX_STATUS_KEY:
|
|
return self._device.lux
|