mirror of https://github.com/simbaja/ha_gehome.git
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
"""Define all of the entity types"""
|
|
|
|
import logging
|
|
from typing import Any, Dict, Optional, TYPE_CHECKING
|
|
|
|
from gekitchen import ErdCodeType, GeAppliance, translate_erd_code
|
|
from gekitchen.erd_types import *
|
|
from gekitchen.erd_constants import *
|
|
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
|
|
from .const import DOMAIN
|
|
from .erd_string_utils import *
|
|
|
|
if TYPE_CHECKING:
|
|
from .appliance_api import ApplianceApi
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOOR_ERD_CODES = {
|
|
ErdCode.DOOR_STATUS
|
|
}
|
|
RAW_TEMPERATURE_ERD_CODES = {
|
|
ErdCode.LOWER_OVEN_RAW_TEMPERATURE,
|
|
ErdCode.LOWER_OVEN_USER_TEMP_OFFSET,
|
|
ErdCode.UPPER_OVEN_RAW_TEMPERATURE,
|
|
ErdCode.UPPER_OVEN_USER_TEMP_OFFSET,
|
|
ErdCode.CURRENT_TEMPERATURE,
|
|
ErdCode.TEMPERATURE_SETTING,
|
|
}
|
|
NONZERO_TEMPERATURE_ERD_CODES = {
|
|
ErdCode.HOT_WATER_SET_TEMP,
|
|
ErdCode.LOWER_OVEN_DISPLAY_TEMPERATURE,
|
|
ErdCode.LOWER_OVEN_PROBE_DISPLAY_TEMP,
|
|
ErdCode.UPPER_OVEN_DISPLAY_TEMPERATURE,
|
|
ErdCode.UPPER_OVEN_PROBE_DISPLAY_TEMP,
|
|
}
|
|
TEMPERATURE_ERD_CODES = RAW_TEMPERATURE_ERD_CODES.union(NONZERO_TEMPERATURE_ERD_CODES)
|
|
TIMER_ERD_CODES = {
|
|
ErdCode.LOWER_OVEN_ELAPSED_COOK_TIME,
|
|
ErdCode.LOWER_OVEN_KITCHEN_TIMER,
|
|
ErdCode.LOWER_OVEN_DELAY_TIME_REMAINING,
|
|
ErdCode.LOWER_OVEN_COOK_TIME_REMAINING,
|
|
ErdCode.LOWER_OVEN_ELAPSED_COOK_TIME,
|
|
ErdCode.ELAPSED_ON_TIME,
|
|
ErdCode.TIME_REMAINING,
|
|
ErdCode.UPPER_OVEN_ELAPSED_COOK_TIME,
|
|
ErdCode.UPPER_OVEN_KITCHEN_TIMER,
|
|
ErdCode.UPPER_OVEN_DELAY_TIME_REMAINING,
|
|
ErdCode.UPPER_OVEN_ELAPSED_COOK_TIME,
|
|
ErdCode.UPPER_OVEN_COOK_TIME_REMAINING,
|
|
}
|
|
|
|
|
|
def get_erd_units(erd_code: ErdCodeType, measurement_units: ErdMeasurementUnits):
|
|
"""Get the units for a sensor."""
|
|
erd_code = translate_erd_code(erd_code)
|
|
if not measurement_units:
|
|
return None
|
|
|
|
if erd_code in TEMPERATURE_ERD_CODES or erd_code in {ErdCode.LOWER_OVEN_COOK_MODE, ErdCode.UPPER_OVEN_COOK_MODE}:
|
|
if measurement_units == ErdMeasurementUnits.METRIC:
|
|
return TEMP_CELSIUS
|
|
return TEMP_FAHRENHEIT
|
|
return None
|
|
|
|
|
|
def get_erd_icon(erd_code: ErdCodeType, value: Any = None) -> Optional[str]:
|
|
"""Select an appropriate icon."""
|
|
erd_code = translate_erd_code(erd_code)
|
|
if not isinstance(erd_code, ErdCode):
|
|
return None
|
|
if erd_code in TIMER_ERD_CODES:
|
|
return "mdi:timer-outline"
|
|
if erd_code in {
|
|
ErdCode.LOWER_OVEN_COOK_MODE,
|
|
ErdCode.LOWER_OVEN_CURRENT_STATE,
|
|
ErdCode.LOWER_OVEN_WARMING_DRAWER_STATE,
|
|
ErdCode.UPPER_OVEN_COOK_MODE,
|
|
ErdCode.UPPER_OVEN_CURRENT_STATE,
|
|
ErdCode.UPPER_OVEN_WARMING_DRAWER_STATE,
|
|
ErdCode.WARMING_DRAWER_STATE,
|
|
}:
|
|
return "mdi:stove"
|
|
if erd_code in {
|
|
ErdCode.TURBO_COOL_STATUS,
|
|
ErdCode.TURBO_FREEZE_STATUS,
|
|
}:
|
|
return "mdi:snowflake"
|
|
if erd_code == ErdCode.SABBATH_MODE:
|
|
return "mdi:judaism"
|
|
|
|
# Let binary sensors assign their own. Might be worth passing
|
|
# the actual entity in if we want to do more of this.
|
|
if erd_code in DOOR_ERD_CODES and isinstance(value, str):
|
|
if "open" in value.lower():
|
|
return "mdi:door-open"
|
|
return "mdi:door-closed"
|
|
|
|
return None
|
|
|
|
|
|
|