27 lines
875 B
Python
27 lines
875 B
Python
import logging
|
|
from typing import List, Any, Optional
|
|
|
|
from gehomesdk import ErdWaterHeaterMode
|
|
from ..common import OptionsConverter
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
class WhHeaterModeConverter(OptionsConverter):
|
|
@property
|
|
def options(self) -> List[str]:
|
|
return [i.stringify() for i in ErdWaterHeaterMode]
|
|
def from_option_string(self, value: str) -> Any:
|
|
enum_val = value.upper().replace(" ","_")
|
|
try:
|
|
return ErdWaterHeaterMode[enum_val]
|
|
except:
|
|
_LOGGER.warning(f"Could not heater mode to {enum_val}")
|
|
return ErdWaterHeaterMode.UNKNOWN
|
|
def to_option_string(self, value: ErdWaterHeaterMode) -> Optional[str]:
|
|
try:
|
|
if value is not None:
|
|
return value.stringify()
|
|
except:
|
|
pass
|
|
return ErdWaterHeaterMode.UNKNOWN.stringify()
|