mirror of https://github.com/simbaja/ha_gehome.git
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
"""GE Home Button Entities"""
|
||
|
|
import async_timeout
|
||
|
|
import logging
|
||
|
|
from typing import Callable
|
||
|
|
|
||
|
|
from homeassistant.config_entries import ConfigEntry
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
|
||
|
|
from .const import DOMAIN
|
||
|
|
from .entities import GeErdButton
|
||
|
|
from .update_coordinator import GeHomeUpdateCoordinator
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: Callable):
|
||
|
|
"""GE Home buttons."""
|
||
|
|
|
||
|
|
coordinator: GeHomeUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
|
||
|
|
|
||
|
|
# 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 = [
|
||
|
|
entity
|
||
|
|
for api in apis
|
||
|
|
for entity in api.entities
|
||
|
|
if isinstance(entity, GeErdButton)
|
||
|
|
]
|
||
|
|
_LOGGER.debug(f'Found {len(entities):d} buttons ')
|
||
|
|
async_add_entities(entities)
|