129 lines
3.9 KiB
Python
129 lines
3.9 KiB
Python
"""Config flow for Co2signal integration."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE
|
|
from homeassistant.data_entry_flow import FlowResult
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
from . import APIRatelimitExceeded, InvalidAuth, get_data
|
|
from .const import CONF_COUNTRY_CODE, DOMAIN
|
|
from .util import get_extra_name
|
|
|
|
TYPE_USE_HOME = "Use home location"
|
|
TYPE_SPECIFY_COORDINATES = "Specify coordinates"
|
|
TYPE_SPECIFY_COUNTRY = "Specify country code"
|
|
|
|
|
|
class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|
"""Handle a config flow for Co2signal."""
|
|
|
|
VERSION = 1
|
|
_data: dict | None
|
|
|
|
async def async_step_user(
|
|
self, user_input: dict[str, Any] | None = None
|
|
) -> FlowResult:
|
|
"""Handle the initial step."""
|
|
data_schema = vol.Schema(
|
|
{
|
|
vol.Required("location", default=TYPE_USE_HOME): vol.In(
|
|
(
|
|
TYPE_USE_HOME,
|
|
TYPE_SPECIFY_COORDINATES,
|
|
TYPE_SPECIFY_COUNTRY,
|
|
)
|
|
),
|
|
vol.Required(CONF_API_KEY): cv.string,
|
|
}
|
|
)
|
|
|
|
if user_input is None:
|
|
return self.async_show_form(
|
|
step_id="user",
|
|
data_schema=data_schema,
|
|
)
|
|
|
|
data = {CONF_API_KEY: user_input[CONF_API_KEY]}
|
|
|
|
if user_input["location"] == TYPE_SPECIFY_COORDINATES:
|
|
self._data = data
|
|
return await self.async_step_coordinates()
|
|
|
|
if user_input["location"] == TYPE_SPECIFY_COUNTRY:
|
|
self._data = data
|
|
return await self.async_step_country()
|
|
|
|
return await self._validate_and_create("user", data_schema, data)
|
|
|
|
async def async_step_coordinates(
|
|
self, user_input: dict[str, Any] | None = None
|
|
) -> FlowResult:
|
|
"""Validate coordinates."""
|
|
data_schema = vol.Schema(
|
|
{
|
|
vol.Required(
|
|
CONF_LATITUDE,
|
|
): cv.latitude,
|
|
vol.Required(
|
|
CONF_LONGITUDE,
|
|
): cv.longitude,
|
|
}
|
|
)
|
|
if user_input is None:
|
|
return self.async_show_form(step_id="coordinates", data_schema=data_schema)
|
|
|
|
assert self._data is not None
|
|
|
|
return await self._validate_and_create(
|
|
"coordinates", data_schema, {**self._data, **user_input}
|
|
)
|
|
|
|
async def async_step_country(
|
|
self, user_input: dict[str, Any] | None = None
|
|
) -> FlowResult:
|
|
"""Validate country."""
|
|
data_schema = vol.Schema(
|
|
{
|
|
vol.Required(CONF_COUNTRY_CODE): cv.string,
|
|
}
|
|
)
|
|
if user_input is None:
|
|
return self.async_show_form(step_id="country", data_schema=data_schema)
|
|
|
|
assert self._data is not None
|
|
|
|
return await self._validate_and_create(
|
|
"country", data_schema, {**self._data, **user_input}
|
|
)
|
|
|
|
async def _validate_and_create(
|
|
self, step_id: str, data_schema: vol.Schema, data: dict
|
|
) -> FlowResult:
|
|
"""Validate data and show form if it is invalid."""
|
|
errors: dict[str, str] = {}
|
|
|
|
try:
|
|
await self.hass.async_add_executor_job(get_data, self.hass, data)
|
|
except InvalidAuth:
|
|
errors["base"] = "invalid_auth"
|
|
except APIRatelimitExceeded:
|
|
errors["base"] = "api_ratelimit"
|
|
except Exception: # pylint: disable=broad-except
|
|
errors["base"] = "unknown"
|
|
else:
|
|
return self.async_create_entry(
|
|
title=get_extra_name(data) or "CO2 Signal",
|
|
data=data,
|
|
)
|
|
|
|
return self.async_show_form(
|
|
step_id=step_id,
|
|
data_schema=data_schema,
|
|
errors=errors,
|
|
)
|