core/homeassistant/components/aqualogic/switch.py

111 lines
3.1 KiB
Python

"""Support for AquaLogic switches."""
import logging
import voluptuous as vol
from homeassistant.components.switch import PLATFORM_SCHEMA, SwitchDevice
from homeassistant.const import CONF_MONITORED_CONDITIONS
from homeassistant.core import callback
import homeassistant.helpers.config_validation as cv
from . import DOMAIN, UPDATE_TOPIC
DEPENDENCIES = ['aqualogic']
_LOGGER = logging.getLogger(__name__)
SWITCH_TYPES = {
'lights': 'Lights',
'filter': 'Filter',
'filter_low_speed': 'Filter Low Speed',
'aux_1': 'Aux 1',
'aux_2': 'Aux 2',
'aux_3': 'Aux 3',
'aux_4': 'Aux 4',
'aux_5': 'Aux 5',
'aux_6': 'Aux 6',
'aux_7': 'Aux 7',
}
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_MONITORED_CONDITIONS, default=list(SWITCH_TYPES)):
vol.All(cv.ensure_list, [vol.In(SWITCH_TYPES)]),
})
async def async_setup_platform(
hass, config, async_add_entities, discovery_info=None):
"""Set up the switch platform."""
switches = []
processor = hass.data[DOMAIN]
for switch_type in config.get(CONF_MONITORED_CONDITIONS):
switches.append(AquaLogicSwitch(processor, switch_type))
async_add_entities(switches)
class AquaLogicSwitch(SwitchDevice):
"""Switch implementation for the AquaLogic component."""
def __init__(self, processor, switch_type):
"""Initialize switch."""
from aqualogic.core import States
self._processor = processor
self._type = switch_type
self._state_name = {
'lights': States.LIGHTS,
'filter': States.FILTER,
'filter_low_speed': States.FILTER_LOW_SPEED,
'aux_1': States.AUX_1,
'aux_2': States.AUX_2,
'aux_3': States.AUX_3,
'aux_4': States.AUX_4,
'aux_5': States.AUX_5,
'aux_6': States.AUX_6,
'aux_7': States.AUX_7
}[switch_type]
@property
def name(self):
"""Return the name of the switch."""
return "AquaLogic {}".format(SWITCH_TYPES[self._type])
@property
def should_poll(self):
"""Return the polling state."""
return False
@property
def is_on(self):
"""Return true if device is on."""
panel = self._processor.panel
if panel is None:
return False
state = panel.get_state(self._state_name)
return state
def turn_on(self, **kwargs):
"""Turn the device on."""
panel = self._processor.panel
if panel is None:
return
panel.set_state(self._state_name, True)
def turn_off(self, **kwargs):
"""Turn the device off."""
panel = self._processor.panel
if panel is None:
return
panel.set_state(self._state_name, False)
async def async_added_to_hass(self):
"""Register callbacks."""
self.hass.helpers.dispatcher.async_dispatcher_connect(
UPDATE_TOPIC, self.async_update_callback)
@callback
def async_update_callback(self):
"""Update callback."""
self.async_schedule_update_ha_state()