mirror of https://github.com/simbaja/ha_gehome.git
24 lines
694 B
Python
24 lines
694 B
Python
import logging
|
|
from typing import Type
|
|
|
|
from gekitchensdk.erd import ErdApplianceType
|
|
|
|
from .base import ApplianceApi
|
|
from .oven import OvenApi
|
|
from .fridge import FridgeApi
|
|
from .dishwasher import DishwasherApi
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
def get_appliance_api_type(appliance_type: ErdApplianceType) -> Type:
|
|
_LOGGER.debug(f"Found device type: {appliance_type}")
|
|
"""Get the appropriate appliance type"""
|
|
if appliance_type == ErdApplianceType.OVEN:
|
|
return OvenApi
|
|
if appliance_type == ErdApplianceType.FRIDGE:
|
|
return FridgeApi
|
|
if appliance_type == ErdApplianceType.DISH_WASHER:
|
|
return DishwasherApi
|
|
# Fallback
|
|
return ApplianceApi
|