96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
"""Support for EDP re:dy plugs/switches."""
|
|
import logging
|
|
|
|
from homeassistant.components.switch import SwitchDevice
|
|
|
|
from . import EDP_REDY, EdpRedyDevice
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DEPENDENCIES = ['edp_redy']
|
|
|
|
# Load power in watts (W)
|
|
ATTR_ACTIVE_POWER = 'active_power'
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass, config, async_add_entities, discovery_info=None):
|
|
"""Perform the setup for re:dy devices."""
|
|
session = hass.data[EDP_REDY]
|
|
devices = []
|
|
for device_json in session.modules_dict.values():
|
|
if 'HA_SWITCH' not in device_json['Capabilities']:
|
|
continue
|
|
devices.append(EdpRedySwitch(session, device_json))
|
|
|
|
async_add_entities(devices, True)
|
|
|
|
|
|
class EdpRedySwitch(EdpRedyDevice, SwitchDevice):
|
|
"""Representation of a Edp re:dy switch (plugs, switches, etc)."""
|
|
|
|
def __init__(self, session, device_json):
|
|
"""Initialize the switch."""
|
|
super().__init__(session, device_json['PKID'], device_json['Name'])
|
|
|
|
self._active_power = None
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Return the icon to use in the frontend."""
|
|
return 'mdi:power-plug'
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if it is on."""
|
|
return self._state
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes."""
|
|
if self._active_power is not None:
|
|
attrs = {ATTR_ACTIVE_POWER: self._active_power}
|
|
else:
|
|
attrs = {}
|
|
attrs.update(super().device_state_attributes)
|
|
return attrs
|
|
|
|
async def async_turn_on(self, **kwargs):
|
|
"""Turn the switch on."""
|
|
if await self._async_send_state_cmd(True):
|
|
self._state = True
|
|
self.async_schedule_update_ha_state()
|
|
|
|
async def async_turn_off(self, **kwargs):
|
|
"""Turn the switch off."""
|
|
if await self._async_send_state_cmd(False):
|
|
self._state = False
|
|
self.async_schedule_update_ha_state()
|
|
|
|
async def _async_send_state_cmd(self, state):
|
|
state_json = {'devModuleId': self._id, 'key': 'RelayState',
|
|
'value': state}
|
|
return await self._session.async_set_state_var(state_json)
|
|
|
|
async def async_update(self):
|
|
"""Parse the data for this switch."""
|
|
if self._id in self._session.modules_dict:
|
|
device_json = self._session.modules_dict[self._id]
|
|
self._parse_data(device_json)
|
|
else:
|
|
self._is_available = False
|
|
|
|
def _parse_data(self, data):
|
|
"""Parse data received from the server."""
|
|
super()._parse_data(data)
|
|
|
|
for state_var in data['StateVars']:
|
|
if state_var['Name'] == 'RelayState':
|
|
self._state = state_var['Value'] == 'true'
|
|
elif state_var['Name'] == 'ActivePower':
|
|
try:
|
|
self._active_power = float(state_var['Value']) * 1000
|
|
except ValueError:
|
|
_LOGGER.error("Could not parse power for %s", self._id)
|
|
self._active_power = None
|