ha_gehome/entities/hood/ge_hood_fan_speed.py

47 lines
1.9 KiB
Python

import logging
from typing import List, Any, Optional
from gehomesdk import ErdCodeType, ErdHoodFanSpeedAvailability, ErdHoodFanSpeed, ErdCode
from ...devices import ApplianceApi
from ..common import GeErdSelect, OptionsConverter
_LOGGER = logging.getLogger(__name__)
class HoodFanSpeedOptionsConverter(OptionsConverter):
def __init__(self, availability: ErdHoodFanSpeedAvailability):
super().__init__()
self.availability = availability
self.excluded_speeds = []
if not availability.off_available:
self.excluded_speeds.append(ErdHoodFanSpeed.OFF)
if not availability.low_available:
self.excluded_speeds.append(ErdHoodFanSpeed.LOW)
if not availability.med_available:
self.excluded_speeds.append(ErdHoodFanSpeed.MEDIUM)
if not availability.high_available:
self.excluded_speeds.append(ErdHoodFanSpeed.HIGH)
if not availability.boost_available:
self.excluded_speeds.append(ErdHoodFanSpeed.BOOST)
@property
def options(self) -> List[str]:
return [i.stringify() for i in ErdHoodFanSpeed if i not in self.excluded_speeds]
def from_option_string(self, value: str) -> Any:
try:
return ErdHoodFanSpeed[value.upper()]
except:
_LOGGER.warning(f"Could not set hood fan speed to {value.upper()}")
return ErdHoodFanSpeed.OFF
def to_option_string(self, value: ErdHoodFanSpeed) -> Optional[str]:
try:
if value is not None:
return value.stringify()
except:
pass
return ErdHoodFanSpeed.OFF.stringify()
class GeHoodFanSpeedSelect(GeErdSelect):
def __init__(self, api: ApplianceApi, erd_code: ErdCodeType):
self._availability: ErdHoodFanSpeedAvailability = api.try_get_erd_value(ErdCode.HOOD_FAN_SPEED_AVAILABILITY)
super().__init__(api, erd_code, HoodFanSpeedOptionsConverter(self._availability))