evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add entity translations to Flux led (#95355)

Browse source
This commit is contained in:
Joost Lekkerkerker 2023-06-27 16:58:57 +02:00 • committed by GitHub
parent 90854df5b2
commit e19b29d6ae
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 104 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

32
homeassistant/components/flux_led/select.py
View file

@ -52,30 +52,22 @@ async def async_setup_entry(
| FluxRemoteConfigSelect
| FluxWhiteChannelSelect
] = []
name = entry.data.get(CONF_NAME, entry.title)
entry.data.get(CONF_NAME, entry.title)
base_unique_id = entry.unique_id or entry.entry_id
if device.device_type == DeviceType.Switch:
entities.append(FluxPowerStateSelect(coordinator.device, entry))
if device.operating_modes:
entities.append(
FluxOperatingModesSelect(
coordinator, base_unique_id, f"{name} Operating Mode", "operating_mode"
)
FluxOperatingModesSelect(coordinator, base_unique_id, "operating_mode")
)
if device.wirings and device.wiring is not None:
entities.append(
FluxWiringsSelect(coordinator, base_unique_id, f"{name} Wiring", "wiring")
)
entities.append(FluxWiringsSelect(coordinator, base_unique_id, "wiring"))
if device.ic_types:
entities.append(
FluxICTypeSelect(coordinator, base_unique_id, f"{name} IC Type", "ic_type")
)
entities.append(FluxICTypeSelect(coordinator, base_unique_id, "ic_type"))
if device.remote_config:
entities.append(
FluxRemoteConfigSelect(
coordinator, base_unique_id, f"{name} Remote Config", "remote_config"
)
FluxRemoteConfigSelect(coordinator, base_unique_id, "remote_config")
)
if FLUX_COLOR_MODE_RGBW in device.color_modes:
entities.append(FluxWhiteChannelSelect(coordinator.device, entry))
@ -98,6 +90,7 @@ class FluxConfigSelect(FluxEntity, SelectEntity):
class FluxPowerStateSelect(FluxConfigAtStartSelect, SelectEntity):
"""Representation of a Flux power restore state option."""
_attr_translation_key = "power_restored"
_attr_icon = "mdi:transmission-tower-off"
_attr_options = list(NAME_TO_POWER_RESTORE_STATE)
@ -108,7 +101,6 @@ class FluxPowerStateSelect(FluxConfigAtStartSelect, SelectEntity):
) -> None:
"""Initialize the power state select."""
super().__init__(device, entry)
self._attr_name = f"{entry.data.get(CONF_NAME, entry.title)} Power Restored"
base_unique_id = entry.unique_id or entry.entry_id
self._attr_unique_id = f"{base_unique_id}_power_restored"
self._async_set_current_option_from_device()
@ -134,6 +126,7 @@ class FluxICTypeSelect(FluxConfigSelect):
"""Representation of Flux ic type."""
_attr_icon = "mdi:chip"
_attr_translation_key = "ic_type"
@property
def options(self) -> list[str]:
@ -156,6 +149,7 @@ class FluxWiringsSelect(FluxConfigSelect):
"""Representation of Flux wirings."""
_attr_icon = "mdi:led-strip-variant"
_attr_translation_key = "wiring"
@property
def options(self) -> list[str]:
@ -176,6 +170,8 @@ class FluxWiringsSelect(FluxConfigSelect):
class FluxOperatingModesSelect(FluxConfigSelect):
"""Representation of Flux operating modes."""
_attr_translation_key = "operating_mode"
@property
def options(self) -> list[str]:
"""Return the current operating mode."""
@ -196,15 +192,16 @@ class FluxOperatingModesSelect(FluxConfigSelect):
class FluxRemoteConfigSelect(FluxConfigSelect):
"""Representation of Flux remote config type."""
_attr_translation_key = "remote_config"
def __init__(
self,
coordinator: FluxLedUpdateCoordinator,
base_unique_id: str,
name: str,
key: str,
) -> None:
"""Initialize the remote config type select."""
super().__init__(coordinator, base_unique_id, name, key)
super().__init__(coordinator, base_unique_id, key)
assert self._device.remote_config is not None
self._name_to_state = {
_human_readable_option(option.name): option for option in RemoteConfig
@ -226,6 +223,8 @@ class FluxRemoteConfigSelect(FluxConfigSelect):
class FluxWhiteChannelSelect(FluxConfigAtStartSelect):
"""Representation of Flux white channel."""
_attr_translation_key = "white_channel"
_attr_options = [_human_readable_option(option.name) for option in WhiteChannelType]
def __init__(
@ -235,7 +234,6 @@ class FluxWhiteChannelSelect(FluxConfigAtStartSelect):
) -> None:
"""Initialize the white channel select."""
super().__init__(device, entry)
self._attr_name = f"{entry.data.get(CONF_NAME, entry.title)} White Channel"
base_unique_id = entry.unique_id or entry.entry_id
self._attr_unique_id = f"{base_unique_id}_white_channel"