mirror of https://github.com/simbaja/ha_gehome.git
25 lines
618 B
Python
25 lines
618 B
Python
|
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, GeDishwasherControlLockedSwitch
|
||
|
|
||
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
||
|
|
||
|
class DryerApi(ApplianceApi):
|
||
|
"""API class for dryer objects"""
|
||
|
APPLIANCE_TYPE = ErdApplianceType.WASHER
|
||
|
|
||
|
def get_all_entities(self) -> List[Entity]:
|
||
|
base_entities = super().get_all_entities()
|
||
|
|
||
|
dryer_entities = [
|
||
|
|
||
|
]
|
||
|
entities = base_entities + dryer_entities
|
||
|
return entities
|
||
|
|