hass-core/homeassistant/components/light/wemo.py

"""
Support for Belkin WeMo lights.

For more details about this component, please refer to the documentation at
https://home-assistant.io/components/light.wemo/
"""
import logging
from datetime import timedelta

import homeassistant.util as util
from homeassistant.components.light import (
    Light, ATTR_BRIGHTNESS)

DEPENDENCIES = ['wemo']

MIN_TIME_BETWEEN_SCANS = timedelta(seconds=10)
MIN_TIME_BETWEEN_FORCED_SCANS = timedelta(milliseconds=100)

_LOGGER = logging.getLogger(__name__)


def setup_platform(hass, config, add_devices_callback, discovery_info=None):
    """Setup WeMo bridges and register connected lights."""
    import pywemo.discovery as discovery

    if discovery_info is not None:
        location = discovery_info[2]
        mac = discovery_info[3]
        device = discovery.device_from_description(location, mac)

        if device:
            setup_bridge(device, add_devices_callback)


def setup_bridge(bridge, add_devices_callback):
    """Setup a WeMo link."""
    lights = {}

    @util.Throttle(MIN_TIME_BETWEEN_SCANS, MIN_TIME_BETWEEN_FORCED_SCANS)
    def update_lights():
        """Update the WeMo led objects with latest info from the bridge."""
        bridge.bridge_get_lights()

        new_lights = []

        for light_id, info in bridge.Lights.items():
            if light_id not in lights:
                lights[light_id] = WemoLight(bridge, light_id, info,
                                             update_lights)
                new_lights.append(lights[light_id])
            else:
                lights[light_id].info = info

        if new_lights:
            add_devices_callback(new_lights)

    update_lights()


class WemoLight(Light):
    """Representation of a WeMo light."""

    def __init__(self, bridge, light_id, info, update_lights):
        """Initialize the light."""
        self.bridge = bridge
        self.light_id = light_id
        self.info = info
        self.update_lights = update_lights

    @property
    def unique_id(self):
        """Return the ID of this light."""
        deviceid = self.bridge.light_get_id(self.info)
        return "{}.{}".format(self.__class__, deviceid)

    @property
    def name(self):
        """Return the name of the light."""
        return self.bridge.light_name(self.info)

    @property
    def brightness(self):
        """Return the brightness of this light between 0..255."""
        state = self.bridge.light_get_state(self.info)
        return int(state['dim'])

    @property
    def is_on(self):
        """True if device is on."""
        state = self.bridge.light_get_state(self.info)
        return int(state['state'])

    def turn_on(self, **kwargs):
        """Turn the light on."""
        dim = kwargs.get(ATTR_BRIGHTNESS, self.brightness)
        self.bridge.light_set_state(self.info, state=1, dim=dim)

    def turn_off(self, **kwargs):
        """Turn the light off."""
        self.bridge.light_set_state(self.info, state=0, dim=0)

    def update(self):
        """Synchronize state with bridge."""
        self.update_lights(no_throttle=True)