From ddea61148f3d69a99be6c138540292e2dbe8388b Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 2 Oct 2024 21:56:26 +0200
Subject: [PATCH] Use _get_reconfigure_entry in brother (#127279)
---
homeassistant/components/brother/config_flow.py | 16 ++++------------
1 file changed, 4 insertions(+), 12 deletions(-)
diff --git a/homeassistant/components/brother/config_flow.py b/homeassistant/components/brother/config_flow.py
index cb98be30f8b..f9c51d3b786 100644
--- a/homeassistant/components/brother/config_flow.py
+++ b/homeassistant/components/brother/config_flow.py
@@ -3,7 +3,7 @@
from __future__ import annotations
from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any
+from typing import Any
from brother import Brother, SnmpError, UnsupportedModelError
import voluptuous as vol
@@ -50,11 +50,12 @@ class BrotherConfigFlow(ConfigFlow, domain=DOMAIN):
VERSION = 1
+ entry: ConfigEntry
+
def __init__(self) -> None:
"""Initialize."""
self.brother: Brother
self.host: str | None = None
- self.entry: ConfigEntry | None = None
async def async_step_user(
self, user_input: dict[str, Any] | None = None
@@ -145,13 +146,7 @@ class BrotherConfigFlow(ConfigFlow, domain=DOMAIN):
self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult:
"""Handle a reconfiguration flow initialized by the user."""
- entry = self.hass.config_entries.async_get_entry(self.context["entry_id"])
-
- if TYPE_CHECKING:
- assert entry is not None
-
- self.entry = entry
-
+ self.entry = self._get_reconfigure_entry()
return await self.async_step_reconfigure_confirm()
async def async_step_reconfigure_confirm(
@@ -160,9 +155,6 @@ class BrotherConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a reconfiguration flow initialized by the user."""
errors = {}
- if TYPE_CHECKING:
- assert self.entry is not None
-
if user_input is not None:
try:
await validate_input(self.hass, user_input, self.entry.unique_id)
--
GitLab