""" Support for WeMo switchces. """ import logging from homeassistant.helpers import ToggleDevice from homeassistant.const import ATTR_FRIENDLY_NAME, CONF_HOSTS from homeassistant.components.switch import ( ATTR_TODAY_MWH, ATTR_CURRENT_POWER_MWH) # pylint: disable=unused-argument def get_devices(hass, config): """ Find and return WeMo switches. """ try: # Pylint does not play nice if not every folders has an __init__.py # pylint: disable=no-name-in-module, import-error import homeassistant.external.pywemo.pywemo as pywemo except ImportError: logging.getLogger(__name__).exception(( "Failed to import pywemo. " "Did you maybe not run `git submodule init` " "and `git submodule update`?")) return [] if CONF_HOSTS in config: switches = (pywemo.device_from_host(host) for host in config[CONF_HOSTS].split(",")) else: logging.getLogger(__name__).info("Scanning for WeMo devices") switches = pywemo.discover_devices() # Filter out the switches and wrap in WemoSwitch object return [WemoSwitch(switch) for switch in switches if isinstance(switch, pywemo.Switch)] class WemoSwitch(ToggleDevice): """ represents a WeMo switch within home assistant. """ def __init__(self, wemo): self.wemo = wemo def get_name(self): """ Returns the name of the switch if any. """ return self.wemo.name def turn_on(self, **kwargs): """ Turns the switch on. """ self.wemo.on() def turn_off(self): """ Turns the switch off. """ self.wemo.off() def is_on(self): """ True if switch is on. """ return self.wemo.get_state(True) def get_state_attributes(self): """ Returns optional state attributes. """ if self.wemo.model.startswith('Belkin Insight'): cur_info = self.wemo.insight_params return { ATTR_FRIENDLY_NAME: self.wemo.name, ATTR_CURRENT_POWER_MWH: cur_info['currentpower'], ATTR_TODAY_MWH: cur_info['todaymw'] } else: return {ATTR_FRIENDLY_NAME: self.wemo.name}