74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
"""Config Flow for the zamg integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import voluptuous as vol
|
|
from zamg import ZamgData
|
|
from zamg.exceptions import ZamgApiError, ZamgNoDataError
|
|
|
|
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
|
|
from .const import CONF_STATION_ID, DOMAIN, LOGGER
|
|
|
|
|
|
class ZamgConfigFlow(ConfigFlow, domain=DOMAIN):
|
|
"""Config flow for zamg integration."""
|
|
|
|
VERSION = 1
|
|
|
|
_client: ZamgData | None = None
|
|
|
|
async def async_step_user(
|
|
self, user_input: dict[str, Any] | None = None
|
|
) -> ConfigFlowResult:
|
|
"""Handle a flow initiated by the user."""
|
|
if self._client is None:
|
|
self._client = ZamgData()
|
|
self._client.session = async_get_clientsession(self.hass)
|
|
|
|
if user_input is None:
|
|
try:
|
|
stations = await self._client.zamg_stations()
|
|
closest_station_id = await self._client.closest_station(
|
|
self.hass.config.latitude,
|
|
self.hass.config.longitude,
|
|
)
|
|
except (ZamgApiError, ZamgNoDataError) as err:
|
|
LOGGER.error("Config_flow: Received error from ZAMG: %s", err)
|
|
return self.async_abort(reason="cannot_connect")
|
|
LOGGER.debug("config_flow: closest station = %s", closest_station_id)
|
|
user_input = {}
|
|
|
|
schema = vol.Schema(
|
|
{
|
|
vol.Required(CONF_STATION_ID, default=closest_station_id): vol.In(
|
|
{
|
|
station: f"{stations[station][2]} ({station})"
|
|
for station in stations
|
|
}
|
|
)
|
|
}
|
|
)
|
|
return self.async_show_form(step_id="user", data_schema=schema)
|
|
|
|
station_id = user_input[CONF_STATION_ID]
|
|
|
|
# Check if already configured
|
|
await self.async_set_unique_id(station_id)
|
|
self._abort_if_unique_id_configured()
|
|
|
|
try:
|
|
self._client.set_default_station(station_id)
|
|
await self._client.update()
|
|
except (ZamgApiError, ZamgNoDataError) as err:
|
|
LOGGER.error("Config_flow: Received error from ZAMG: %s", err)
|
|
return self.async_abort(reason="cannot_connect")
|
|
|
|
return self.async_create_entry(
|
|
title=self._client.get_station_name,
|
|
data={CONF_STATION_ID: station_id},
|
|
)
|