jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 05:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

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" 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): class WAQIConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow for World Air Quality Index (WAQI).""" """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) session=async_get_clientsession(self.hass)
) as waqi_client: ) as waqi_client:
waqi_client.authenticate(self.data[CONF_API_KEY]) waqi_client.authenticate(self.data[CONF_API_KEY])
try: station_number = user_input[CONF_STATION_NUMBER]
measuring_station = await waqi_client.get_by_station_number( measuring_station, errors = await get_by_station_number(
user_input[CONF_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: if measuring_station:
errors["base"] = "cannot_connect"
except Exception as exc: # pylint: disable=broad-except
_LOGGER.exception(exc)
errors["base"] = "unknown"
else:
return await self._async_create_entry(measuring_station) return await self._async_create_entry(measuring_station)
return self.async_show_form( return self.async_show_form(
step_id=CONF_STATION_NUMBER, step_id=CONF_STATION_NUMBER,