config_flow.py

Go to the documentation of this file.

"""Config flow for Sanix integration."""
 
import logging
from typing import Any
 
from sanix import Sanix
from sanix.exceptions import SanixException, SanixInvalidAuthException
import voluptuous as vol
 
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_TOKEN
 
from .const import CONF_SERIAL_NUMBER, DOMAIN, MANUFACTURER
 
_LOGGER = logging.getLogger(__name__)
 
 
STEP_USER_DATA_SCHEMA = vol.Schema(
    {
        vol.Required(CONF_SERIAL_NUMBER): str,
        vol.Required(CONF_TOKEN): str,
    }
)
 
 
class SanixConfigFlow(ConfigFlow, domain=DOMAIN):
    """Handle a config flow for Sanix."""
 
    VERSION = 1
 
    async def async_step_user(
        self, user_input: dict[str, Any] | None = None
    ) -> ConfigFlowResult:
        """Handle the initial step."""
        errors: dict[str, str] = {}
 
        if user_input:
            await self.async_set_unique_idasync_set_unique_id(user_input[CONF_SERIAL_NUMBER])
            self._abort_if_unique_id_configured_abort_if_unique_id_configured()
 
            sanix_api = Sanix(user_input[CONF_SERIAL_NUMBER], user_input[CONF_TOKEN])
 
            try:
                await self.hass.async_add_executor_job(sanix_api.fetch_data)
            except SanixInvalidAuthException:
                errors["base"] = "invalid_auth"
            except SanixException:
                errors["base"] = "unknown"
            else:
                return self.async_create_entryasync_create_entryasync_create_entry(
                    title=MANUFACTURER,
                    data=user_input,
                )
 
        return self.async_show_formasync_show_formasync_show_form(
            step_id="user",
            description_placeholders={"dashboard_url": "https://sanix.bitcomplex.pl/"},
            data_schema=STEP_USER_DATA_SCHEMA,
            errors=errors,
        )