Add DALI Center integration (home-assistant#151479)

Co-authored-by: Norbert Rittel <[email protected]>
Co-authored-by: Joost Lekkerkerker <[email protected]>

Author: 3 people

CODEOWNERS

@@ -0,0 +1,88 @@
+"""The DALI Center integration."""
+
+from __future__ import annotations
+
+import logging
+
+from PySrDaliGateway import DaliGateway
+from PySrDaliGateway.exceptions import DaliGatewayError
+
+from homeassistant.const import (
+    CONF_HOST,
+    CONF_NAME,
+    CONF_PASSWORD,
+    CONF_PORT,
+    CONF_USERNAME,
+    Platform,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.dispatcher import async_dispatcher_send
+
+from .const import CONF_SERIAL_NUMBER, DOMAIN, MANUFACTURER
+from .types import DaliCenterConfigEntry, DaliCenterData
+
+_PLATFORMS: list[Platform] = [Platform.LIGHT]
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: DaliCenterConfigEntry) -> bool:
+    """Set up DALI Center from a config entry."""
+
+    gateway = DaliGateway(
+        entry.data[CONF_SERIAL_NUMBER],
+        entry.data[CONF_HOST],
+        entry.data[CONF_PORT],
+        entry.data[CONF_USERNAME],
+        entry.data[CONF_PASSWORD],
+        name=entry.data[CONF_NAME],
+    )
+    gw_sn = gateway.gw_sn
+
+    try:
+        await gateway.connect()
+    except DaliGatewayError as exc:
+        raise ConfigEntryNotReady(
+            "You can try to delete the gateway and add it again"
+        ) from exc
+
+    def on_online_status(dev_id: str, available: bool) -> None:
+        signal = f"{DOMAIN}_update_available_{dev_id}"
+        hass.add_job(async_dispatcher_send, hass, signal, available)
+
+    gateway.on_online_status = on_online_status
+
+    try:
+        devices = await gateway.discover_devices()
+    except DaliGatewayError as exc:
+        raise ConfigEntryNotReady(
+            "Unable to discover devices from the gateway"
+        ) from exc
+
+    _LOGGER.debug("Discovered %d devices on gateway %s", len(devices), gw_sn)
+
+    dev_reg = dr.async_get(hass)
+    dev_reg.async_get_or_create(
+        config_entry_id=entry.entry_id,
+        identifiers={(DOMAIN, gw_sn)},
+        manufacturer=MANUFACTURER,
+        name=gateway.name,
+        model="SR-GW-EDA",
+        serial_number=gw_sn,
+    )
+
+    entry.runtime_data = DaliCenterData(
+        gateway=gateway,
+        devices=devices,
+    )
+    await hass.config_entries.async_forward_entry_setups(entry, _PLATFORMS)
+
+    return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: DaliCenterConfigEntry) -> bool:
+    """Unload a config entry."""
+    if unload_ok := await hass.config_entries.async_unload_platforms(entry, _PLATFORMS):
+        await entry.runtime_data.gateway.disconnect()
+    return unload_ok

homeassistant/components/sunricher_dali_center/__init__.py

@@ -0,0 +1,134 @@
+"""Config flow for the DALI Center integration."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from PySrDaliGateway import DaliGateway
+from PySrDaliGateway.discovery import DaliGatewayDiscovery
+from PySrDaliGateway.exceptions import DaliGatewayError
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import (
+    CONF_HOST,
+    CONF_NAME,
+    CONF_PASSWORD,
+    CONF_PORT,
+    CONF_USERNAME,
+)
+from homeassistant.helpers.selector import (
+    SelectOptionDict,
+    SelectSelector,
+    SelectSelectorConfig,
+)
+
+from .const import CONF_SERIAL_NUMBER, DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class DaliCenterConfigFlow(ConfigFlow, domain=DOMAIN):
+    """Handle a config flow for DALI Center."""
+
+    VERSION = 1
+
+    def __init__(self) -> None:
+        """Initialize the config flow."""
+        self._discovered_gateways: dict[str, DaliGateway] = {}
+
+    async def async_step_user(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Handle the initial step."""
+        if user_input is not None:
+            return await self.async_step_select_gateway()
+
+        return self.async_show_form(
+            step_id="user",
+            data_schema=vol.Schema({}),
+        )
+
+    async def async_step_select_gateway(
+        self, discovery_info: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Handle gateway discovery."""
+        errors: dict[str, str] = {}
+
+        if discovery_info and "selected_gateway" in discovery_info:
+            selected_sn = discovery_info["selected_gateway"]
+            selected_gateway = self._discovered_gateways[selected_sn]
+
+            await self.async_set_unique_id(selected_gateway.gw_sn)
+            self._abort_if_unique_id_configured()
+
+            try:
+                await selected_gateway.connect()
+            except DaliGatewayError as err:
+                _LOGGER.debug(
+                    "Failed to connect to gateway %s during config flow",
+                    selected_gateway.gw_sn,
+                    exc_info=err,
+                )
+                errors["base"] = "cannot_connect"
+            else:
+                await selected_gateway.disconnect()
+                return self.async_create_entry(
+                    title=selected_gateway.name,
+                    data={
+                        CONF_SERIAL_NUMBER: selected_gateway.gw_sn,
+                        CONF_HOST: selected_gateway.gw_ip,
+                        CONF_PORT: selected_gateway.port,
+                        CONF_NAME: selected_gateway.name,
+                        CONF_USERNAME: selected_gateway.username,
+                        CONF_PASSWORD: selected_gateway.passwd,
+                    },
+                )
+
+        if not self._discovered_gateways:
+            _LOGGER.debug("Starting gateway discovery")
+            discovery = DaliGatewayDiscovery()
+            try:
+                discovered = await discovery.discover_gateways()
+            except DaliGatewayError as err:
+                _LOGGER.debug("Gateway discovery failed", exc_info=err)
+                errors["base"] = "discovery_failed"
+            else:
+                configured_gateways = {
+                    entry.data[CONF_SERIAL_NUMBER]
+                    for entry in self.hass.config_entries.async_entries(DOMAIN)
+                }
+
+                self._discovered_gateways = {
+                    gw.gw_sn: gw
+                    for gw in discovered
+                    if gw.gw_sn not in configured_gateways
+                }
+
+        if not self._discovered_gateways:
+            return self.async_show_form(
+                step_id="select_gateway",
+                errors=errors if errors else {"base": "no_devices_found"},
+                data_schema=vol.Schema({}),
+            )
+
+        gateway_options = [
+            SelectOptionDict(
+                value=sn,
+                label=f"{gateway.name} [SN {sn}, IP {gateway.gw_ip}]",
+            )
+            for sn, gateway in self._discovered_gateways.items()
+        ]
+
+        return self.async_show_form(
+            step_id="select_gateway",
+            data_schema=vol.Schema(
+                {
+                    vol.Optional("selected_gateway"): SelectSelector(
+                        SelectSelectorConfig(options=gateway_options, sort=True)
+                    ),
+                }
+            ),
+            errors=errors,
+        )

homeassistant/components/sunricher_dali_center/config_flow.py

@@ -0,0 +1,5 @@
+"""Constants for the DALI Center integration."""
+
+DOMAIN = "sunricher_dali_center"
+MANUFACTURER = "Sunricher"
+CONF_SERIAL_NUMBER = "serial_number"