Add async_get_options_flow type hints (n-z) (#73431)

This commit is contained in:
epenet 2022-06-13 13:30:41 +02:00 committed by GitHub
parent b589700651
commit 42ed0fd47b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
20 changed files with 106 additions and 35 deletions

View file

@ -1,4 +1,6 @@
"""Config flow for Risco integration."""
from __future__ import annotations
import logging
from pyrisco import CannotConnectError, RiscoAPI, UnauthorizedError
@ -67,7 +69,9 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
@staticmethod
@core.callback
def async_get_options_flow(config_entry):
def async_get_options_flow(
config_entry: config_entries.ConfigEntry,
) -> RiscoOptionsFlowHandler:
"""Define the config flow to handle options."""
return RiscoOptionsFlowHandler(config_entry)
@ -98,7 +102,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
class RiscoOptionsFlowHandler(config_entries.OptionsFlow):
"""Handle a Risco options flow."""
def __init__(self, config_entry):
def __init__(self, config_entry: config_entries.ConfigEntry) -> None:
"""Initialize."""
self.config_entry = config_entry
self._data = {**DEFAULT_OPTIONS, **config_entry.options}