__init__.py

"""The Recteq integration."""

import logging
import async_timeout

from .const import (
    DOMAIN,
    PROJECT,
    VERSION,
    ISSUE_LINK,
    PLATFORMS,
    CONF_DEVICE_ID,
    CONF_IP_ADDRESS,
    CONF_LOCAL_KEY,
    CONF_PROTOCOL
)

from .device import TuyaOutletDeviceWrapper, RecteqDevice

from homeassistant import config_entries
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady

from integrationhelper.const import CC_STARTUP_VERSION

_LOGGER = logging.getLogger(__name__)

async def async_setup(hass: HomeAssistant, config):
    hass.data[DOMAIN] = {}

    _LOGGER.info(CC_STARTUP_VERSION.format(
        name=PROJECT,
        version=VERSION,
        issue_link=ISSUE_LINK
    ))

    return True

async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
    try:
        async with async_timeout.timeout(10):
            device = TuyaOutletDeviceWrapper(
                entry.data[CONF_DEVICE_ID],
                entry.data[CONF_IP_ADDRESS],
                entry.data[CONF_LOCAL_KEY],
                entry.data[CONF_PROTOCOL]
            )
    except ConnectionError as err:
        raise ConfigEntryNotReady from err
        
    recteq = hass.data[DOMAIN][entry.entry_id] = RecteqDevice(
        hass, entry, device
    )

    for PLATFORM in PLATFORMS:
        hass.async_create_task(
            hass.config_entries.async_forward_entry_setup(entry, PLATFORM)
        )
    return True

async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
    unload_ok = all(
        await asyncio.gather(
            *[
                hass.config_entries.async_forward_entry_unload(entry, PLATFORM)
                for PLATFORM in PLATFORMS
            ]
        )
    )
    if unload_ok:
        await hass.data[DOMAIN].pop(entry.entry_id).shutdown()

    return unload_ok