chickliu/ha_xiaomi_home
1
0
Fork 0
mirror of https://github.com/XiaoMi/ha_xiaomi_home.git synced 2026-01-14 13:20:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 84e79c3435 into 5d4b975f85

Browse Source
This commit is contained in:
Li Shuzhen 2025-01-07 20:49:18 +08:00 committed by GitHub
commit a2a24eb47a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
custom_components/xiaomi_home/water_heater.py
View File

@ -93,7 +93,7 @@ class WaterHeater(MIoTServiceEntity, WaterHeaterEntity):
_prop_target_temp: Optional[MIoTSpecProperty]
_prop_mode: Optional[MIoTSpecProperty]
_mode_list: Optional[dict[Any, Any]]
_mode_map: Optional[dict[int, str]]
def __init__(
self, miot_device: MIoTDevice, entity_data: MIoTEntityData
@ -113,6 +113,8 @@ class WaterHeater(MIoTServiceEntity, WaterHeaterEntity):
# on
if prop.name == 'on':
self._prop_on = prop
self._attr_supported_features |= (
WaterHeaterEntityFeature.ON_OFF)
# temperature
if prop.name == 'temperature':
if isinstance(prop.value_range, dict):
@ -145,10 +147,10 @@ class WaterHeater(MIoTServiceEntity, WaterHeaterEntity):
_LOGGER.error(
'mode value_list is None, %s', self.entity_id)
continue
self._mode_list = {
self._mode_map = {
item['value']: item['description']
for item in prop.value_list}
self._attr_operation_list = list(self._mode_list.values())
self._attr_operation_list = list(self._mode_map.values())
self._attr_supported_features |= (
WaterHeaterEntityFeature.OPERATION_MODE)
self._prop_mode = prop
@ -184,7 +186,8 @@ class WaterHeater(MIoTServiceEntity, WaterHeaterEntity):
prop=self._prop_on, value=True, update=False)
await self.set_property_async(
prop=self._prop_mode,
value=self.__get_mode_value(description=operation_mode))
value=self.get_map_value(
map_=self._mode_map, description=operation_mode))
async def async_turn_away_mode_on(self) -> None:
"""Set the water heater to away mode."""
@ -207,20 +210,4 @@ class WaterHeater(MIoTServiceEntity, WaterHeaterEntity):
return STATE_OFF
if not self._prop_mode and self.get_prop_value(prop=self._prop_on):
return STATE_ON
return self.__get_mode_description(
key=self.get_prop_value(prop=self._prop_mode))
def __get_mode_description(self, key: int) -> Optional[str]:
"""Convert mode value to description."""
if self._mode_list is None:
return None
return self._mode_list.get(key, None)
def __get_mode_value(self, description: str) -> Optional[int]:
"""Convert mode description to value."""
if self._mode_list is None:
return None
for key, value in self._mode_list.items():
if value == description:
return key
return None
return self._mode_map[self.get_prop_value(prop=self._prop_mode)]