ha_gehome/sharkiq/vacuum.py

23 lines
795 B
Python
Raw Normal View History

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)