d86837cc4d
Was improperly generating a warning each restart. Was not properly handling a password change in configuration. Was not properly removing config entries for accounts removed from configuration.
97 lines
3.4 KiB
Python
97 lines
3.4 KiB
Python
"""Config flow to configure Life360 integration."""
|
|
from collections import OrderedDict
|
|
import logging
|
|
|
|
from life360 import LoginError
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
|
|
from .const import CONF_AUTHORIZATION, DOMAIN
|
|
from .helpers import get_api
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOCS_URL = 'https://www.home-assistant.io/components/life360'
|
|
|
|
|
|
@config_entries.HANDLERS.register(DOMAIN)
|
|
class Life360ConfigFlow(config_entries.ConfigFlow):
|
|
"""Life360 integration config flow."""
|
|
|
|
VERSION = 1
|
|
CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL
|
|
|
|
def __init__(self):
|
|
"""Initialize."""
|
|
self._api = get_api()
|
|
self._username = vol.UNDEFINED
|
|
self._password = vol.UNDEFINED
|
|
|
|
@property
|
|
def configured_usernames(self):
|
|
"""Return tuple of configured usernames."""
|
|
entries = self.hass.config_entries.async_entries(DOMAIN)
|
|
if entries:
|
|
return (entry.data[CONF_USERNAME] for entry in entries)
|
|
return ()
|
|
|
|
async def async_step_user(self, user_input=None):
|
|
"""Handle a user initiated config flow."""
|
|
errors = {}
|
|
|
|
if user_input is not None:
|
|
self._username = user_input[CONF_USERNAME]
|
|
self._password = user_input[CONF_PASSWORD]
|
|
try:
|
|
# pylint: disable=no-value-for-parameter
|
|
vol.Email()(self._username)
|
|
authorization = self._api.get_authorization(
|
|
self._username, self._password)
|
|
except vol.Invalid:
|
|
errors[CONF_USERNAME] = 'invalid_username'
|
|
except LoginError:
|
|
errors['base'] = 'invalid_credentials'
|
|
else:
|
|
if self._username in self.configured_usernames:
|
|
errors['base'] = 'user_already_configured'
|
|
else:
|
|
return self.async_create_entry(
|
|
title=self._username,
|
|
data={
|
|
CONF_USERNAME: self._username,
|
|
CONF_PASSWORD: self._password,
|
|
CONF_AUTHORIZATION: authorization
|
|
},
|
|
description_placeholders={'docs_url': DOCS_URL}
|
|
)
|
|
|
|
data_schema = OrderedDict()
|
|
data_schema[vol.Required(CONF_USERNAME, default=self._username)] = str
|
|
data_schema[vol.Required(CONF_PASSWORD, default=self._password)] = str
|
|
|
|
return self.async_show_form(
|
|
step_id='user',
|
|
data_schema=vol.Schema(data_schema),
|
|
errors=errors,
|
|
description_placeholders={'docs_url': DOCS_URL}
|
|
)
|
|
|
|
async def async_step_import(self, user_input):
|
|
"""Import a config flow from configuration."""
|
|
username = user_input[CONF_USERNAME]
|
|
password = user_input[CONF_PASSWORD]
|
|
try:
|
|
authorization = self._api.get_authorization(username, password)
|
|
except LoginError:
|
|
_LOGGER.error('Invalid credentials for %s', username)
|
|
return self.async_abort(reason='invalid_credentials')
|
|
return self.async_create_entry(
|
|
title='{} (from configuration)'.format(username),
|
|
data={
|
|
CONF_USERNAME: username,
|
|
CONF_PASSWORD: password,
|
|
CONF_AUTHORIZATION: authorization
|
|
}
|
|
)
|