82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
|
"""Config flow for Group integration."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Any, cast
|
||
|
|
|
||
|
|
import voluptuous as vol
|
||
|
|
|
||
|
|
from homeassistant.config_entries import ConfigEntry
|
||
|
|
from homeassistant.const import CONF_ENTITIES
|
||
|
|
from homeassistant.helpers import helper_config_entry_flow, selector
|
||
|
|
|
||
|
|
from . import DOMAIN
|
||
|
|
|
||
|
|
|
||
|
|
def basic_group_options_schema(domain: str) -> vol.Schema:
|
||
|
|
"""Generate options schema."""
|
||
|
|
return vol.Schema(
|
||
|
|
{
|
||
|
|
vol.Required(CONF_ENTITIES): selector.selector(
|
||
|
|
{"entity": {"domain": domain, "multiple": True}}
|
||
|
|
),
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def basic_group_config_schema(domain: str) -> vol.Schema:
|
||
|
|
"""Generate config schema."""
|
||
|
|
return vol.Schema({vol.Required("name"): selector.selector({"text": {}})}).extend(
|
||
|
|
basic_group_options_schema(domain).schema
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
STEPS = {
|
||
|
|
"init": vol.Schema(
|
||
|
|
{
|
||
|
|
vol.Required("group_type"): selector.selector(
|
||
|
|
{
|
||
|
|
"select": {
|
||
|
|
"options": [
|
||
|
|
"cover",
|
||
|
|
"fan",
|
||
|
|
"light",
|
||
|
|
"media_player",
|
||
|
|
]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
)
|
||
|
|
}
|
||
|
|
),
|
||
|
|
"cover": basic_group_config_schema("cover"),
|
||
|
|
"fan": basic_group_config_schema("fan"),
|
||
|
|
"light": basic_group_config_schema("light"),
|
||
|
|
"media_player": basic_group_config_schema("media_player"),
|
||
|
|
"cover_options": basic_group_options_schema("cover"),
|
||
|
|
"fan_options": basic_group_options_schema("fan"),
|
||
|
|
"light_options": basic_group_options_schema("light"),
|
||
|
|
"media_player_options": basic_group_options_schema("media_player"),
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class GroupConfigFlowHandler(
|
||
|
|
helper_config_entry_flow.HelperConfigFlowHandler, domain=DOMAIN
|
||
|
|
):
|
||
|
|
"""Handle a config or options flow for Switch Light."""
|
||
|
|
|
||
|
|
steps = STEPS
|
||
|
|
|
||
|
|
def async_config_entry_title(self, user_input: dict[str, Any]) -> str:
|
||
|
|
"""Return config entry title."""
|
||
|
|
return cast(str, user_input["name"]) if "name" in user_input else ""
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def async_initial_options_step(config_entry: ConfigEntry) -> str:
|
||
|
|
"""Return initial options step."""
|
||
|
|
return f"{config_entry.options['group_type']}_options"
|
||
|
|
|
||
|
|
def async_next_step(self, step_id: str, user_input: dict[str, Any]) -> str | None:
|
||
|
|
"""Return next step_id."""
|
||
|
|
if step_id == "init":
|
||
|
|
return cast(str, user_input["group_type"])
|
||
|
|
return None
|