123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
"""Support for Yeelight Sunflower color bulbs (not Yeelight Blue or WiFi)."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
import yeelightsunflower
|
|
|
|
from homeassistant.components.light import (
|
|
ATTR_BRIGHTNESS,
|
|
ATTR_HS_COLOR,
|
|
PLATFORM_SCHEMA,
|
|
SUPPORT_BRIGHTNESS,
|
|
SUPPORT_COLOR,
|
|
LightEntity,
|
|
)
|
|
from homeassistant.const import CONF_HOST
|
|
from homeassistant.core import HomeAssistant
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
import homeassistant.util.color as color_util
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SUPPORT_YEELIGHT_SUNFLOWER = SUPPORT_BRIGHTNESS | SUPPORT_COLOR
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({vol.Required(CONF_HOST): cv.string})
|
|
|
|
|
|
def setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the Yeelight Sunflower Light platform."""
|
|
host = config.get(CONF_HOST)
|
|
hub = yeelightsunflower.Hub(host)
|
|
|
|
if not hub.available:
|
|
_LOGGER.error("Could not connect to Yeelight Sunflower hub")
|
|
return
|
|
|
|
add_entities(SunflowerBulb(light) for light in hub.get_lights())
|
|
|
|
|
|
class SunflowerBulb(LightEntity):
|
|
"""Representation of a Yeelight Sunflower Light."""
|
|
|
|
def __init__(self, light):
|
|
"""Initialize a Yeelight Sunflower bulb."""
|
|
self._light = light
|
|
self._available = light.available
|
|
self._brightness = light.brightness
|
|
self._is_on = light.is_on
|
|
self._rgb_color = light.rgb_color
|
|
self._unique_id = light.zid
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the display name of this light."""
|
|
return f"sunflower_{self._light.zid}"
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the unique ID of this light."""
|
|
return self._unique_id
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return True if entity is available."""
|
|
return self._available
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if light is on."""
|
|
return self._is_on
|
|
|
|
@property
|
|
def brightness(self):
|
|
"""Return the brightness is 0-255; Yeelight's brightness is 0-100."""
|
|
return int(self._brightness / 100 * 255)
|
|
|
|
@property
|
|
def hs_color(self):
|
|
"""Return the color property."""
|
|
return color_util.color_RGB_to_hs(*self._rgb_color)
|
|
|
|
@property
|
|
def supported_features(self):
|
|
"""Flag supported features."""
|
|
return SUPPORT_YEELIGHT_SUNFLOWER
|
|
|
|
def turn_on(self, **kwargs):
|
|
"""Instruct the light to turn on, optionally set colour/brightness."""
|
|
# when no arguments, just turn light on (full brightness)
|
|
if not kwargs:
|
|
self._light.turn_on()
|
|
else:
|
|
if ATTR_HS_COLOR in kwargs and ATTR_BRIGHTNESS in kwargs:
|
|
rgb = color_util.color_hs_to_RGB(*kwargs[ATTR_HS_COLOR])
|
|
bright = int(kwargs[ATTR_BRIGHTNESS] / 255 * 100)
|
|
self._light.set_all(rgb[0], rgb[1], rgb[2], bright)
|
|
elif ATTR_HS_COLOR in kwargs:
|
|
rgb = color_util.color_hs_to_RGB(*kwargs[ATTR_HS_COLOR])
|
|
self._light.set_rgb_color(rgb[0], rgb[1], rgb[2])
|
|
elif ATTR_BRIGHTNESS in kwargs:
|
|
bright = int(kwargs[ATTR_BRIGHTNESS] / 255 * 100)
|
|
self._light.set_brightness(bright)
|
|
|
|
def turn_off(self, **kwargs):
|
|
"""Instruct the light to turn off."""
|
|
self._light.turn_off()
|
|
|
|
def update(self):
|
|
"""Fetch new state data for this light and update local values."""
|
|
self._light.update()
|
|
self._available = self._light.available
|
|
self._brightness = self._light.brightness
|
|
self._is_on = self._light.is_on
|
|
self._rgb_color = self._light.rgb_color
|