ha_gehome/ge_kitchen/entities/fridge/ge_abstract_fridge.py

165 lines
5.7 KiB
Python
Raw Normal View History

"""GE Kitchen Sensor Entities - Abstract Fridge"""
import sys
import os
import abc
import logging
from typing import Any, Dict, List, Optional
from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
from gekitchensdk import (
ErdCode,
ErdOnOff,
ErdFullNotFull,
FridgeDoorStatus,
FridgeSetPointLimits,
FridgeSetPoints,
FridgeIceBucketStatus,
IceMakerControlStatus
)
from ge_kitchen.const import DOMAIN
from ..common import GeWaterHeater
from .const import *
_LOGGER = logging.getLogger(__name__)
class GeAbstractFridge(GeWaterHeater):
"""Mock a fridge or freezer as a water heater."""
@property
def heater_type(self) -> str:
raise NotImplementedError
@property
def turbo_erd_code(self) -> str:
raise NotImplementedError
@property
def turbo_mode(self) -> str:
raise NotImplementedError
@property
def operation_list(self) -> List[str]:
return [OP_MODE_NORMAL, OP_MODE_SABBATH, self.turbo_mode]
@property
def unique_id(self) -> str:
2020-12-17 21:26:42 -07:00
return f"{DOMAIN}_{self.serial_number}_{self.heater_type}"
@property
def name(self) -> Optional[str]:
2020-12-17 22:00:48 -07:00
return f"{self.serial_number} {self.heater_type.title()}"
@property
def target_temps(self) -> FridgeSetPoints:
"""Get the current temperature settings tuple."""
return self.appliance.get_erd_value(ErdCode.TEMPERATURE_SETTING)
@property
def target_temperature(self) -> int:
"""Return the temperature we try to reach."""
return getattr(self.target_temps, self.heater_type)
@property
def current_temperature(self) -> int:
"""Return the current temperature."""
current_temps = self.appliance.get_erd_value(ErdCode.CURRENT_TEMPERATURE)
current_temp = getattr(current_temps, self.heater_type)
if current_temp is None:
_LOGGER.exception(f"{self.name} has None for current_temperature (available: {self.available})!")
return current_temp
async def async_set_temperature(self, **kwargs):
target_temp = kwargs.get(ATTR_TEMPERATURE)
if target_temp is None:
return
if not self.min_temp <= target_temp <= self.max_temp:
raise ValueError("Tried to set temperature out of device range")
if self.heater_type == HEATER_TYPE_FRIDGE:
new_temp = FridgeSetPoints(fridge=target_temp, freezer=self.target_temps.freezer)
elif self.heater_type == HEATER_TYPE_FREEZER:
new_temp = FridgeSetPoints(fridge=self.target_temps.fridge, freezer=target_temp)
else:
raise ValueError("Invalid heater_type")
await self.appliance.async_set_erd_value(ErdCode.TEMPERATURE_SETTING, new_temp)
@property
def supported_features(self):
return GE_FRIDGE_SUPPORT
@property
def setpoint_limits(self) -> FridgeSetPointLimits:
return self.appliance.get_erd_value(ErdCode.SETPOINT_LIMITS)
@property
def min_temp(self):
"""Return the minimum temperature."""
return getattr(self.setpoint_limits, f"{self.heater_type}_min")
@property
def max_temp(self):
"""Return the maximum temperature."""
return getattr(self.setpoint_limits, f"{self.heater_type}_max")
@property
def current_operation(self) -> str:
"""Get the current operation mode."""
if self.appliance.get_erd_value(ErdCode.SABBATH_MODE):
return OP_MODE_SABBATH
if self.appliance.get_erd_value(self.turbo_erd_code):
return self.turbo_mode
return OP_MODE_NORMAL
async def async_set_sabbath_mode(self, sabbath_on: bool = True):
"""Set sabbath mode if it's changed"""
if self.appliance.get_erd_value(ErdCode.SABBATH_MODE) == sabbath_on:
return
await self.appliance.async_set_erd_value(ErdCode.SABBATH_MODE, sabbath_on)
async def async_set_operation_mode(self, operation_mode):
"""Set the operation mode."""
if operation_mode not in self.operation_list:
raise ValueError("Invalid operation mode")
if operation_mode == self.current_operation:
return
sabbath_mode = operation_mode == OP_MODE_SABBATH
await self.async_set_sabbath_mode(sabbath_mode)
if not sabbath_mode:
await self.appliance.async_set_erd_value(self.turbo_erd_code, operation_mode == self.turbo_mode)
@property
def door_status(self) -> FridgeDoorStatus:
"""Shorthand to get door status."""
return self.appliance.get_erd_value(ErdCode.DOOR_STATUS)
@property
def ice_maker_state_attrs(self) -> Dict[str, Any]:
"""Get state attributes for the ice maker, if applicable."""
data = {}
erd_val: FridgeIceBucketStatus = self.appliance.get_erd_value(ErdCode.ICE_MAKER_BUCKET_STATUS)
ice_bucket_status = getattr(erd_val, f"state_full_{self.heater_type}")
if ice_bucket_status != ErdFullNotFull.NA:
2020-12-29 08:03:37 -07:00
data["ice_bucket"] = self._stringify(ice_bucket_status)
erd_val: IceMakerControlStatus = self.appliance.get_erd_value(ErdCode.ICE_MAKER_CONTROL)
ice_control_status = getattr(erd_val, f"status_{self.heater_type}")
if ice_control_status != ErdOnOff.NA:
2020-12-29 08:03:37 -07:00
data["ice_maker"] = self._stringify(ice_control_status)
return data
@property
def door_state_attrs(self) -> Dict[str, Any]:
"""Get state attributes for the doors."""
return {}
@property
def device_state_attributes(self) -> Dict[str, Any]:
door_attrs = self.door_state_attrs
ice_maker_attrs = self.ice_maker_state_attrs
other_attrs = self.other_state_attrs
return {**door_attrs, **ice_maker_attrs, **other_attrs}