ha_gehome/ge_kitchen/binary_sensor.py

34 lines
1.1 KiB
Python
Raw Normal View History

2020-08-13 15:10:55 -06:00
"""GE Kitchen Sensor Entities"""
import async_timeout
import logging
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from typing import Callable
from .update_coordinator import GeKitchenUpdateCoordinator
2020-08-25 11:09:28 -06:00
from .appliance_api import GeErdBinarySensor, GeErdSwitch
2020-08-13 15:10:55 -06:00
from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: Callable):
"""GE Kitchen sensors."""
coordinator = hass.data[DOMAIN][config_entry.entry_id] # type: GeKitchenUpdateCoordinator
# This should be a NOP, but let's be safe
with async_timeout.timeout(20):
await coordinator.initialization_future
apis = coordinator.appliance_apis.values()
_LOGGER.debug(f'Found {len(apis):d} appliance APIs')
entities = [
2020-08-13 15:10:55 -06:00
entity
for api in apis
for entity in api.entities
2020-08-25 11:09:28 -06:00
if isinstance(entity, GeErdBinarySensor) and not isinstance(entity, GeErdSwitch)
2020-08-13 15:10:55 -06:00
]
_LOGGER.debug(f'Found {len(entities):d} binary sensors ')
async_add_entities(entities)