From d6747d6aafa51d37356af177715f10156c31ec29 Mon Sep 17 00:00:00 2001
From: Matthew Garrett <mjg59@coreos.com>
Date: Fri, 13 Jan 2017 22:15:43 -0800
Subject: [PATCH] Add support for Zengge Bluetooth bulbs (#5196)

* Add support for Zengge Bluetooth bulbs

Adds support for the Zengge Bluetooth RGBW bulbs. These are sold under a
number of brands, including Flux. The bulbs do not support full RGBW
control - they turn off the RGB LEDs when white is enabled, and vice versa.

* Update zengge.py
---
 .coveragerc                              |   1 +
 homeassistant/components/light/zengge.py | 146 +++++++++++++++++++++++
 requirements_all.txt                     |   3 +
 3 files changed, 150 insertions(+)
 create mode 100644 homeassistant/components/light/zengge.py

diff --git a/.coveragerc b/.coveragerc
index 74faed07c4a..3862b3015e4 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -195,6 +195,7 @@ omit =
     homeassistant/components/light/tikteck.py
     homeassistant/components/light/x10.py
     homeassistant/components/light/yeelight.py
+    homeassistant/components/light/zengge.py
     homeassistant/components/lirc.py
     homeassistant/components/media_player/aquostv.py
     homeassistant/components/media_player/braviatv.py
diff --git a/homeassistant/components/light/zengge.py b/homeassistant/components/light/zengge.py
new file mode 100644
index 00000000000..74992a9d633
--- /dev/null
+++ b/homeassistant/components/light/zengge.py
@@ -0,0 +1,146 @@
+"""
+Support for Zengge lights.
+
+For more details about this platform, please refer to the documentation at
+https://home-assistant.io/components/light.zengge/
+"""
+import logging
+
+import voluptuous as vol
+
+from homeassistant.const import CONF_DEVICES, CONF_NAME
+from homeassistant.components.light import (
+    ATTR_RGB_COLOR, ATTR_WHITE_VALUE,
+    SUPPORT_RGB_COLOR, SUPPORT_WHITE_VALUE, Light, PLATFORM_SCHEMA)
+import homeassistant.helpers.config_validation as cv
+
+REQUIREMENTS = ['zengge==0.2']
+
+_LOGGER = logging.getLogger(__name__)
+
+SUPPORT_ZENGGE_LED = (SUPPORT_RGB_COLOR | SUPPORT_WHITE_VALUE)
+
+DEVICE_SCHEMA = vol.Schema({
+    vol.Optional(CONF_NAME): cv.string,
+})
+
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Optional(CONF_DEVICES, default={}): {cv.string: DEVICE_SCHEMA},
+})
+
+
+# pylint: disable=unused-argument
+def setup_platform(hass, config, add_devices, discovery_info=None):
+    """Set up the Zengge platform."""
+    lights = []
+    for address, device_config in config[CONF_DEVICES].items():
+        device = {}
+        device['name'] = device_config[CONF_NAME]
+        device['address'] = address
+        light = ZenggeLight(device)
+        if light.is_valid:
+            lights.append(light)
+
+    add_devices(lights)
+
+
+class ZenggeLight(Light):
+    """Representation of a Zengge light."""
+
+    def __init__(self, device):
+        """Initialize the light."""
+        import zengge
+
+        self._name = device['name']
+        self._address = device['address']
+        self.is_valid = True
+        self._bulb = zengge.zengge(self._address)
+        self._white = 0
+        self._rgb = (0, 0, 0)
+        self._state = False
+        if self._bulb.connect() is False:
+            self.is_valid = False
+            _LOGGER.error(
+                "Failed to connect to bulb %s, %s", self._address, self._name)
+            return
+        self.update()
+
+    @property
+    def unique_id(self):
+        """Return the ID of this light."""
+        return "{}.{}".format(self.__class__, self._address)
+
+    @property
+    def name(self):
+        """Return the name of the device if any."""
+        return self._name
+
+    @property
+    def is_on(self):
+        """Return true if device is on."""
+        return self._state
+
+    @property
+    def rgb_color(self):
+        """Return the color property."""
+        return self._rgb
+
+    @property
+    def white_value(self):
+        """Return the white property."""
+        return self._white
+
+    @property
+    def supported_features(self):
+        """Flag supported features."""
+        return SUPPORT_ZENGGE_LED
+
+    @property
+    def should_poll(self):
+        """Feel free to poll."""
+        return True
+
+    @property
+    def assumed_state(self):
+        """We can report the actual state."""
+        return False
+
+    def set_rgb(self, red, green, blue):
+        """Set the rgb state."""
+        return self._bulb.set_rgb(red, green, blue)
+
+    def set_white(self, white):
+        """Set the white state."""
+        return self._bulb.set_white(white)
+
+    def turn_on(self, **kwargs):
+        """Turn the specified light on."""
+        self._state = True
+        self._bulb.on()
+
+        rgb = kwargs.get(ATTR_RGB_COLOR)
+        white = kwargs.get(ATTR_WHITE_VALUE)
+
+        if white is not None:
+            self._white = white
+            self._rgb = (0, 0, 0)
+
+        if rgb is not None:
+            self._white = 0
+            self._rgb = rgb
+
+        if self._white != 0:
+            self.set_white(self._white)
+        else:
+            self.set_rgb(self._rgb[0], self._rgb[1], self._rgb[2])
+
+    def turn_off(self, **kwargs):
+        """Turn the specified light off."""
+        self._state = False
+        self._bulb.off()
+
+    def update(self):
+        """Synchronise internal state with the actual light state."""
+        self._rgb = self._bulb.get_colour()
+        self._white = self._bulb.get_white()
+        self._state = self._bulb.get_on()
diff --git a/requirements_all.txt b/requirements_all.txt
index 1b13985963a..099711751ce 100755
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -659,5 +659,8 @@ yahoo-finance==1.4.0
 # homeassistant.components.sensor.yweather
 yahooweather==0.8
 
+# homeassistant.components.light.zengge
+zengge==0.2
+
 # homeassistant.components.zeroconf
 zeroconf==0.17.6