config_flow.py

Go to the documentation of this file.

"""Adds config flow for SabNzbd."""
 
from __future__ import annotations
 
import logging
from typing import Any
 
import voluptuous as vol
import yarl
 
from homeassistant.config_entries import (
    SOURCE_RECONFIGURE,
    ConfigFlow,
    ConfigFlowResult,
)
from homeassistant.const import CONF_API_KEY, CONF_URL
from homeassistant.helpers.selector import (
    TextSelector,
    TextSelectorConfig,
    TextSelectorType,
)
from homeassistant.util import slugify
 
from .const import DOMAIN
from .helpers import get_client
 
_LOGGER = logging.getLogger(__name__)
 
USER_SCHEMA = vol.Schema(
    {
        vol.Required(CONF_URL): TextSelector(
            TextSelectorConfig(
                type=TextSelectorType.URL,
            )
        ),
        vol.Required(CONF_API_KEY): TextSelector(
            TextSelectorConfig(
                type=TextSelectorType.PASSWORD,
            )
        ),
    }
)
 
 
class SABnzbdConfigFlow(ConfigFlow, domain=DOMAIN):
    """Sabnzbd config flow."""
 
    VERSION = 1
 
    async def async_step_reconfigure(
        self, user_input: dict[str, Any] | None = None
    ) -> ConfigFlowResult:
        """Handle reconfiguration flow."""
        return await self.async_step_userasync_step_userasync_step_user(user_input)
 
    async def async_step_user(
        self, user_input: dict[str, Any] | None = None
    ) -> ConfigFlowResult:
        """Handle a flow initialized by the user."""
        errors = {}
 
        if user_input is not None:
            sab_api = await get_client(self.hass, user_input)
            if not sab_api:
                errors["base"] = "cannot_connect"
            else:
                self._async_abort_entries_match_async_abort_entries_match(
                    {
                        CONF_URL: user_input[CONF_URL],
                        CONF_API_KEY: user_input[CONF_API_KEY],
                    }
                )
 
                if self.sourcesourcesourcesource == SOURCE_RECONFIGURE:
                    return self.async_update_reload_and_abortasync_update_reload_and_abort(
                        self._get_reconfigure_entry_get_reconfigure_entry(), data_updates=user_input
                    )
 
                parsed_url = yarl.URL(user_input[CONF_URL])
                return self.async_create_entryasync_create_entryasync_create_entry(
                    title=slugify(parsed_url.host), data=user_input
                )
 
        return self.async_show_formasync_show_formasync_show_form(
            step_id="user",
            data_schema=self.add_suggested_values_to_schemaadd_suggested_values_to_schema(
                USER_SCHEMA,
                self._get_reconfigure_entry_get_reconfigure_entry().data
                if self.sourcesourcesourcesource == SOURCE_RECONFIGURE
                else user_input,
            ),
            errors=errors,
        )