Add config flow to enigma2 (#106348)
* add config flow to enigma2 * Apply suggestions from code review Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com> * fix suggested change * use parametrize for config flow tests * Restore PLATFORM_SCHEMA and add create_issue to async_setup_platform * fix docstring * remove name, refactor config flow * bump dependency * remove name, add verify_ssl, use async_create_clientsession * use translation key, change integration type to device * Bump openwebifpy to 4.2.1 * cleanup, remove CONF_NAME from entity, add async_set_unique_id * clear unneeded constants, fix tests * fix tests * move _attr_translation_key out of init * update test requirement * Address review comments * address review comments * clear strings.json * Review coments --------- Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
This commit is contained in:
parent
764a0f29cc
commit
be0926b7b8
14 changed files with 565 additions and 40 deletions
158
homeassistant/components/enigma2/config_flow.py
Normal file
158
homeassistant/components/enigma2/config_flow.py
Normal file
|
@ -0,0 +1,158 @@
|
|||
"""Config flow for Enigma2."""
|
||||
|
||||
from typing import Any
|
||||
|
||||
from aiohttp.client_exceptions import ClientError
|
||||
from openwebif.api import OpenWebIfDevice
|
||||
from openwebif.error import InvalidAuthError
|
||||
import voluptuous as vol
|
||||
from yarl import URL
|
||||
|
||||
from homeassistant.components.homeassistant import DOMAIN as HOMEASSISTANT_DOMAIN
|
||||
from homeassistant.config_entries import SOURCE_USER, ConfigFlow, ConfigFlowResult
|
||||
from homeassistant.const import (
|
||||
CONF_HOST,
|
||||
CONF_PASSWORD,
|
||||
CONF_PORT,
|
||||
CONF_SSL,
|
||||
CONF_USERNAME,
|
||||
CONF_VERIFY_SSL,
|
||||
)
|
||||
from homeassistant.helpers import selector
|
||||
from homeassistant.helpers.aiohttp_client import async_create_clientsession
|
||||
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
|
||||
|
||||
from .const import (
|
||||
CONF_DEEP_STANDBY,
|
||||
CONF_SOURCE_BOUQUET,
|
||||
CONF_USE_CHANNEL_ICON,
|
||||
DEFAULT_PORT,
|
||||
DEFAULT_SSL,
|
||||
DEFAULT_VERIFY_SSL,
|
||||
DOMAIN,
|
||||
)
|
||||
|
||||
CONFIG_SCHEMA = vol.Schema(
|
||||
{
|
||||
vol.Required(CONF_HOST): selector.TextSelector(),
|
||||
vol.Required(CONF_PORT, default=DEFAULT_PORT): vol.All(
|
||||
selector.NumberSelector(
|
||||
selector.NumberSelectorConfig(
|
||||
min=1, max=65535, mode=selector.NumberSelectorMode.BOX
|
||||
)
|
||||
),
|
||||
vol.Coerce(int),
|
||||
),
|
||||
vol.Optional(CONF_USERNAME): selector.TextSelector(),
|
||||
vol.Optional(CONF_PASSWORD): selector.TextSelector(
|
||||
selector.TextSelectorConfig(type=selector.TextSelectorType.PASSWORD)
|
||||
),
|
||||
vol.Required(CONF_SSL, default=DEFAULT_SSL): selector.BooleanSelector(),
|
||||
vol.Required(
|
||||
CONF_VERIFY_SSL, default=DEFAULT_VERIFY_SSL
|
||||
): selector.BooleanSelector(),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
class Enigma2ConfigFlowHandler(ConfigFlow, domain=DOMAIN):
|
||||
"""Handle a config flow for Enigma2."""
|
||||
|
||||
DATA_KEYS = (
|
||||
CONF_HOST,
|
||||
CONF_PORT,
|
||||
CONF_USERNAME,
|
||||
CONF_PASSWORD,
|
||||
CONF_SSL,
|
||||
CONF_VERIFY_SSL,
|
||||
)
|
||||
OPTIONS_KEYS = (CONF_DEEP_STANDBY, CONF_SOURCE_BOUQUET, CONF_USE_CHANNEL_ICON)
|
||||
|
||||
def __init__(self) -> None:
|
||||
"""Initialize the config flow."""
|
||||
super().__init__()
|
||||
self.errors: dict[str, str] = {}
|
||||
self._data: dict[str, Any] = {}
|
||||
self._options: dict[str, Any] = {}
|
||||
|
||||
async def validate_user_input(self, user_input: dict[str, Any]) -> dict[str, Any]:
|
||||
"""Validate user input."""
|
||||
|
||||
self.errors = {}
|
||||
|
||||
self._async_abort_entries_match({CONF_HOST: user_input[CONF_HOST]})
|
||||
|
||||
base_url = URL.build(
|
||||
scheme="http" if not user_input[CONF_SSL] else "https",
|
||||
host=user_input[CONF_HOST],
|
||||
port=user_input[CONF_PORT],
|
||||
user=user_input.get(CONF_USERNAME),
|
||||
password=user_input.get(CONF_PASSWORD),
|
||||
)
|
||||
|
||||
session = async_create_clientsession(
|
||||
self.hass, verify_ssl=user_input[CONF_VERIFY_SSL], base_url=base_url
|
||||
)
|
||||
|
||||
try:
|
||||
about = await OpenWebIfDevice(session).get_about()
|
||||
except InvalidAuthError:
|
||||
self.errors["base"] = "invalid_auth"
|
||||
except ClientError:
|
||||
self.errors["base"] = "cannot_connect"
|
||||
except Exception: # pylint: disable=broad-except
|
||||
self.errors["base"] = "unknown"
|
||||
else:
|
||||
await self.async_set_unique_id(about["info"]["ifaces"][0]["mac"])
|
||||
self._abort_if_unique_id_configured()
|
||||
|
||||
return user_input
|
||||
|
||||
async def async_step_user(
|
||||
self, user_input: dict[str, Any] | None = None
|
||||
) -> ConfigFlowResult:
|
||||
"""Handle the user step."""
|
||||
if user_input is None:
|
||||
return self.async_show_form(step_id=SOURCE_USER, data_schema=CONFIG_SCHEMA)
|
||||
|
||||
data = await self.validate_user_input(user_input)
|
||||
if "base" in self.errors:
|
||||
return self.async_show_form(
|
||||
step_id=SOURCE_USER, data_schema=CONFIG_SCHEMA, errors=self.errors
|
||||
)
|
||||
return self.async_create_entry(
|
||||
data=data, title=data[CONF_HOST], options=self._options
|
||||
)
|
||||
|
||||
async def async_step_import(self, user_input: dict[str, Any]) -> ConfigFlowResult:
|
||||
"""Validate import."""
|
||||
if CONF_PORT not in user_input:
|
||||
user_input[CONF_PORT] = DEFAULT_PORT
|
||||
if CONF_SSL not in user_input:
|
||||
user_input[CONF_SSL] = DEFAULT_SSL
|
||||
user_input[CONF_VERIFY_SSL] = DEFAULT_VERIFY_SSL
|
||||
|
||||
async_create_issue(
|
||||
self.hass,
|
||||
HOMEASSISTANT_DOMAIN,
|
||||
f"deprecated_yaml_{DOMAIN}",
|
||||
breaks_in_ha_version="2024.11.0",
|
||||
is_fixable=False,
|
||||
is_persistent=False,
|
||||
issue_domain=DOMAIN,
|
||||
severity=IssueSeverity.WARNING,
|
||||
translation_key="deprecated_yaml",
|
||||
translation_placeholders={
|
||||
"domain": DOMAIN,
|
||||
"integration_title": "Enigma2",
|
||||
},
|
||||
)
|
||||
|
||||
self._data = {
|
||||
key: user_input[key] for key in user_input if key in self.DATA_KEYS
|
||||
}
|
||||
self._options = {
|
||||
key: user_input[key] for key in user_input if key in self.OPTIONS_KEYS
|
||||
}
|
||||
|
||||
return await self.async_step_user(self._data)
|
Loading…
Add table
Add a link
Reference in a new issue