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

Use enums to define choices in tolo (#113992)

Browse source 
* use enums defined in HA to define choices

* move enums to .const
This commit is contained in:
Matthias Lohr 2024-03-25 12:47:56 +01:00 • committed by GitHub
parent 49a4fe41c6
commit 95ddade83c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 23 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
homeassistant/components/tolo/const.py
View file

@ -1,7 +1,24 @@
"""Constants for the tolo integration."""
from enum import Enum
from tololib import AromaTherapySlot as ToloAromaTherapySlot, LampMode as ToloLampMode
DOMAIN = "tolo"
DEFAULT_NAME = "TOLO Sauna"
DEFAULT_RETRY_TIMEOUT = 1
DEFAULT_RETRY_COUNT = 3
class AromaTherapySlot(Enum):
"""Mapping to TOLO Aroma Therapy Slot."""
A = ToloAromaTherapySlot.A
B = ToloAromaTherapySlot.B
class LampMode(Enum):
"""Mapping to TOLO Lamp Mode."""
MANUAL = ToloLampMode.MANUAL
AUTOMATIC = ToloLampMode.AUTOMATIC

10
homeassistant/components/tolo/select.py
View file

@ -5,7 +5,7 @@ from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
from tololib import AromaTherapySlot, LampMode, ToloClient, ToloSettings
from tololib import ToloClient, ToloSettings
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.config_entries import ConfigEntry
@ -14,7 +14,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
from .const import DOMAIN
from .const import DOMAIN, AromaTherapySlot, LampMode
@dataclass(frozen=True, kw_only=True)
@ -32,7 +32,9 @@ SELECTS = (
translation_key="lamp_mode",
options=[lamp_mode.name.lower() for lamp_mode in LampMode],
getter=lambda settings: settings.lamp_mode.name.lower(),
setter=lambda client, option: client.set_lamp_mode(LampMode[option.upper()]),
setter=lambda client, option: client.set_lamp_mode(
LampMode[option.upper()].value
),
),
ToloSelectEntityDescription(
key="aroma_therapy_slot",
@ -42,7 +44,7 @@ SELECTS = (
],
getter=lambda settings: settings.aroma_therapy_slot.name.lower(),
setter=lambda client, option: client.set_aroma_therapy_slot(
AromaTherapySlot[option.upper()]
AromaTherapySlot[option.upper()].value
),
),
)