jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 15:17:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use reauth helpers in homewizard (#128628)

Browse Source
This commit is contained in:
epenet 2024-10-18 09:22:31 +02:00 committed by GitHub
parent 5986646af4
commit c696a3b789
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/homewizard/config_flow.py
View File

@ -12,7 +12,7 @@ from homewizard_energy.models import Device
from voluptuous import Required, Schema from voluptuous import Required, Schema
from homeassistant.components import onboarding, zeroconf from homeassistant.components import onboarding, zeroconf
from homeassistant.config_entries import ConfigEntry, ConfigFlow, ConfigFlowResult from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_IP_ADDRESS, CONF_PATH from homeassistant.const import CONF_IP_ADDRESS, CONF_PATH
from homeassistant.data_entry_flow import AbortFlow from homeassistant.data_entry_flow import AbortFlow
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
@ -43,7 +43,6 @@ class HomeWizardConfigFlow(ConfigFlow, domain=DOMAIN):
VERSION = 1 VERSION = 1
discovery: DiscoveryData discovery: DiscoveryData
entry: ConfigEntry | None
async def async_step_user( async def async_step_user(
self, user_input: dict[str, Any] | None = None self, user_input: dict[str, Any] | None = None
@ -151,7 +150,6 @@ class HomeWizardConfigFlow(ConfigFlow, domain=DOMAIN):
self, entry_data: Mapping[str, Any] self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult: ) -> ConfigFlowResult:
"""Handle re-auth if API was disabled.""" """Handle re-auth if API was disabled."""
self.entry = self.hass.config_entries.async_get_entry(self.context["entry_id"])
return await self.async_step_reauth_confirm() return await self.async_step_reauth_confirm()
async def async_step_reauth_confirm( async def async_step_reauth_confirm(
@ -160,20 +158,17 @@ class HomeWizardConfigFlow(ConfigFlow, domain=DOMAIN):
"""Confirm reauth dialog.""" """Confirm reauth dialog."""
errors: dict[str, str] | None = None errors: dict[str, str] | None = None
if user_input is not None: if user_input is not None:
assert self.entry is not None reauth_entry = self._get_reauth_entry()
try: try:
await self._async_try_connect(self.entry.data[CONF_IP_ADDRESS]) await self._async_try_connect(reauth_entry.data[CONF_IP_ADDRESS])
except RecoverableError as ex: except RecoverableError as ex:
_LOGGER.error(ex) _LOGGER.error(ex)
errors = {"base": ex.error_code} errors = {"base": ex.error_code}
else: else:
await self.hass.config_entries.async_reload(self.entry.entry_id) await self.hass.config_entries.async_reload(reauth_entry.entry_id)
return self.async_abort(reason="reauth_successful") return self.async_abort(reason="reauth_successful")
return self.async_show_form( return self.async_show_form(step_id="reauth_confirm", errors=errors)
step_id="reauth_confirm",
errors=errors,
)
@staticmethod @staticmethod
async def _async_try_connect(ip_address: str) -> Device: async def _async_try_connect(ip_address: str) -> Device: