diff --git a/homeassistant/components/hvv_departures/config_flow.py b/homeassistant/components/hvv_departures/config_flow.py index 488579af12b..d96ab359dda 100644 --- a/homeassistant/components/hvv_departures/config_flow.py +++ b/homeassistant/components/hvv_departures/config_flow.py @@ -1,5 +1,8 @@ """Config flow for HVV integration.""" +from __future__ import annotations + import logging +from typing import Any from pygti.auth import GTI_DEFAULT_HOST from pygti.exceptions import CannotConnect, InvalidAuth @@ -122,7 +125,9 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): @staticmethod @callback - def async_get_options_flow(config_entry): + def async_get_options_flow( + config_entry: config_entries.ConfigEntry, + ) -> OptionsFlowHandler: """Get options flow.""" return OptionsFlowHandler(config_entry) @@ -130,12 +135,11 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): class OptionsFlowHandler(config_entries.OptionsFlow): """Options flow handler.""" - def __init__(self, config_entry): + def __init__(self, config_entry: config_entries.ConfigEntry) -> None: """Initialize HVV Departures options flow.""" self.config_entry = config_entry self.options = dict(config_entry.options) - self.departure_filters = {} - self.hub = None + self.departure_filters: dict[str, Any] = {} async def async_step_init(self, user_input=None): """Manage the options.""" @@ -143,10 +147,10 @@ class OptionsFlowHandler(config_entries.OptionsFlow): if not self.departure_filters: departure_list = {} - self.hub = self.hass.data[DOMAIN][self.config_entry.entry_id] + hub: GTIHub = self.hass.data[DOMAIN][self.config_entry.entry_id] try: - departure_list = await self.hub.gti.departureList( + departure_list = await hub.gti.departureList( { "station": self.config_entry.data[CONF_STATION], "time": {"date": "heute", "time": "jetzt"},