maxclaey
4218b31e7b
* Added support for security system, switch and thermostat * Processing review * Only perform set call when the call didn't come from HomeKit * Added support for alarm_code * Take into account review remarks * Provide tests for HomeKit security systems, switches and thermostats * Support STATE_AUTO * Guard if state exists * Improve support for thermostat auto mode * Provide both high and low at the same time for home assistant * Set default values within accepted ranges * Added tests for auto mode * Fix thermostat test error * Use attributes.get instead of indexing for safety * Avoid hardcoded attributes in tests
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""Class to hold all switch accessories."""
|
|
import logging
|
|
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.core import split_entity_id
|
|
from homeassistant.helpers.event import async_track_state_change
|
|
|
|
from . import TYPES
|
|
from .accessories import HomeAccessory, add_preload_service
|
|
from .const import SERV_SWITCH, CHAR_ON
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@TYPES.register('Switch')
|
|
class Switch(HomeAccessory):
|
|
"""Generate a Switch accessory."""
|
|
|
|
def __init__(self, hass, entity_id, display_name):
|
|
"""Initialize a Switch accessory object to represent a remote."""
|
|
super().__init__(display_name, entity_id, 'SWITCH')
|
|
|
|
self._hass = hass
|
|
self._entity_id = entity_id
|
|
self._domain = split_entity_id(entity_id)[0]
|
|
|
|
self.flag_target_state = False
|
|
|
|
self.service_switch = add_preload_service(self, SERV_SWITCH)
|
|
self.char_on = self.service_switch.get_characteristic(CHAR_ON)
|
|
self.char_on.value = False
|
|
self.char_on.setter_callback = self.set_state
|
|
|
|
def run(self):
|
|
"""Method called be object after driver is started."""
|
|
state = self._hass.states.get(self._entity_id)
|
|
self.update_state(new_state=state)
|
|
|
|
async_track_state_change(self._hass, self._entity_id,
|
|
self.update_state)
|
|
|
|
def set_state(self, value):
|
|
"""Move switch state to value if call came from HomeKit."""
|
|
_LOGGER.debug("%s: Set switch state to %s",
|
|
self._entity_id, value)
|
|
self.flag_target_state = True
|
|
service = 'turn_on' if value else 'turn_off'
|
|
self._hass.services.call(self._domain, service,
|
|
{ATTR_ENTITY_ID: self._entity_id})
|
|
|
|
def update_state(self, entity_id=None, old_state=None, new_state=None):
|
|
"""Update switch state after state changed."""
|
|
if new_state is None:
|
|
return
|
|
|
|
current_state = (new_state.state == 'on')
|
|
if not self.flag_target_state:
|
|
_LOGGER.debug("%s: Set current state to %s",
|
|
self._entity_id, current_state)
|
|
self.char_on.set_value(current_state, should_callback=False)
|
|
else:
|
|
self.flag_target_state = False
|