__init__.py

Go to the documentation of this file.

"""SMLIGHT SLZB Zigbee device integration."""
 
from __future__ import annotations
 
from dataclasses import dataclass
 
from pysmlight import Api2
 
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
from .coordinator import SmDataUpdateCoordinator, SmFirmwareUpdateCoordinator
 
PLATFORMS: list[Platform] = [
    Platform.BINARY_SENSOR,
    Platform.BUTTON,
    Platform.SENSOR,
    Platform.SWITCH,
    Platform.UPDATE,
]
 
 
@dataclass(kw_only=True)
class SmlightData:
    """Coordinator data class."""
 
    data: SmDataUpdateCoordinator
    firmware: SmFirmwareUpdateCoordinator
 
 
type SmConfigEntry = ConfigEntry[SmlightData]
 
 
async def async_setup_entry(hass: HomeAssistant, entry: SmConfigEntry) -> bool:
    """Set up SMLIGHT Zigbee from a config entry."""
    client = Api2(host=entry.data[CONF_HOST], session=async_get_clientsession(hass))
 
    data_coordinator = SmDataUpdateCoordinator(hass, entry.data[CONF_HOST], client)
    firmware_coordinator = SmFirmwareUpdateCoordinator(
        hass, entry.data[CONF_HOST], client
    )
 
    await data_coordinator.async_config_entry_first_refresh()
    await firmware_coordinator.async_config_entry_first_refresh()
 
    if data_coordinator.data.info.legacy_api < 2:
        entry.async_create_background_task(
            hass, client.sse.client(), "smlight-sse-client"
        )
 
    entry.runtime_data = SmlightData(
        data=data_coordinator, firmware=firmware_coordinator
    )
 
    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
    return True
 
 
async def async_unload_entry(hass: HomeAssistant, entry: SmConfigEntry) -> bool:
    """Unload a config entry."""
    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)