"""Config flow for Oncue integration.""" from __future__ import annotations import logging from typing import Any from aiooncue import LoginFailedException, Oncue import voluptuous as vol from homeassistant import config_entries from homeassistant.const import CONF_PASSWORD, CONF_USERNAME from homeassistant.data_entry_flow import FlowResult from homeassistant.helpers.aiohttp_client import async_get_clientsession from .const import CONNECTION_EXCEPTIONS, DOMAIN _LOGGER = logging.getLogger(__name__) class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): """Handle a config flow for Oncue.""" VERSION = 1 async def async_step_user( self, user_input: dict[str, Any] | None = None ) -> FlowResult: """Handle the initial step.""" errors = {} if user_input is not None: try: await Oncue( user_input[CONF_USERNAME], user_input[CONF_PASSWORD], async_get_clientsession(self.hass), ).async_login() except CONNECTION_EXCEPTIONS: errors["base"] = "cannot_connect" except LoginFailedException: errors["base"] = "invalid_auth" except Exception: # pylint: disable=broad-except _LOGGER.exception("Unexpected exception") errors["base"] = "unknown" else: normalized_username = user_input[CONF_USERNAME].lower() await self.async_set_unique_id(normalized_username) self._abort_if_unique_id_configured() return self.async_create_entry( title=normalized_username, data=user_input ) return self.async_show_form( step_id="user", data_schema=vol.Schema( { vol.Required(CONF_USERNAME): str, vol.Required(CONF_PASSWORD): str, } ), errors=errors, )