home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-08 07:05:07 +02:00
Code Issues Packages Projects Releases Wiki Activity

Improve Shelly color mode switch for dual mode bulbs (#58971)

Browse Source 
* Shelly color mode switch for SHCB-1

* Update light.py

* Update light.py

* Update homeassistant/components/shelly/light.py

Co-authored-by: Shay Levy <levyshay1@gmail.com>

* Update light.py

Co-authored-by: Shay Levy <levyshay1@gmail.com>
This commit is contained in:
Gábor Kiss
2021-11-09 18:24:40 +01:00
committed by GitHub
parent 59a0e5a773
commit 032786fcd8
1 changed files with 6 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
homeassistant/components/shelly/light.py
View File

@@ -1,12 +1,10 @@
"""Light for Shelly.""" """Light for Shelly."""
from __future__ import annotations from __future__ import annotations
import asyncio
import logging import logging
from typing import Any, Final, cast from typing import Any, Final, cast
from aioshelly.block_device import Block from aioshelly.block_device import Block
import async_timeout
from homeassistant.components.light import ( from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_BRIGHTNESS,
@@ -35,10 +33,10 @@ from homeassistant.util.color import (
from . import BlockDeviceWrapper, RpcDeviceWrapper from . import BlockDeviceWrapper, RpcDeviceWrapper
from .const import ( from .const import (
AIOSHELLY_DEVICE_TIMEOUT_SEC,
BLOCK, BLOCK,
DATA_CONFIG_ENTRY, DATA_CONFIG_ENTRY,
DOMAIN, DOMAIN,
DUAL_MODE_LIGHT_MODELS,
FIRMWARE_PATTERN, FIRMWARE_PATTERN,
KELVIN_MAX_VALUE, KELVIN_MAX_VALUE,
KELVIN_MIN_VALUE_COLOR, KELVIN_MIN_VALUE_COLOR,
@@ -136,7 +134,6 @@ class BlockShellyLight(ShellyBlockEntity, LightEntity):
"""Initialize light.""" """Initialize light."""
super().__init__(wrapper, block) super().__init__(wrapper, block)
self.control_result: dict[str, Any] | None = None self.control_result: dict[str, Any] | None = None
self.mode_result: dict[str, Any] | None = None
self._supported_color_modes: set[str] = set() self._supported_color_modes: set[str] = set()
self._supported_features: int = 0 self._supported_features: int = 0
self._min_kelvin: int = KELVIN_MIN_VALUE_WHITE self._min_kelvin: int = KELVIN_MIN_VALUE_WHITE
@@ -185,8 +182,8 @@ class BlockShellyLight(ShellyBlockEntity, LightEntity):
@property @property
def mode(self) -> str: def mode(self) -> str:
"""Return the color mode of the light.""" """Return the color mode of the light."""
if self.mode_result: if self.control_result and self.control_result.get("mode"):
return cast(str, self.mode_result["mode"]) return cast(str, self.control_result["mode"])
if hasattr(self.block, "mode"): if hasattr(self.block, "mode"):
return cast(str, self.block.mode) return cast(str, self.block.mode)
@@ -369,9 +366,10 @@ class BlockShellyLight(ShellyBlockEntity, LightEntity):
self.wrapper.model, self.wrapper.model,
) )
if await self.set_light_mode(set_mode): if set_mode and self.wrapper.model in DUAL_MODE_LIGHT_MODELS:
self.control_result = await self.set_state(**params) params["mode"] = set_mode
self.control_result = await self.set_state(**params)
self.async_write_ha_state() self.async_write_ha_state()
async def async_turn_off(self, **kwargs: Any) -> None: async def async_turn_off(self, **kwargs: Any) -> None:
@@ -387,32 +385,10 @@ class BlockShellyLight(ShellyBlockEntity, LightEntity):
self.async_write_ha_state() self.async_write_ha_state()
async def set_light_mode(self, set_mode: str | None) -> bool:
"""Change device mode color/white if mode has changed."""
if set_mode is None or self.mode == set_mode:
return True
_LOGGER.debug("Setting light mode for entity %s, mode: %s", self.name, set_mode)
try:
async with async_timeout.timeout(AIOSHELLY_DEVICE_TIMEOUT_SEC):
self.mode_result = await self.wrapper.device.switch_light_mode(set_mode)
except (asyncio.TimeoutError, OSError) as err:
_LOGGER.error(
"Setting light mode for entity %s failed, state: %s, error: %s",
self.name,
set_mode,
repr(err),
)
self.wrapper.last_update_success = False
return False
return True
@callback @callback
def _update_callback(self) -> None: def _update_callback(self) -> None:
"""When device updates, clear control & mode result that overrides state.""" """When device updates, clear control & mode result that overrides state."""
self.control_result = None self.control_result = None
self.mode_result = None
super()._update_callback() super()._update_callback()