Use new config entry update/abort helper in samsungtv

Use new config entry update/abort helper in samsungtv uses the new helper from #108034
home-assistant · Jan 22, 2024 · 3cc5432 · 3cc5432
1 parent bd7eb01
commit 3cc5432
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/homeassistant/components/samsungtv/config_flow.py b/homeassistant/components/samsungtv/config_flow.py
@@ -528,11 +528,10 @@ async def async_step_reauth_confirm(
             if result == RESULT_SUCCESS:
                 new_data = dict(self._reauth_entry.data)
                 new_data[CONF_TOKEN] = bridge.token
-                self.hass.config_entries.async_update_entry(
-                    self._reauth_entry, data=new_data
+                return self.async_update_reload_and_abort(
+                    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):
                 return self.async_abort(reason=result)
 
@@ -569,16 +568,14 @@ async def async_step_reauth_confirm_encrypted(
                 and (token := await self._authenticator.try_pin(pin))
                 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,
                     data={
                         **self._reauth_entry.data,
                         CONF_TOKEN: token,
                         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}