"""The Nanoleaf integration.""" from aionanoleaf import InvalidToken, Nanoleaf, Unavailable from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_HOST, CONF_TOKEN from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady from homeassistant.helpers.aiohttp_client import async_get_clientsession from .const import DOMAIN async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Set up Nanoleaf from a config entry.""" nanoleaf = Nanoleaf( async_get_clientsession(hass), entry.data[CONF_HOST], entry.data[CONF_TOKEN] ) try: await nanoleaf.get_info() except Unavailable as err: raise ConfigEntryNotReady from err except InvalidToken as err: raise ConfigEntryAuthFailed from err hass.data.setdefault(DOMAIN, {})[entry.entry_id] = nanoleaf hass.async_create_task( hass.config_entries.async_forward_entry_setup(entry, "light") ) return True