__init__.py

Go to the documentation of this file.

"""The IronOS integration."""
 
from __future__ import annotations
 
import logging
from typing import TYPE_CHECKING
 
from aiogithubapi import GitHubAPI
from pynecil import Pynecil
 
from homeassistant.components import bluetooth
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType
from homeassistant.util.hass_dict import HassKey
 
from .const import DOMAIN
from .coordinator import IronOSFirmwareUpdateCoordinator, IronOSLiveDataCoordinator
 
PLATFORMS: list[Platform] = [Platform.NUMBER, Platform.SENSOR, Platform.UPDATE]
 
 
type IronOSConfigEntry = ConfigEntry[IronOSLiveDataCoordinator]
IRON_OS_KEY: HassKey[IronOSFirmwareUpdateCoordinator] = HassKey(DOMAIN)
 
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
 
_LOGGER = logging.getLogger(__name__)
 
 
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
    """Set up IronOS firmware update coordinator."""
 
    session = async_get_clientsession(hass)
    github = GitHubAPI(session=session)
 
    hass.data[IRON_OS_KEY] = IronOSFirmwareUpdateCoordinator(hass, github)
    await hass.data[IRON_OS_KEY].async_request_refresh()
    return True
 
 
async def async_setup_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bool:
    """Set up IronOS from a config entry."""
    if TYPE_CHECKING:
        assert entry.unique_id
    ble_device = bluetooth.async_ble_device_from_address(
        hass, entry.unique_id, connectable=True
    )
    if not ble_device:
        raise ConfigEntryNotReady(
            translation_domain=DOMAIN,
            translation_key="setup_device_unavailable_exception",
            translation_placeholders={CONF_NAME: entry.title},
        )
 
    device = Pynecil(ble_device)
 
    coordinator = IronOSLiveDataCoordinator(hass, device)
    await coordinator.async_config_entry_first_refresh()
 
    entry.runtime_data = coordinator
    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
    return True
 
 
async def async_unload_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bool:
    """Unload a config entry."""
    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)