mirror of
https://github.com/home-assistant/core.git
synced 2025-06-05 13:47:06 +00:00
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.
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
"""Config Flow for Flick Electric integration."""
|
|
import asyncio
|
|
import logging
|
|
|
|
import async_timeout
|
|
from pyflick.authentication import AuthException, SimpleFlickAuth
|
|
from pyflick.const import DEFAULT_CLIENT_ID, DEFAULT_CLIENT_SECRET
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries, exceptions
|
|
from homeassistant.const import (
|
|
CONF_CLIENT_ID,
|
|
CONF_CLIENT_SECRET,
|
|
CONF_PASSWORD,
|
|
CONF_USERNAME,
|
|
)
|
|
from homeassistant.helpers import aiohttp_client
|
|
|
|
from .const import DOMAIN # pylint: disable=unused-import
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DATA_SCHEMA = vol.Schema(
|
|
{
|
|
vol.Required(CONF_USERNAME): str,
|
|
vol.Required(CONF_PASSWORD): str,
|
|
vol.Optional(CONF_CLIENT_ID): str,
|
|
vol.Optional(CONF_CLIENT_SECRET): str,
|
|
}
|
|
)
|
|
|
|
|
|
class FlickConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|
"""Flick config flow."""
|
|
|
|
VERSION = 1
|
|
CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL
|
|
|
|
async def _validate_input(self, user_input):
|
|
auth = SimpleFlickAuth(
|
|
username=user_input[CONF_USERNAME],
|
|
password=user_input[CONF_PASSWORD],
|
|
websession=aiohttp_client.async_get_clientsession(self.hass),
|
|
client_id=user_input.get(CONF_CLIENT_ID, DEFAULT_CLIENT_ID),
|
|
client_secret=user_input.get(CONF_CLIENT_SECRET, DEFAULT_CLIENT_SECRET),
|
|
)
|
|
|
|
try:
|
|
with async_timeout.timeout(60):
|
|
token = await auth.async_get_access_token()
|
|
except asyncio.TimeoutError as err:
|
|
raise CannotConnect() from err
|
|
except AuthException as err:
|
|
raise InvalidAuth() from err
|
|
else:
|
|
return token is not None
|
|
|
|
async def async_step_user(self, user_input=None):
|
|
"""Handle gathering login info."""
|
|
errors = {}
|
|
if user_input is not None:
|
|
try:
|
|
await self._validate_input(user_input)
|
|
except CannotConnect:
|
|
errors["base"] = "cannot_connect"
|
|
except InvalidAuth:
|
|
errors["base"] = "invalid_auth"
|
|
except Exception: # pylint: disable=broad-except
|
|
_LOGGER.exception("Unexpected exception")
|
|
errors["base"] = "unknown"
|
|
else:
|
|
await self.async_set_unique_id(
|
|
f"flick_electric_{user_input[CONF_USERNAME]}"
|
|
)
|
|
self._abort_if_unique_id_configured()
|
|
|
|
return self.async_create_entry(
|
|
title=f"Flick Electric: {user_input[CONF_USERNAME]}",
|
|
data=user_input,
|
|
)
|
|
|
|
return self.async_show_form(
|
|
step_id="user", data_schema=DATA_SCHEMA, errors=errors
|
|
)
|
|
|
|
|
|
class CannotConnect(exceptions.HomeAssistantError):
|
|
"""Error to indicate we cannot connect."""
|
|
|
|
|
|
class InvalidAuth(exceptions.HomeAssistantError):
|
|
"""Error to indicate there is invalid auth."""
|