__init__.py

Go to the documentation of this file.

"""The P1 Monitor integration."""
 
from __future__ import annotations
 
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
 
from .const import LOGGER
from .coordinator import P1MonitorDataUpdateCoordinator
 
PLATFORMS: list[Platform] = [Platform.SENSOR]
 
type P1MonitorConfigEntry = ConfigEntry[P1MonitorDataUpdateCoordinator]
 
 
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
    """Set up P1 Monitor from a config entry."""
 
    coordinator = P1MonitorDataUpdateCoordinator(hass)
    try:
        await coordinator.async_config_entry_first_refresh()
    except ConfigEntryNotReady:
        await coordinator.p1monitor.close()
        raise
 
    entry.runtime_data = coordinator
 
    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
    return True
 
 
async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
    """Migrate old entry."""
    LOGGER.debug("Migrating from version %s", config_entry.version)
 
    if config_entry.version == 1:
        # Migrate to split host and port
        host = config_entry.data[CONF_HOST]
        if ":" in host:
            host, port = host.split(":")
        else:
            port = 80
 
        new_data = {
            **config_entry.data,
            CONF_HOST: host,
            CONF_PORT: int(port),
        }
 
        hass.config_entries.async_update_entry(config_entry, data=new_data, version=2)
        LOGGER.debug("Migration to version %s successful", config_entry.version)
    return True
 
 
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
    """Unload P1 Monitor config entry."""
    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)