evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Try negative WAQI station number before aborting (#102550)

Browse source
This commit is contained in:
Joost Lekkerkerker 2023-10-23 11:55:12 +02:00 • committed by GitHub
parent e27baedf32
commit 8a7de27946
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
homeassistant/components/waqi/config_flow.py
View file

@ -39,6 +39,22 @@ _LOGGER = logging.getLogger(__name__)
CONF_MAP = "map"
async def get_by_station_number(
client: WAQIClient, station_number: int
) -> tuple[WAQIAirQuality | None, dict[str, str]]:
"""Get measuring station by station number."""
errors: dict[str, str] = {}
measuring_station: WAQIAirQuality | None = None
try:
measuring_station = await client.get_by_station_number(station_number)
except WAQIConnectionError:
errors["base"] = "cannot_connect"
except Exception as exc: # pylint: disable=broad-except
_LOGGER.exception(exc)
errors["base"] = "unknown"
return measuring_station, errors
class WAQIConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow for World Air Quality Index (WAQI)."""
@ -141,16 +157,16 @@ class WAQIConfigFlow(ConfigFlow, domain=DOMAIN):
session=async_get_clientsession(self.hass)
) as waqi_client:
waqi_client.authenticate(self.data[CONF_API_KEY])
try:
measuring_station = await waqi_client.get_by_station_number(
user_input[CONF_STATION_NUMBER]
station_number = user_input[CONF_STATION_NUMBER]
measuring_station, errors = await get_by_station_number(
waqi_client, abs(station_number)
)
if not measuring_station:
measuring_station, _ = await get_by_station_number(
waqi_client,
abs(station_number) - station_number - station_number,
)
except WAQIConnectionError:
errors["base"] = "cannot_connect"
except Exception as exc: # pylint: disable=broad-except
_LOGGER.exception(exc)
errors["base"] = "unknown"
else:
if measuring_station:
return await self._async_create_entry(measuring_station)
return self.async_show_form(
step_id=CONF_STATION_NUMBER,