mirror of
https://github.com/home-assistant/core.git
synced 2025-07-22 20:57:21 +00:00
Use SelectEntityDescription
for Xiaomi Miio integration (#53907)
* Use SelectEntityDescription * Use SelectEntityDescription * Remove service field from XiaomiMiioSelectDescription class * Fix typo * Use lowercase for options
This commit is contained in:
parent
672a74fa37
commit
2105419a4e
@ -1,11 +1,13 @@
|
|||||||
"""Support led_brightness for Mi Air Humidifier."""
|
"""Support led_brightness for Mi Air Humidifier."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
|
|
||||||
from miio.airhumidifier import LedBrightness as AirhumidifierLedBrightness
|
from miio.airhumidifier import LedBrightness as AirhumidifierLedBrightness
|
||||||
from miio.airhumidifier_miot import LedBrightness as AirhumidifierMiotLedBrightness
|
from miio.airhumidifier_miot import LedBrightness as AirhumidifierMiotLedBrightness
|
||||||
|
|
||||||
from homeassistant.components.select import SelectEntity
|
from homeassistant.components.select import SelectEntity, SelectEntityDescription
|
||||||
from homeassistant.core import callback
|
from homeassistant.core import callback
|
||||||
|
|
||||||
from .const import (
|
from .const import (
|
||||||
@ -18,7 +20,6 @@ from .const import (
|
|||||||
KEY_DEVICE,
|
KEY_DEVICE,
|
||||||
MODELS_HUMIDIFIER_MIIO,
|
MODELS_HUMIDIFIER_MIIO,
|
||||||
MODELS_HUMIDIFIER_MIOT,
|
MODELS_HUMIDIFIER_MIOT,
|
||||||
SERVICE_SET_LED_BRIGHTNESS,
|
|
||||||
)
|
)
|
||||||
from .device import XiaomiCoordinatedMiioEntity
|
from .device import XiaomiCoordinatedMiioEntity
|
||||||
|
|
||||||
@ -34,23 +35,19 @@ LED_BRIGHTNESS_REVERSE_MAP_MIOT = {
|
|||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class SelectorType:
|
class XiaomiMiioSelectDescription(SelectEntityDescription):
|
||||||
"""Class that holds device specific info for a xiaomi aqara or humidifier selectors."""
|
"""A class that describes select entities."""
|
||||||
|
|
||||||
name: str = None
|
options: tuple = ()
|
||||||
icon: str = None
|
|
||||||
short_name: str = None
|
|
||||||
options: list = None
|
|
||||||
service: str = None
|
|
||||||
|
|
||||||
|
|
||||||
SELECTOR_TYPES = {
|
SELECTOR_TYPES = {
|
||||||
FEATURE_SET_LED_BRIGHTNESS: SelectorType(
|
FEATURE_SET_LED_BRIGHTNESS: XiaomiMiioSelectDescription(
|
||||||
name="Led brightness",
|
key=ATTR_LED_BRIGHTNESS,
|
||||||
|
name="Led Brightness",
|
||||||
icon="mdi:brightness-6",
|
icon="mdi:brightness-6",
|
||||||
short_name=ATTR_LED_BRIGHTNESS,
|
device_class="xiaomi_miio__led_brightness",
|
||||||
options=["Bright", "Dim", "Off"],
|
options=("bright", "dim", "off"),
|
||||||
service=SERVICE_SET_LED_BRIGHTNESS,
|
|
||||||
),
|
),
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -71,15 +68,15 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
|
|||||||
else:
|
else:
|
||||||
return
|
return
|
||||||
|
|
||||||
selector = SELECTOR_TYPES[FEATURE_SET_LED_BRIGHTNESS]
|
description = SELECTOR_TYPES[FEATURE_SET_LED_BRIGHTNESS]
|
||||||
entities.append(
|
entities.append(
|
||||||
entity_class(
|
entity_class(
|
||||||
f"{config_entry.title} {selector.name}",
|
f"{config_entry.title} {description.name}",
|
||||||
device,
|
device,
|
||||||
config_entry,
|
config_entry,
|
||||||
f"{selector.short_name}_{config_entry.unique_id}",
|
f"{description.key}_{config_entry.unique_id}",
|
||||||
selector,
|
|
||||||
hass.data[DOMAIN][config_entry.entry_id][KEY_COORDINATOR],
|
hass.data[DOMAIN][config_entry.entry_id][KEY_COORDINATOR],
|
||||||
|
description,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -89,12 +86,11 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
|
|||||||
class XiaomiSelector(XiaomiCoordinatedMiioEntity, SelectEntity):
|
class XiaomiSelector(XiaomiCoordinatedMiioEntity, SelectEntity):
|
||||||
"""Representation of a generic Xiaomi attribute selector."""
|
"""Representation of a generic Xiaomi attribute selector."""
|
||||||
|
|
||||||
def __init__(self, name, device, entry, unique_id, selector, coordinator):
|
def __init__(self, name, device, entry, unique_id, coordinator, description):
|
||||||
"""Initialize the generic Xiaomi attribute selector."""
|
"""Initialize the generic Xiaomi attribute selector."""
|
||||||
super().__init__(name, device, entry, unique_id, coordinator)
|
super().__init__(name, device, entry, unique_id, coordinator)
|
||||||
self._attr_icon = selector.icon
|
self._attr_options = list(description.options)
|
||||||
self._controller = selector
|
self.entity_description = description
|
||||||
self._attr_options = self._controller.options
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _extract_value_from_attribute(state, attribute):
|
def _extract_value_from_attribute(state, attribute):
|
||||||
@ -108,33 +104,33 @@ class XiaomiSelector(XiaomiCoordinatedMiioEntity, SelectEntity):
|
|||||||
class XiaomiAirHumidifierSelector(XiaomiSelector):
|
class XiaomiAirHumidifierSelector(XiaomiSelector):
|
||||||
"""Representation of a Xiaomi Air Humidifier selector."""
|
"""Representation of a Xiaomi Air Humidifier selector."""
|
||||||
|
|
||||||
def __init__(self, name, device, entry, unique_id, controller, coordinator):
|
def __init__(self, name, device, entry, unique_id, coordinator, description):
|
||||||
"""Initialize the plug switch."""
|
"""Initialize the plug switch."""
|
||||||
super().__init__(name, device, entry, unique_id, controller, coordinator)
|
super().__init__(name, device, entry, unique_id, coordinator, description)
|
||||||
self._current_led_brightness = self._extract_value_from_attribute(
|
self._current_led_brightness = self._extract_value_from_attribute(
|
||||||
self.coordinator.data, self._controller.short_name
|
self.coordinator.data, self.entity_description.key
|
||||||
)
|
)
|
||||||
|
|
||||||
@callback
|
@callback
|
||||||
def _handle_coordinator_update(self):
|
def _handle_coordinator_update(self):
|
||||||
"""Fetch state from the device."""
|
"""Fetch state from the device."""
|
||||||
self._current_led_brightness = self._extract_value_from_attribute(
|
self._current_led_brightness = self._extract_value_from_attribute(
|
||||||
self.coordinator.data, self._controller.short_name
|
self.coordinator.data, self.entity_description.key
|
||||||
)
|
)
|
||||||
self.async_write_ha_state()
|
self.async_write_ha_state()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def current_option(self):
|
def current_option(self):
|
||||||
"""Return the current option."""
|
"""Return the current option."""
|
||||||
return self.led_brightness
|
return self.led_brightness.lower()
|
||||||
|
|
||||||
async def async_select_option(self, option: str) -> None:
|
async def async_select_option(self, option: str) -> None:
|
||||||
"""Set an option of the miio device."""
|
"""Set an option of the miio device."""
|
||||||
if option not in self.options:
|
if option not in self.options:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"Selection '{option}' is not a valid {self._controller.name}"
|
f"Selection '{option}' is not a valid {self.entity_description.name}"
|
||||||
)
|
)
|
||||||
await self.async_set_led_brightness(option)
|
await self.async_set_led_brightness(option.title())
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def led_brightness(self):
|
def led_brightness(self):
|
||||||
|
9
homeassistant/components/xiaomi_miio/strings.select.json
Normal file
9
homeassistant/components/xiaomi_miio/strings.select.json
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
{
|
||||||
|
"state": {
|
||||||
|
"xiaomi_miio__led_brightness": {
|
||||||
|
"bright": "Bright",
|
||||||
|
"dim": "Dim",
|
||||||
|
"off": "Off"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user