139 lines
4.5 KiB
Python
139 lines
4.5 KiB
Python
"""
|
|
This platform provides sensors for OpenUV data.
|
|
|
|
For more details about this platform, please refer to the documentation at
|
|
https://home-assistant.io/components/sensor.openuv/
|
|
"""
|
|
import logging
|
|
|
|
from homeassistant.const import CONF_MONITORED_CONDITIONS
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.components.openuv import (
|
|
DATA_UV, DOMAIN, SENSORS, TOPIC_UPDATE, TYPE_CURRENT_OZONE_LEVEL,
|
|
TYPE_CURRENT_UV_INDEX, TYPE_CURRENT_UV_LEVEL, TYPE_MAX_UV_INDEX,
|
|
TYPE_SAFE_EXPOSURE_TIME_1, TYPE_SAFE_EXPOSURE_TIME_2,
|
|
TYPE_SAFE_EXPOSURE_TIME_3, TYPE_SAFE_EXPOSURE_TIME_4,
|
|
TYPE_SAFE_EXPOSURE_TIME_5, TYPE_SAFE_EXPOSURE_TIME_6, OpenUvEntity)
|
|
from homeassistant.util.dt import as_local, parse_datetime
|
|
|
|
DEPENDENCIES = ['openuv']
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ATTR_MAX_UV_TIME = 'time'
|
|
|
|
EXPOSURE_TYPE_MAP = {
|
|
TYPE_SAFE_EXPOSURE_TIME_1: 'st1',
|
|
TYPE_SAFE_EXPOSURE_TIME_2: 'st2',
|
|
TYPE_SAFE_EXPOSURE_TIME_3: 'st3',
|
|
TYPE_SAFE_EXPOSURE_TIME_4: 'st4',
|
|
TYPE_SAFE_EXPOSURE_TIME_5: 'st5',
|
|
TYPE_SAFE_EXPOSURE_TIME_6: 'st6'
|
|
}
|
|
|
|
UV_LEVEL_EXTREME = "Extreme"
|
|
UV_LEVEL_VHIGH = "Very High"
|
|
UV_LEVEL_HIGH = "High"
|
|
UV_LEVEL_MODERATE = "Moderate"
|
|
UV_LEVEL_LOW = "Low"
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass, config, async_add_entities, discovery_info=None):
|
|
"""Set up the OpenUV binary sensor platform."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
openuv = hass.data[DOMAIN]
|
|
|
|
sensors = []
|
|
for sensor_type in discovery_info[CONF_MONITORED_CONDITIONS]:
|
|
name, icon, unit = SENSORS[sensor_type]
|
|
sensors.append(OpenUvSensor(openuv, sensor_type, name, icon, unit))
|
|
|
|
async_add_entities(sensors, True)
|
|
|
|
|
|
class OpenUvSensor(OpenUvEntity):
|
|
"""Define a binary sensor for OpenUV."""
|
|
|
|
def __init__(self, openuv, sensor_type, name, icon, unit):
|
|
"""Initialize the sensor."""
|
|
super().__init__(openuv)
|
|
|
|
self._icon = icon
|
|
self._latitude = openuv.client.latitude
|
|
self._longitude = openuv.client.longitude
|
|
self._name = name
|
|
self._sensor_type = sensor_type
|
|
self._state = None
|
|
self._unit = unit
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Return the icon."""
|
|
return self._icon
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Disable polling."""
|
|
return False
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the status of the sensor."""
|
|
return self._state
|
|
|
|
@property
|
|
def unique_id(self) -> str:
|
|
"""Return a unique, HASS-friendly identifier for this entity."""
|
|
return '{0}_{1}_{2}'.format(
|
|
self._latitude, self._longitude, self._sensor_type)
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit the value is expressed in."""
|
|
return self._unit
|
|
|
|
@callback
|
|
def _update_data(self):
|
|
"""Update the state."""
|
|
self.async_schedule_update_ha_state(True)
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Register callbacks."""
|
|
async_dispatcher_connect(self.hass, TOPIC_UPDATE, self._update_data)
|
|
|
|
async def async_update(self):
|
|
"""Update the state."""
|
|
data = self.openuv.data[DATA_UV]['result']
|
|
if self._sensor_type == TYPE_CURRENT_OZONE_LEVEL:
|
|
self._state = data['ozone']
|
|
elif self._sensor_type == TYPE_CURRENT_UV_INDEX:
|
|
self._state = data['uv']
|
|
elif self._sensor_type == TYPE_CURRENT_UV_LEVEL:
|
|
if data['uv'] >= 11:
|
|
self._state = UV_LEVEL_EXTREME
|
|
elif data['uv'] >= 8:
|
|
self._state = UV_LEVEL_VHIGH
|
|
elif data['uv'] >= 6:
|
|
self._state = UV_LEVEL_HIGH
|
|
elif data['uv'] >= 3:
|
|
self._state = UV_LEVEL_MODERATE
|
|
else:
|
|
self._state = UV_LEVEL_LOW
|
|
elif self._sensor_type == TYPE_MAX_UV_INDEX:
|
|
self._state = data['uv_max']
|
|
self._attrs.update({
|
|
ATTR_MAX_UV_TIME: as_local(
|
|
parse_datetime(data['uv_max_time']))
|
|
})
|
|
elif self._sensor_type in (TYPE_SAFE_EXPOSURE_TIME_1,
|
|
TYPE_SAFE_EXPOSURE_TIME_2,
|
|
TYPE_SAFE_EXPOSURE_TIME_3,
|
|
TYPE_SAFE_EXPOSURE_TIME_4,
|
|
TYPE_SAFE_EXPOSURE_TIME_5,
|
|
TYPE_SAFE_EXPOSURE_TIME_6):
|
|
self._state = data['safe_exposure_time'][EXPOSURE_TYPE_MAP[
|
|
self._sensor_type]]
|