ha_gehome/custom_components/ge_home/devices/washer_dryer.py

42 lines
1.5 KiB
Python

import logging
from typing import List
from homeassistant.helpers.entity import Entity
from gehomesdk.erd import ErdCode, ErdApplianceType
from .base import ApplianceApi
from ..entities import GeErdSensor, GeErdBinarySensor
_LOGGER = logging.getLogger(__name__)
class WasherDryerApi(ApplianceApi):
"""API class for washer/dryer objects"""
APPLIANCE_TYPE = ErdApplianceType.COMBINATION_WASHER_DRYER
def get_all_entities(self) -> List[Entity]:
base_entities = super().get_all_entities()
washer_entities = [
GeErdSensor(self, ErdCode.LAUNDRY_MACHINE_STATE),
GeErdSensor(self, ErdCode.LAUNDRY_MACHINE_SUBCYCLE),
GeErdBinarySensor(self, ErdCode.LAUNDRY_END_OF_CYCLE),
GeErdSensor(self, ErdCode.LAUNDRY_TIME_REMAINING),
GeErdSensor(self, ErdCode.LAUNDRY_CYCLE),
GeErdSensor(self, ErdCode.LAUNDRY_DELAY_TIME_REMAINING),
GeErdSensor(self, ErdCode.LAUNDRY_DOOR),
GeErdBinarySensor(self, ErdCode.LAUNDRY_DOOR_LOCK),
GeErdSensor(self, ErdCode.LAUNDRY_SOIL_LEVEL),
GeErdSensor(self, ErdCode.LAUNDRY_WASHTEMP_LEVEL),
GeErdSensor(self, ErdCode.LAUNDRY_SPINTIME_LEVEL),
GeErdSensor(self, ErdCode.LAUNDRY_RINSE_OPTION),
GeErdBinarySensor(self, ErdCode.LAUNDRY_REMOTE_STATUS)
]
dryer_entities = [
]
entities = base_entities + washer_entities + dryer_entities
return entities