Use _get_reauth_entry in trafikverket_weatherstation (#127316)

This commit is contained in:
epenet 2024-10-02 15:35:34 +02:00 committed by GitHub
parent 781c3eed2f
commit 56e79de707
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -26,7 +26,7 @@ class TVWeatherConfigFlow(ConfigFlow, domain=DOMAIN):
VERSION = 1
entry: ConfigEntry | None = None
entry: ConfigEntry
async def validate_input(self, sensor_api: str, station: str) -> None:
"""Validate input from user input."""
@ -80,7 +80,7 @@ class TVWeatherConfigFlow(ConfigFlow, domain=DOMAIN):
) -> ConfigFlowResult:
"""Handle re-authentication with Trafikverket."""
self.entry = self.hass.config_entries.async_get_entry(self.context["entry_id"])
self.entry = self._get_reauth_entry()
return await self.async_step_reauth_confirm()
async def async_step_reauth_confirm(
@ -92,8 +92,6 @@ class TVWeatherConfigFlow(ConfigFlow, domain=DOMAIN):
if user_input:
api_key = user_input[CONF_API_KEY]
assert self.entry is not None
try:
await self.validate_input(api_key, self.entry.data[CONF_STATION])
except InvalidAuthentication: