jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 14:17:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement base entity class for Twinkly (#134382)

Browse Source 
* implement base entity class for twinkly

* Update homeassistant/components/twinkly/entity.py

Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>

* super init

---------

Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
This commit is contained in:
Sven Naumann 2025-01-01 13:49:13 +01:00 committed by GitHub
parent 18e8a3b185
commit 55dc4b0d2c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 31 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
homeassistant/components/twinkly/entity.py Normal file
View File

@ -0,0 +1,27 @@
"""Base entity for Twinkly."""
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DEV_MODEL, DEV_NAME, DOMAIN
from .coordinator import TwinklyCoordinator
class TwinklyEntity(CoordinatorEntity[TwinklyCoordinator]):
"""Defines a base Twinkly entity."""
_attr_has_entity_name = True
def __init__(self, coordinator: TwinklyCoordinator) -> None:
"""Initialize."""
super().__init__(coordinator)
device_info = coordinator.data.device_info
mac = device_info["mac"]
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, mac)},
connections={(CONNECTION_NETWORK_MAC, mac)},
manufacturer="LEDWORKS",
model=device_info[DEV_MODEL],
name=device_info[DEV_NAME],
sw_version=coordinator.software_version,
)

25
homeassistant/components/twinkly/light.py
View File

@ -15,19 +15,11 @@ from homeassistant.components.light import (
LightEntityFeature, LightEntityFeature,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import TwinklyConfigEntry, TwinklyCoordinator from . import TwinklyConfigEntry, TwinklyCoordinator
from .const import ( from .const import DEV_LED_PROFILE, DEV_PROFILE_RGB, DEV_PROFILE_RGBW
DEV_LED_PROFILE, from .entity import TwinklyEntity
DEV_MODEL,
DEV_NAME,
DEV_PROFILE_RGB,
DEV_PROFILE_RGBW,
DOMAIN,
)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -43,10 +35,9 @@ async def async_setup_entry(
async_add_entities([entity], update_before_add=True) async_add_entities([entity], update_before_add=True)
class TwinklyLight(CoordinatorEntity[TwinklyCoordinator], LightEntity): class TwinklyLight(TwinklyEntity, LightEntity):
"""Implementation of the light for the Twinkly service.""" """Implementation of the light for the Twinkly service."""
_attr_has_entity_name = True
_attr_name = None _attr_name = None
_attr_translation_key = "light" _attr_translation_key = "light"
@ -54,7 +45,7 @@ class TwinklyLight(CoordinatorEntity[TwinklyCoordinator], LightEntity):
"""Initialize a TwinklyLight entity.""" """Initialize a TwinklyLight entity."""
super().__init__(coordinator) super().__init__(coordinator)
device_info = coordinator.data.device_info device_info = coordinator.data.device_info
self._attr_unique_id = mac = device_info["mac"] self._attr_unique_id = device_info["mac"]
if device_info.get(DEV_LED_PROFILE) == DEV_PROFILE_RGBW: if device_info.get(DEV_LED_PROFILE) == DEV_PROFILE_RGBW:
self._attr_supported_color_modes = {ColorMode.RGBW} self._attr_supported_color_modes = {ColorMode.RGBW}
@ -68,14 +59,6 @@ class TwinklyLight(CoordinatorEntity[TwinklyCoordinator], LightEntity):
self._attr_supported_color_modes = {ColorMode.BRIGHTNESS} self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
self._attr_color_mode = ColorMode.BRIGHTNESS self._attr_color_mode = ColorMode.BRIGHTNESS
self.client = coordinator.client self.client = coordinator.client
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, mac)},
connections={(CONNECTION_NETWORK_MAC, mac)},
manufacturer="LEDWORKS",
model=device_info[DEV_MODEL],
name=device_info[DEV_NAME],
sw_version=coordinator.software_version,
)
if coordinator.supports_effects: if coordinator.supports_effects:
self._attr_supported_features = LightEntityFeature.EFFECT self._attr_supported_features = LightEntityFeature.EFFECT
self._update_attr() self._update_attr()