b4bac0f7a0
* Remove unnecessary exception re-wraps * Preserve exception chains on re-raise We slap "from cause" to almost all possible cases here. In some cases it could conceivably be better to do "from None" if we really want to hide the cause. However those should be in the minority, and "from cause" should be an improvement over the corresponding raise without a "from" in all cases anyway. The only case where we raise from None here is in plex, where the exception for an original invalid SSL cert is not the root cause for failure to validate a newly fetched one. Follow local convention on exception variable names if there is a consistent one, otherwise `err` to match with majority of codebase. * Fix mistaken re-wrap in homematicip_cloud/hap.py Missed the difference between HmipConnectionError and HmipcConnectionError. * Do not hide original error on plex new cert validation error Original is not the cause for the new one, but showing old in the traceback is useful nevertheless.
148 lines
4.1 KiB
Python
148 lines
4.1 KiB
Python
"""Config flow for Monoprice 6-Zone Amplifier integration."""
|
|
import logging
|
|
|
|
from pymonoprice import get_async_monoprice
|
|
from serial import SerialException
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries, core, exceptions
|
|
from homeassistant.const import CONF_PORT
|
|
|
|
from .const import (
|
|
CONF_SOURCE_1,
|
|
CONF_SOURCE_2,
|
|
CONF_SOURCE_3,
|
|
CONF_SOURCE_4,
|
|
CONF_SOURCE_5,
|
|
CONF_SOURCE_6,
|
|
CONF_SOURCES,
|
|
)
|
|
from .const import DOMAIN # pylint:disable=unused-import
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SOURCES = [
|
|
CONF_SOURCE_1,
|
|
CONF_SOURCE_2,
|
|
CONF_SOURCE_3,
|
|
CONF_SOURCE_4,
|
|
CONF_SOURCE_5,
|
|
CONF_SOURCE_6,
|
|
]
|
|
|
|
OPTIONS_FOR_DATA = {vol.Optional(source): str for source in SOURCES}
|
|
|
|
DATA_SCHEMA = vol.Schema({vol.Required(CONF_PORT): str, **OPTIONS_FOR_DATA})
|
|
|
|
|
|
@core.callback
|
|
def _sources_from_config(data):
|
|
sources_config = {
|
|
str(idx + 1): data.get(source) for idx, source in enumerate(SOURCES)
|
|
}
|
|
|
|
return {
|
|
index: name.strip()
|
|
for index, name in sources_config.items()
|
|
if (name is not None and name.strip() != "")
|
|
}
|
|
|
|
|
|
async def validate_input(hass: core.HomeAssistant, data):
|
|
"""Validate the user input allows us to connect.
|
|
|
|
Data has the keys from DATA_SCHEMA with values provided by the user.
|
|
"""
|
|
try:
|
|
await get_async_monoprice(data[CONF_PORT], hass.loop)
|
|
except SerialException as err:
|
|
_LOGGER.error("Error connecting to Monoprice controller")
|
|
raise CannotConnect from err
|
|
|
|
sources = _sources_from_config(data)
|
|
|
|
# Return info that you want to store in the config entry.
|
|
return {CONF_PORT: data[CONF_PORT], CONF_SOURCES: sources}
|
|
|
|
|
|
class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|
"""Handle a config flow for Monoprice 6-Zone Amplifier."""
|
|
|
|
VERSION = 1
|
|
CONNECTION_CLASS = config_entries.CONN_CLASS_LOCAL_POLL
|
|
|
|
async def async_step_user(self, user_input=None):
|
|
"""Handle the initial step."""
|
|
errors = {}
|
|
if user_input is not None:
|
|
try:
|
|
info = await validate_input(self.hass, user_input)
|
|
|
|
return self.async_create_entry(title=user_input[CONF_PORT], data=info)
|
|
except CannotConnect:
|
|
errors["base"] = "cannot_connect"
|
|
except Exception: # pylint: disable=broad-except
|
|
_LOGGER.exception("Unexpected exception")
|
|
errors["base"] = "unknown"
|
|
|
|
return self.async_show_form(
|
|
step_id="user", data_schema=DATA_SCHEMA, errors=errors
|
|
)
|
|
|
|
@staticmethod
|
|
@core.callback
|
|
def async_get_options_flow(config_entry):
|
|
"""Define the config flow to handle options."""
|
|
return MonopriceOptionsFlowHandler(config_entry)
|
|
|
|
|
|
@core.callback
|
|
def _key_for_source(index, source, previous_sources):
|
|
if str(index) in previous_sources:
|
|
key = vol.Optional(
|
|
source, description={"suggested_value": previous_sources[str(index)]}
|
|
)
|
|
else:
|
|
key = vol.Optional(source)
|
|
|
|
return key
|
|
|
|
|
|
class MonopriceOptionsFlowHandler(config_entries.OptionsFlow):
|
|
"""Handle a Monoprice options flow."""
|
|
|
|
def __init__(self, config_entry):
|
|
"""Initialize."""
|
|
self.config_entry = config_entry
|
|
|
|
@core.callback
|
|
def _previous_sources(self):
|
|
if CONF_SOURCES in self.config_entry.options:
|
|
previous = self.config_entry.options[CONF_SOURCES]
|
|
else:
|
|
previous = self.config_entry.data[CONF_SOURCES]
|
|
|
|
return previous
|
|
|
|
async def async_step_init(self, user_input=None):
|
|
"""Manage the options."""
|
|
if user_input is not None:
|
|
return self.async_create_entry(
|
|
title="", data={CONF_SOURCES: _sources_from_config(user_input)}
|
|
)
|
|
|
|
previous_sources = self._previous_sources()
|
|
|
|
options = {
|
|
_key_for_source(idx + 1, source, previous_sources): str
|
|
for idx, source in enumerate(SOURCES)
|
|
}
|
|
|
|
return self.async_show_form(
|
|
step_id="init",
|
|
data_schema=vol.Schema(options),
|
|
)
|
|
|
|
|
|
class CannotConnect(exceptions.HomeAssistantError):
|
|
"""Error to indicate we cannot connect."""
|