Use new config entry update/abort helper in samsungtv (#108683)

Use new config entry update/abort helper in samsungtv

uses the new helper from https://github.com/home-assistant/core/pull/108034
This commit is contained in:
J. Nick Koston 2024-01-22 11:59:50 -10:00 committed by GitHub
parent faf52aa2ed
commit 17202e21f3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -528,11 +528,10 @@ class SamsungTVConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
if result == RESULT_SUCCESS: if result == RESULT_SUCCESS:
new_data = dict(self._reauth_entry.data) new_data = dict(self._reauth_entry.data)
new_data[CONF_TOKEN] = bridge.token new_data[CONF_TOKEN] = bridge.token
self.hass.config_entries.async_update_entry( return self.async_update_reload_and_abort(
self._reauth_entry, data=new_data self._reauth_entry,
data=new_data,
) )
await self.hass.config_entries.async_reload(self._reauth_entry.entry_id)
return self.async_abort(reason="reauth_successful")
if result not in (RESULT_AUTH_MISSING, RESULT_CANNOT_CONNECT): if result not in (RESULT_AUTH_MISSING, RESULT_CANNOT_CONNECT):
return self.async_abort(reason=result) return self.async_abort(reason=result)
@ -569,7 +568,7 @@ class SamsungTVConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
and (token := await self._authenticator.try_pin(pin)) and (token := await self._authenticator.try_pin(pin))
and (session_id := await self._authenticator.get_session_id_and_close()) and (session_id := await self._authenticator.get_session_id_and_close())
): ):
self.hass.config_entries.async_update_entry( return self.async_update_reload_and_abort(
self._reauth_entry, self._reauth_entry,
data={ data={
**self._reauth_entry.data, **self._reauth_entry.data,
@ -577,8 +576,6 @@ class SamsungTVConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
CONF_SESSION_ID: session_id, CONF_SESSION_ID: session_id,
}, },
) )
await self.hass.config_entries.async_reload(self._reauth_entry.entry_id)
return self.async_abort(reason="reauth_successful")
errors = {"base": RESULT_INVALID_PIN} errors = {"base": RESULT_INVALID_PIN}