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

Improve type hints in growatt_server config flow (#124901)

Browse Source
This commit is contained in:
epenet 2024-08-30 11:25:29 +02:00 committed by GitHub
parent febb382030
commit afa02dcce9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/growatt_server/config_flow.py
View File

@ -23,9 +23,10 @@ class GrowattServerConfigFlow(ConfigFlow, domain=DOMAIN):
VERSION = 1 VERSION = 1
api: growattServer.GrowattApi
def __init__(self) -> None: def __init__(self) -> None:
"""Initialise growatt server flow.""" """Initialise growatt server flow."""
self.api: growattServer.GrowattApi | None = None
self.user_id = None self.user_id = None
self.data: dict[str, Any] = {} self.data: dict[str, Any] = {}
@ -70,7 +71,9 @@ class GrowattServerConfigFlow(ConfigFlow, domain=DOMAIN):
self.data = user_input self.data = user_input
return await self.async_step_plant() return await self.async_step_plant()
async def async_step_plant(self, user_input=None): async def async_step_plant(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Handle adding a "plant" to Home Assistant.""" """Handle adding a "plant" to Home Assistant."""
plant_info = await self.hass.async_add_executor_job( plant_info = await self.hass.async_add_executor_job(
self.api.plant_list, self.user_id self.api.plant_list, self.user_id
@ -86,7 +89,8 @@ class GrowattServerConfigFlow(ConfigFlow, domain=DOMAIN):
return self.async_show_form(step_id="plant", data_schema=data_schema) return self.async_show_form(step_id="plant", data_schema=data_schema)
if user_input is None and len(plant_info["data"]) == 1: if user_input is None:
# single plant => mark it as selected
user_input = {CONF_PLANT_ID: plant_info["data"][0]["plantId"]} user_input = {CONF_PLANT_ID: plant_info["data"][0]["plantId"]}
user_input[CONF_NAME] = plants[user_input[CONF_PLANT_ID]] user_input[CONF_NAME] = plants[user_input[CONF_PLANT_ID]]