core/homeassistant/components/switch/tellduslive.py

"""
homeassistant.components.switch.tellduslive
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Support for Tellstick switches using Tellstick Net and
the Telldus Live online service.

For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/switch.tellduslive/

"""
import logging

from homeassistant.components import tellduslive
from homeassistant.helpers.entity import ToggleEntity

_LOGGER = logging.getLogger(__name__)


def setup_platform(hass, config, add_devices, discovery_info=None):
    """ Find and return Tellstick switches. """
    if discovery_info is None:
        return
    add_devices(TelldusLiveSwitch(switch) for switch in discovery_info)


class TelldusLiveSwitch(ToggleEntity):
    """ Represents a Tellstick switch. """

    def __init__(self, switch_id):
        self._id = switch_id
        self.update()
        _LOGGER.debug("created switch %s", self)

    def update(self):
        tellduslive.NETWORK.update_switches()
        self._switch = tellduslive.NETWORK.get_switch(self._id)

    @property
    def should_poll(self):
        """ Tells Home Assistant to poll this entity. """
        return True

    @property
    def assumed_state(self):
        """Return True if unable to access real state of entity."""
        return True

    @property
    def name(self):
        """ Returns the name of the switch if any. """
        return self._switch["name"]

    @property
    def available(self):
        return not self._switch.get("offline", False)

    @property
    def is_on(self):
        """ True if switch is on. """
        from tellive.live import const
        return self._switch["state"] == const.TELLSTICK_TURNON

    def turn_on(self, **kwargs):
        """ Turns the switch on. """
        tellduslive.NETWORK.turn_switch_on(self._id)

    def turn_off(self, **kwargs):
        """ Turns the switch off. """
        tellduslive.NETWORK.turn_switch_off(self._id)