jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 05:07:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1401 from balloob/wemo_insight_updates

Browse Source 
Wemo updates
This commit is contained in:
Greg Dowling 2016-02-26 00:23:17 +00:00
commit f92802fd5c
2 changed files with 34 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
homeassistant/components/switch/wemo.py
View File

@ -9,7 +9,8 @@ https://home-assistant.io/components/switch.wemo/
import logging import logging
from homeassistant.components.switch import SwitchDevice from homeassistant.components.switch import SwitchDevice
from homeassistant.const import STATE_OFF, STATE_ON, STATE_STANDBY from homeassistant.const import (
STATE_OFF, STATE_ON, STATE_STANDBY, STATE_UNKNOWN)
from homeassistant.loader import get_component from homeassistant.loader import get_component
DEPENDENCIES = ['wemo'] DEPENDENCIES = ['wemo']
@ -18,10 +19,18 @@ _LOGGER = logging.getLogger(__name__)
ATTR_SENSOR_STATE = "sensor_state" ATTR_SENSOR_STATE = "sensor_state"
ATTR_SWITCH_MODE = "switch_mode" ATTR_SWITCH_MODE = "switch_mode"
ATTR_CURRENT_STATE_DETAIL = 'state_detail'
MAKER_SWITCH_MOMENTARY = "momentary" MAKER_SWITCH_MOMENTARY = "momentary"
MAKER_SWITCH_TOGGLE = "toggle" MAKER_SWITCH_TOGGLE = "toggle"
MAKER_SWITCH_MOMENTARY = "momentary"
MAKER_SWITCH_TOGGLE = "toggle"
WEMO_ON = 1
WEMO_OFF = 0
WEMO_STANDBY = 8
# pylint: disable=unused-argument, too-many-function-args # pylint: disable=unused-argument, too-many-function-args
def setup_platform(hass, config, add_devices_callback, discovery_info=None): def setup_platform(hass, config, add_devices_callback, discovery_info=None):
@ -43,6 +52,7 @@ class WemoSwitch(SwitchDevice):
self.wemo = device self.wemo = device
self.insight_params = None self.insight_params = None
self.maker_params = None self.maker_params = None
self._state = None
wemo = get_component('wemo') wemo = get_component('wemo')
wemo.SUBSCRIPTION_REGISTRY.register(self.wemo) wemo.SUBSCRIPTION_REGISTRY.register(self.wemo)
@ -88,17 +98,10 @@ class WemoSwitch(SwitchDevice):
else: else:
attr[ATTR_SWITCH_MODE] = MAKER_SWITCH_TOGGLE attr[ATTR_SWITCH_MODE] = MAKER_SWITCH_TOGGLE
return attr if self.insight_params:
attr[ATTR_CURRENT_STATE_DETAIL] = self.detail_state
@property return attr
def state(self):
"""Returns the state."""
is_on = self.is_on
if not is_on:
return STATE_OFF
elif self.is_standby:
return STATE_STANDBY
return STATE_ON
@property @property
def current_power_mwh(self): def current_power_mwh(self):
@ -113,21 +116,23 @@ class WemoSwitch(SwitchDevice):
return self.insight_params['todaymw'] return self.insight_params['todaymw']
@property @property
def is_standby(self): def detail_state(self):
"""Is the device on - or in standby.""" """Is the device on - or in standby."""
if self.insight_params: if self.insight_params:
standby_state = self.insight_params['state'] standby_state = int(self.insight_params['state'])
# Standby is actually '8' but seems more defensive if standby_state == WEMO_ON:
# to check for the On and Off states return STATE_OFF
if standby_state == '1' or standby_state == '0': elif standby_state == WEMO_OFF:
return False return STATE_OFF
elif standby_state == WEMO_STANDBY:
return STATE_STANDBY
else: else:
return True return STATE_UNKNOWN
@property @property
def is_on(self): def is_on(self):
"""True if switch is on.""" """True if switch is on. Standby is on!"""
return self.wemo.get_state() return self._state
@property @property
def available(self): def available(self):
@ -143,16 +148,20 @@ class WemoSwitch(SwitchDevice):
def turn_on(self, **kwargs): def turn_on(self, **kwargs):
"""Turns the switch on.""" """Turns the switch on."""
self._state = WEMO_ON
self.update_ha_state()
self.wemo.on() self.wemo.on()
def turn_off(self): def turn_off(self):
"""Turns the switch off.""" """Turns the switch off."""
self._state = WEMO_OFF
self.update_ha_state()
self.wemo.off() self.wemo.off()
def update(self): def update(self):
"""Update WeMo state.""" """Update WeMo state."""
try: try:
self.wemo.get_state(True) self._state = self.wemo.get_state(True)
if self.wemo.model_name == 'Insight': if self.wemo.model_name == 'Insight':
self.insight_params = self.wemo.insight_params self.insight_params = self.wemo.insight_params
self.insight_params['standby_state'] = ( self.insight_params['standby_state'] = (

8
homeassistant/components/wemo.py
View File

@ -25,6 +25,7 @@ WEMO_MODEL_DISPATCH = {
'Maker': DISCOVER_SWITCHES, 'Maker': DISCOVER_SWITCHES,
'Motion': DISCOVER_MOTION, 'Motion': DISCOVER_MOTION,
'Socket': DISCOVER_SWITCHES, 'Socket': DISCOVER_SWITCHES,
'LightSwitch': DISCOVER_SWITCHES
} }
WEMO_SERVICE_DISPATCH = { WEMO_SERVICE_DISPATCH = {
DISCOVER_LIGHTS: 'light', DISCOVER_LIGHTS: 'light',
@ -64,12 +65,11 @@ def setup(hass, config):
return return
KNOWN_DEVICES.append(mac) KNOWN_DEVICES.append(mac)
service = WEMO_MODEL_DISPATCH.get(model_name) service = WEMO_MODEL_DISPATCH.get(model_name) or DISCOVER_SWITCHES
component = WEMO_SERVICE_DISPATCH.get(service) component = WEMO_SERVICE_DISPATCH.get(service)
if service is not None: discovery.discover(hass, service, discovery_info,
discovery.discover(hass, service, discovery_info, component, config)
component, config)
discovery.listen(hass, discovery.SERVICE_WEMO, discovery_dispatch) discovery.listen(hass, discovery.SERVICE_WEMO, discovery_dispatch)