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

Cleanup the Fritz!Smarthome light entity (#123488)

Browse Source 
* use self.data when ever possible

* use short hand attributes for color mode and supported color modes

* color mode can change during runtime

* group executer jobs
This commit is contained in:
Michael 2024-08-18 14:53:35 +02:00 committed by GitHub
parent 04b0760e27
commit 0c5a3fab22
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
homeassistant/components/fritzbox/light.py
View File

@ -20,8 +20,6 @@ from . import FritzboxDataUpdateCoordinator, FritzBoxDeviceEntity
from .const import COLOR_MODE, LOGGER from .const import COLOR_MODE, LOGGER
from .coordinator import FritzboxConfigEntry from .coordinator import FritzboxConfigEntry
SUPPORTED_COLOR_MODES = {ColorMode.COLOR_TEMP, ColorMode.HS}
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
@ -61,6 +59,12 @@ class FritzboxLight(FritzBoxDeviceEntity, LightEntity):
super().__init__(coordinator, ain, None) super().__init__(coordinator, ain, None)
self._supported_hs: dict[int, list[int]] = {} self._supported_hs: dict[int, list[int]] = {}
self._attr_supported_color_modes = {ColorMode.ONOFF}
if self.data.has_color:
self._attr_supported_color_modes = {ColorMode.COLOR_TEMP, ColorMode.HS}
elif self.data.has_level:
self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
@property @property
def is_on(self) -> bool: def is_on(self) -> bool:
"""If the light is currently on or off.""" """If the light is currently on or off."""
@ -95,15 +99,6 @@ class FritzboxLight(FritzBoxDeviceEntity, LightEntity):
return ColorMode.BRIGHTNESS return ColorMode.BRIGHTNESS
return ColorMode.ONOFF return ColorMode.ONOFF
@property
def supported_color_modes(self) -> set[ColorMode]:
"""Flag supported color modes."""
if self.data.has_color:
return SUPPORTED_COLOR_MODES
if self.data.has_level:
return {ColorMode.BRIGHTNESS}
return {ColorMode.ONOFF}
async def async_turn_on(self, **kwargs: Any) -> None: async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the light on.""" """Turn the light on."""
if kwargs.get(ATTR_BRIGHTNESS) is not None: if kwargs.get(ATTR_BRIGHTNESS) is not None:
@ -157,12 +152,14 @@ class FritzboxLight(FritzBoxDeviceEntity, LightEntity):
async def async_added_to_hass(self) -> None: async def async_added_to_hass(self) -> None:
"""Get light attributes from device after entity is added to hass.""" """Get light attributes from device after entity is added to hass."""
await super().async_added_to_hass() await super().async_added_to_hass()
supported_colors = await self.hass.async_add_executor_job(
self.coordinator.data.devices[self.ain].get_colors def _get_color_data() -> tuple[dict, list]:
) return (self.data.get_colors(), self.data.get_color_temps())
supported_color_temps = await self.hass.async_add_executor_job(
self.coordinator.data.devices[self.ain].get_color_temps (
) supported_colors,
supported_color_temps,
) = await self.hass.async_add_executor_job(_get_color_data)
if supported_color_temps: if supported_color_temps:
# only available for color bulbs # only available for color bulbs