mirror of https://github.com/simbaja/ha_gehome.git
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
|
from custom_components.ge_kitchen.entities.common.ge_erd_binary_sensor import GeErdBinarySensor
|
||
|
|
import logging
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from homeassistant.helpers.entity import Entity
|
||
|
|
from gekitchensdk.erd import ErdCode, ErdApplianceType
|
||
|
|
|
||
|
|
from .base import ApplianceApi
|
||
|
|
from ..entities import GeErdSensor, GeErdBinarySensor
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class WasherApi(ApplianceApi):
|
||
|
|
"""API class for washer objects"""
|
||
|
|
APPLIANCE_TYPE = ErdApplianceType.WASHER
|
||
|
|
|
||
|
|
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)
|
||
|
|
]
|
||
|
|
entities = base_entities + washer_entities
|
||
|
|
return entities
|
||
|
|
|