__init__.py

Go to the documentation of this file.

"""The Open Thread Border Router integration."""
 
from __future__ import annotations
 
import logging
 
import aiohttp
import python_otbr_api
 
from homeassistant.components.thread import async_add_dataset
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
from homeassistant.helpers import config_validation as cv, issue_registry as ir
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType
 
from . import websocket_api
from .const import DOMAIN
from .util import (
    GetBorderAgentIdNotSupported,
    OTBRData,
    update_issues,
    update_unique_id,
)
 
_LOGGER = logging.getLogger(__name__)
 
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
 
type OTBRConfigEntry = ConfigEntry[OTBRData]
 
 
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
    """Set up the Open Thread Border Router component."""
    websocket_api.async_setup(hass)
    return True
 
 
async def async_setup_entry(hass: HomeAssistant, entry: OTBRConfigEntry) -> bool:
    """Set up an Open Thread Border Router config entry."""
    api = python_otbr_api.OTBR(entry.data["url"], async_get_clientsession(hass), 10)
 
    otbrdata = OTBRData(entry.data["url"], api, entry.entry_id)
    try:
        border_agent_id = await otbrdata.get_border_agent_id()
        dataset_tlvs = await otbrdata.get_active_dataset_tlvs()
        extended_address = await otbrdata.get_extended_address()
    except GetBorderAgentIdNotSupported:
        ir.async_create_issue(
            hass,
            DOMAIN,
            f"get_get_border_agent_id_unsupported_{otbrdata.entry_id}",
            is_fixable=False,
            is_persistent=False,
            severity=ir.IssueSeverity.WARNING,
            translation_key="get_get_border_agent_id_unsupported",
        )
        return False
    except (
        HomeAssistantError,
        aiohttp.ClientError,
        TimeoutError,
    ) as err:
        raise ConfigEntryNotReady("Unable to connect") from err
    await update_unique_id(hass, entry, border_agent_id)
    if dataset_tlvs:
        await update_issues(hass, otbrdata, dataset_tlvs)
        await async_add_dataset(
            hass,
            DOMAIN,
            dataset_tlvs.hex(),
            preferred_border_agent_id=border_agent_id.hex(),
            preferred_extended_address=extended_address.hex(),
        )
 
    entry.async_on_unload(entry.add_update_listener(async_reload_entry))
    entry.runtime_data = otbrdata
 
    return True
 
 
async def async_unload_entry(hass: HomeAssistant, entry: OTBRConfigEntry) -> bool:
    """Unload a config entry."""
    return True
 
 
async def async_reload_entry(hass: HomeAssistant, entry: OTBRConfigEntry) -> None:
    """Handle an options update."""
    await hass.config_entries.async_reload(entry.entry_id)