core/homeassistant/components/switch/isy994.py

74 lines
2.4 KiB
Python

"""
Support for ISY994 switches.
For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/switch.isy994/
"""
import logging
from typing import Callable
from homeassistant.components.switch import SwitchDevice, DOMAIN
from homeassistant.components.isy994 import (ISY994_NODES, ISY994_PROGRAMS,
ISYDevice)
from homeassistant.helpers.typing import ConfigType
_LOGGER = logging.getLogger(__name__)
def setup_platform(hass, config: ConfigType,
add_entities: Callable[[list], None], discovery_info=None):
"""Set up the ISY994 switch platform."""
devices = []
for node in hass.data[ISY994_NODES][DOMAIN]:
if not node.dimmable:
devices.append(ISYSwitchDevice(node))
for name, status, actions in hass.data[ISY994_PROGRAMS][DOMAIN]:
devices.append(ISYSwitchProgram(name, status, actions))
add_entities(devices)
class ISYSwitchDevice(ISYDevice, SwitchDevice):
"""Representation of an ISY994 switch device."""
@property
def is_on(self) -> bool:
"""Get whether the ISY994 device is in the on state."""
return bool(self.value)
def turn_off(self, **kwargs) -> None:
"""Send the turn on command to the ISY994 switch."""
if not self._node.off():
_LOGGER.debug('Unable to turn on switch.')
def turn_on(self, **kwargs) -> None:
"""Send the turn off command to the ISY994 switch."""
if not self._node.on():
_LOGGER.debug('Unable to turn on switch.')
class ISYSwitchProgram(ISYSwitchDevice):
"""A representation of an ISY994 program switch."""
def __init__(self, name: str, node, actions) -> None:
"""Initialize the ISY994 switch program."""
super().__init__(node)
self._name = name
self._actions = actions
@property
def is_on(self) -> bool:
"""Get whether the ISY994 switch program is on."""
return bool(self.value)
def turn_on(self, **kwargs) -> None:
"""Send the turn on command to the ISY994 switch program."""
if not self._actions.runThen():
_LOGGER.error('Unable to turn on switch')
def turn_off(self, **kwargs) -> None:
"""Send the turn off command to the ISY994 switch program."""
if not self._actions.runElse():
_LOGGER.error('Unable to turn off switch')