51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
""" Support for WeMo switchces. """
|
|
import logging
|
|
|
|
import homeassistant.external.wink.pywink as pywink
|
|
|
|
from homeassistant.helpers import ToggleDevice
|
|
from homeassistant.const import ATTR_FRIENDLY_NAME
|
|
|
|
|
|
# pylint: disable=unused-argument
|
|
def get_devices(hass, config):
|
|
""" Find and return Wink switches. """
|
|
token = config.get('access_token')
|
|
|
|
if token is None:
|
|
logging.getLogger(__name__).error(
|
|
"Missing wink access_token - "
|
|
"get one at https://winkbearertoken.appspot.com/")
|
|
return False
|
|
|
|
pywink.set_bearer_token(token)
|
|
|
|
return [WinkSwitch(switch) for switch in pywink.get_switches()]
|
|
|
|
|
|
class WinkSwitch(ToggleDevice):
|
|
""" represents a WeMo switch within home assistant. """
|
|
|
|
def __init__(self, wink):
|
|
self.wink = wink
|
|
self.state_attr = {ATTR_FRIENDLY_NAME: wink.name()}
|
|
|
|
def get_name(self):
|
|
""" Returns the name of the switch if any. """
|
|
return self.wink.name()
|
|
|
|
def turn_on(self, **kwargs):
|
|
""" Turns the switch on. """
|
|
self.wink.setState(True)
|
|
|
|
def turn_off(self):
|
|
""" Turns the switch off. """
|
|
self.wink.setState(False)
|
|
|
|
def is_on(self):
|
|
""" True if switch is on. """
|
|
return self.wink.state()
|
|
|
|
def get_state_attributes(self):
|
|
""" Returns optional state attributes. """
|
|
return self.state_attr
|