evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use OptionsFlowWithConfigEntry in androidtv (#82805)

Browse source
This commit is contained in:
epenet 2022-11-28 09:47:40 +01:00 • committed by GitHub
parent 6e37aa425b
commit b2b3e14810
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
homeassistant/components/androidtv/config_flow.py
View file

@ -9,7 +9,11 @@ from typing import Any
from androidtv import state_detection_rules_validator from androidtv import state_detection_rules_validator
import voluptuous as vol import voluptuous as vol
from homeassistant.config_entries import ConfigEntry, ConfigFlow, OptionsFlow from homeassistant.config_entries import (
ConfigEntry,
ConfigFlow,
OptionsFlowWithConfigEntry,
)
from homeassistant.const import CONF_DEVICE_CLASS, CONF_HOST, CONF_PORT from homeassistant.const import CONF_DEVICE_CLASS, CONF_HOST, CONF_PORT
from homeassistant.core import callback from homeassistant.core import callback
from homeassistant.data_entry_flow import FlowResult from homeassistant.data_entry_flow import FlowResult
@ -168,22 +172,22 @@ class AndroidTVFlowHandler(ConfigFlow, domain=DOMAIN):
@staticmethod @staticmethod
@callback @callback
def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlow: def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlowHandler:
"""Get the options flow for this handler.""" """Get the options flow for this handler."""
return OptionsFlowHandler(config_entry) return OptionsFlowHandler(config_entry)
class OptionsFlowHandler(OptionsFlow): class OptionsFlowHandler(OptionsFlowWithConfigEntry):
"""Handle an option flow for Android TV.""" """Handle an option flow for Android TV."""
def __init__(self, config_entry: ConfigEntry) -> None: def __init__(self, config_entry: ConfigEntry) -> None:
"""Initialize options flow.""" """Initialize options flow."""
self.config_entry = config_entry super().__init__(config_entry)
apps = config_entry.options.get(CONF_APPS, {}) self._apps: dict[str, Any] = self.options.setdefault(CONF_APPS, {})
det_rules = config_entry.options.get(CONF_STATE_DETECTION_RULES, {}) self._state_det_rules: dict[str, Any] = self.options.setdefault(
self._apps: dict[str, Any] = apps.copy() CONF_STATE_DETECTION_RULES, {}
self._state_det_rules: dict[str, Any] = det_rules.copy() )
self._conf_app_id: str | None = None self._conf_app_id: str | None = None
self._conf_rule_id: str | None = None self._conf_rule_id: str | None = None
@ -222,7 +226,7 @@ class OptionsFlowHandler(OptionsFlow):
apps_list = {k: f"{v} ({k})" if v else k for k, v in self._apps.items()} apps_list = {k: f"{v} ({k})" if v else k for k, v in self._apps.items()}
apps = {APPS_NEW_ID: "Add new", **apps_list} apps = {APPS_NEW_ID: "Add new", **apps_list}
rules = [RULES_NEW_ID] + list(self._state_det_rules) rules = [RULES_NEW_ID] + list(self._state_det_rules)
options = self.config_entry.options options = self.options
data_schema = vol.Schema( data_schema = vol.Schema(
{ {