mirror of
https://github.com/home-assistant/core
synced 2024-10-13 14:52:10 +02:00
b78d156f0e
* Add MELCloud integration * Provides a climate and sensor platforms. Multiple platforms on one go is not the best option, but it does not make sense to remove them and commit them later either. * Email and access token are stored to the ConfigEntry. The token can be updated by adding the integration again with the same email address. The config flow is aborted and the update is performed on the background. * Run isort * Fix pylint errors * Run black * Increase coverage * Update pymelcloud dependency * Add HVAC_MODE_OFF emulation * Remove print * Update pymelcloud to enable device type filtering * Collapse except blocks and chain ClientNotReadys * Add preliminary documentation URL * Use list comp for creating model info Filters out empty model names form units. * f-string galore Dropped 'HVAC' from AtaDevice name template. * Delegate fan mode mapping to pymelcloud * Fix type annotation * Access AtaDevice through self._device * Prefer list comprehension * Update pymelcloud to leverage device type grouping The updated backend lib returns devices in a dict grouped by the device type. The devices do not necessarily need to be in a single list and this way isinstance is not required to extract devices by type. * Remove DOMAIN presence check This does not seem to make much sense after all. * Fix async_setup_entry Entry setup used half-baked naming from few experimentations back. The naming conventiens were unified to match the platforms. A redundant noneness check was also removed after evaluating the possible return values from the backend lib. * Simplify empty model name check * Improve config validation * Use config_validation strings. * Add CONF_EMAIL to config schema. The value is not strictly required when configuring through configuration.yaml, but having it there makes things more consistent. * Use dict[key] to access required properties. * Add DOMAIN in config check back to async_setup. This is required if an integration is configured throught config_flow. * Remove unused manifest properties * Remove redundant ClimateDevice property override * Add __init__.py to coverage exclusion * Use CONF_USERNAME instead of CONF_EMAIL * Use asyncio.gather instead of asyncio.wait * Misc fixes * any -> Any * Better names for dict iterations * Proper dict access with mandatory/known keys * Remove unused 'name' argument * Remove unnecessary platform info from unique_ids * Remove redundant methods from climate platform * Remove redundant default value from dict get * Update ConfigFlow sub-classing * Define sensors in a dict instead of a list * Use _abort_if_unique_id_configured to update token * Fix them tests * Remove current state guards * Fix that gather call * Implement sensor definitions without str manipulation * Use relative intra-package imports * Update homeassistant/components/melcloud/config_flow.py Co-Authored-By: Martin Hjelmare <marhje52@gmail.com> Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
"""Config flow for the MELCloud platform."""
|
|
import asyncio
|
|
import logging
|
|
from typing import Optional
|
|
|
|
from aiohttp import ClientError, ClientResponseError
|
|
from async_timeout import timeout
|
|
import pymelcloud
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.const import CONF_PASSWORD, CONF_TOKEN, CONF_USERNAME
|
|
|
|
from .const import DOMAIN # pylint: disable=unused-import
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class FlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
|
|
"""Handle a config flow."""
|
|
|
|
VERSION = 1
|
|
CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL
|
|
|
|
async def _create_entry(self, username: str, token: str):
|
|
"""Register new entry."""
|
|
await self.async_set_unique_id(username)
|
|
self._abort_if_unique_id_configured({CONF_TOKEN: token})
|
|
return self.async_create_entry(
|
|
title=username, data={CONF_USERNAME: username, CONF_TOKEN: token},
|
|
)
|
|
|
|
async def _create_client(
|
|
self,
|
|
username: str,
|
|
*,
|
|
password: Optional[str] = None,
|
|
token: Optional[str] = None,
|
|
):
|
|
"""Create client."""
|
|
if password is None and token is None:
|
|
raise ValueError(
|
|
"Invalid internal state. Called without either password or token",
|
|
)
|
|
|
|
try:
|
|
with timeout(10):
|
|
acquired_token = token
|
|
if acquired_token is None:
|
|
acquired_token = await pymelcloud.login(
|
|
username,
|
|
password,
|
|
self.hass.helpers.aiohttp_client.async_get_clientsession(),
|
|
)
|
|
await pymelcloud.get_devices(
|
|
acquired_token,
|
|
self.hass.helpers.aiohttp_client.async_get_clientsession(),
|
|
)
|
|
except ClientResponseError as err:
|
|
if err.status == 401 or err.status == 403:
|
|
return self.async_abort(reason="invalid_auth")
|
|
return self.async_abort(reason="cannot_connect")
|
|
except (asyncio.TimeoutError, ClientError):
|
|
return self.async_abort(reason="cannot_connect")
|
|
|
|
return await self._create_entry(username, acquired_token)
|
|
|
|
async def async_step_user(self, user_input=None):
|
|
"""User initiated config flow."""
|
|
if user_input is None:
|
|
return self.async_show_form(
|
|
step_id="user",
|
|
data_schema=vol.Schema(
|
|
{vol.Required(CONF_USERNAME): str, vol.Required(CONF_PASSWORD): str}
|
|
),
|
|
)
|
|
username = user_input[CONF_USERNAME]
|
|
return await self._create_client(username, password=user_input[CONF_PASSWORD])
|
|
|
|
async def async_step_import(self, user_input):
|
|
"""Import a config entry."""
|
|
return await self._create_client(
|
|
user_input[CONF_USERNAME], token=user_input[CONF_TOKEN]
|
|
)
|