mirror of https://github.com/simbaja/ha_gehome.git
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
import logging
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from homeassistant.helpers.entity import Entity
|
||
|
|
from gehomesdk import (
|
||
|
|
ErdCode,
|
||
|
|
ErdApplianceType,
|
||
|
|
ErdOnOff
|
||
|
|
)
|
||
|
|
|
||
|
|
from .base import ApplianceApi
|
||
|
|
from ..entities import (
|
||
|
|
GeErdSensor,
|
||
|
|
GeErdSelect,
|
||
|
|
GeErdPropertySensor,
|
||
|
|
GeErdSwitch,
|
||
|
|
ErdOnOffBoolConverter,
|
||
|
|
GeDehumidifierFanSpeedSensor,
|
||
|
|
GeDehumidifier
|
||
|
|
)
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class DehumidifierApi(ApplianceApi):
|
||
|
|
"""API class for Dehumidifier objects"""
|
||
|
|
APPLIANCE_TYPE = ErdApplianceType.DEHUMIDIFIER
|
||
|
|
|
||
|
|
def get_all_entities(self) -> List[Entity]:
|
||
|
|
base_entities = super().get_all_entities()
|
||
|
|
|
||
|
|
dhum_entities = [
|
||
|
|
GeErdSwitch(self, ErdCode.AC_POWER_STATUS, bool_converter=ErdOnOffBoolConverter(), icon_on_override="mdi:power-on", icon_off_override="mdi:power-off"),
|
||
|
|
GeDehumidifierFanSpeedSensor(self, ErdCode.AC_FAN_SETTING, icon_override="mdi:fan"),
|
||
|
|
GeErdSensor(self, ErdCode.DHUM_CURRENT_HUMIDITY),
|
||
|
|
GeErdSensor(self, ErdCode.DHUM_TARGET_HUMIDITY),
|
||
|
|
GeErdPropertySensor(self, ErdCode.DHUM_MAINTENANCE, "empty_bucket", device_class_override="problem"),
|
||
|
|
GeErdPropertySensor(self, ErdCode.DHUM_MAINTENANCE, "clean_filter", device_class_override="problem"),
|
||
|
|
GeDehumidifier(self)
|
||
|
|
]
|
||
|
|
|
||
|
|
entities = base_entities + dhum_entities
|
||
|
|
return entities
|
||
|
|
|