ha_gehome/ge_kitchen/devices/fridge.py

32 lines
1.1 KiB
Python
Raw Normal View History

import logging
from typing import List
from homeassistant.helpers.entity import Entity
from gekitchen.erd import ErdCode, ErdApplianceType
from .base import ApplianceApi
from ..entities import GeErdSensor, GeErdSwitch, GeFridgeEntity, GeFreezerEntity
_LOGGER = logging.getLogger(__name__)
class FridgeApi(ApplianceApi):
"""API class for oven objects"""
APPLIANCE_TYPE = ErdApplianceType.FRIDGE
def get_all_entities(self) -> List[Entity]:
base_entities = super().get_all_entities()
fridge_entities = [
GeErdSensor(self, ErdCode.AIR_FILTER_STATUS),
GeErdSensor(self, ErdCode.DOOR_STATUS),
GeErdSensor(self, ErdCode.FRIDGE_MODEL_INFO),
GeErdSensor(self, ErdCode.HOT_WATER_IN_USE),
GeErdSensor(self, ErdCode.HOT_WATER_SET_TEMP),
GeErdSensor(self, ErdCode.HOT_WATER_STATUS),
GeErdSwitch(self, ErdCode.SABBATH_MODE),
GeFreezerEntity(self),
GeFridgeEntity(self),
]
entities = base_entities + fridge_entities
return entities