2020-07-21 08:29:36 -06:00
|
|
|
"""Shark IQ robot vacuums"""
|
|
|
|
|
import logging
|
|
|
|
|
from typing import List, TYPE_CHECKING
|
|
|
|
|
from .const import DOMAIN, SHARKIQ_SESSION
|
|
|
|
|
from .sharkiq import SharkVacuumEntity
|
|
|
|
|
|
|
|
|
|
if TYPE_CHECKING:
|
2020-07-22 07:40:39 -06:00
|
|
|
from sharkiqpy import AylaApi, SharkIqVacuum
|
2020-07-21 08:29:36 -06:00
|
|
|
|
|
|
|
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
|
|
|
"""Set up the Shark IQ vacuum cleaner."""
|
|
|
|
|
domain_data = hass.data[DOMAIN][config_entry.entry_id]
|
|
|
|
|
ayla_api = domain_data[SHARKIQ_SESSION] # type: AylaApi
|
|
|
|
|
|
|
|
|
|
devices = await ayla_api.async_get_devices() # type: List[SharkIqVacuum]
|
|
|
|
|
device_names = ', '.join([d.name for d in devices])
|
|
|
|
|
_LOGGER.debug("Found % Shark IQ device(s): %s", len(devices), device_names)
|
|
|
|
|
async_add_entities([SharkVacuumEntity(d) for d in devices], True)
|
