136 lines
4.2 KiB
Python
136 lines
4.2 KiB
Python
|
"""
|
||
|
|
Support for EDP re:dy.
|
||
|
|
|
||
|
|
For more details about this component, please refer to the documentation at
|
||
|
|
https://home-assistant.io/components/edp_redy/
|
||
|
|
"""
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
import voluptuous as vol
|
||
|
|
|
||
|
|
from homeassistant.const import (CONF_USERNAME, CONF_PASSWORD,
|
||
|
|
EVENT_HOMEASSISTANT_START)
|
||
|
|
from homeassistant.core import callback
|
||
|
|
from homeassistant.helpers import discovery, dispatcher, aiohttp_client
|
||
|
|
import homeassistant.helpers.config_validation as cv
|
||
|
|
from homeassistant.helpers.entity import Entity
|
||
|
|
from homeassistant.helpers.event import async_track_point_in_time
|
||
|
|
from homeassistant.util import dt as dt_util
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
DOMAIN = 'edp_redy'
|
||
|
|
EDP_REDY = 'edp_redy'
|
||
|
|
DATA_UPDATE_TOPIC = '{0}_data_update'.format(DOMAIN)
|
||
|
|
UPDATE_INTERVAL = 30
|
||
|
|
|
||
|
|
REQUIREMENTS = ['edp_redy==0.0.2']
|
||
|
|
|
||
|
|
CONFIG_SCHEMA = vol.Schema({
|
||
|
|
DOMAIN: vol.Schema({
|
||
|
|
vol.Required(CONF_USERNAME): cv.string,
|
||
|
|
vol.Required(CONF_PASSWORD): cv.string
|
||
|
|
})
|
||
|
|
}, extra=vol.ALLOW_EXTRA)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_setup(hass, config):
|
||
|
|
"""Set up the EDP re:dy component."""
|
||
|
|
from edp_redy import EdpRedySession
|
||
|
|
|
||
|
|
session = EdpRedySession(config[DOMAIN][CONF_USERNAME],
|
||
|
|
config[DOMAIN][CONF_PASSWORD],
|
||
|
|
aiohttp_client.async_get_clientsession(hass),
|
||
|
|
hass.loop)
|
||
|
|
hass.data[EDP_REDY] = session
|
||
|
|
platform_loaded = False
|
||
|
|
|
||
|
|
async def async_update_and_sched(time):
|
||
|
|
update_success = await session.async_update()
|
||
|
|
|
||
|
|
if update_success:
|
||
|
|
nonlocal platform_loaded
|
||
|
|
# pylint: disable=used-before-assignment
|
||
|
|
if not platform_loaded:
|
||
|
|
for component in ['sensor', 'switch']:
|
||
|
|
await discovery.async_load_platform(hass, component,
|
||
|
|
DOMAIN, {}, config)
|
||
|
|
platform_loaded = True
|
||
|
|
|
||
|
|
dispatcher.async_dispatcher_send(hass, DATA_UPDATE_TOPIC)
|
||
|
|
|
||
|
|
# schedule next update
|
||
|
|
async_track_point_in_time(hass, async_update_and_sched,
|
||
|
|
time + timedelta(seconds=UPDATE_INTERVAL))
|
||
|
|
|
||
|
|
async def start_component(event):
|
||
|
|
_LOGGER.debug("Starting updates")
|
||
|
|
await async_update_and_sched(dt_util.utcnow())
|
||
|
|
|
||
|
|
# only start fetching data after HA boots to prevent delaying the boot
|
||
|
|
# process
|
||
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_START, start_component)
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
class EdpRedyDevice(Entity):
|
||
|
|
"""Representation a base re:dy device."""
|
||
|
|
|
||
|
|
def __init__(self, session, device_id, name):
|
||
|
|
"""Initialize the device."""
|
||
|
|
self._session = session
|
||
|
|
self._state = None
|
||
|
|
self._is_available = True
|
||
|
|
self._device_state_attributes = {}
|
||
|
|
self._id = device_id
|
||
|
|
self._unique_id = device_id
|
||
|
|
self._name = name if name else device_id
|
||
|
|
|
||
|
|
async def async_added_to_hass(self):
|
||
|
|
"""Subscribe to the data updates topic."""
|
||
|
|
dispatcher.async_dispatcher_connect(
|
||
|
|
self.hass, DATA_UPDATE_TOPIC, self._data_updated)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self):
|
||
|
|
"""Return the name of the device."""
|
||
|
|
return self._name
|
||
|
|
|
||
|
|
@property
|
||
|
|
def unique_id(self) -> str:
|
||
|
|
"""Return a unique ID."""
|
||
|
|
return self._unique_id
|
||
|
|
|
||
|
|
@property
|
||
|
|
def available(self):
|
||
|
|
"""Return True if entity is available."""
|
||
|
|
return self._is_available
|
||
|
|
|
||
|
|
@property
|
||
|
|
def should_poll(self):
|
||
|
|
"""Return the polling state. No polling needed."""
|
||
|
|
return False
|
||
|
|
|
||
|
|
@property
|
||
|
|
def device_state_attributes(self):
|
||
|
|
"""Return the state attributes."""
|
||
|
|
return self._device_state_attributes
|
||
|
|
|
||
|
|
@callback
|
||
|
|
def _data_updated(self):
|
||
|
|
"""Update state, trigger updates."""
|
||
|
|
self.async_schedule_update_ha_state(True)
|
||
|
|
|
||
|
|
def _parse_data(self, data):
|
||
|
|
"""Parse data received from the server."""
|
||
|
|
if "OutOfOrder" in data:
|
||
|
|
try:
|
||
|
|
self._is_available = not data['OutOfOrder']
|
||
|
|
except ValueError:
|
||
|
|
_LOGGER.error(
|
||
|
|
"Could not parse OutOfOrder for %s", self._id)
|
||
|
|
self._is_available = False
|