config_flow.py

Go to the documentation of this file.

"""Config flow for Times of the Day integration."""
 
from __future__ import annotations
 
from collections.abc import Mapping
from typing import Any, cast
 
import voluptuous as vol
 
from homeassistant.const import CONF_NAME
from homeassistant.helpers import selector
from homeassistant.helpers.schema_config_entry_flow import (
    SchemaConfigFlowHandler,
    SchemaFlowFormStep,
)
 
from .const import CONF_AFTER_TIME, CONF_BEFORE_TIME, DOMAIN
 
OPTIONS_SCHEMA = vol.Schema(
    {
        vol.Required(CONF_AFTER_TIME): selector.TimeSelector(),
        vol.Required(CONF_BEFORE_TIME): selector.TimeSelector(),
    }
)
 
CONFIG_SCHEMA = vol.Schema(
    {
        vol.Required(CONF_NAME): selector.TextSelector(),
    }
).extend(OPTIONS_SCHEMA.schema)
 
CONFIG_FLOW = {
    "user": SchemaFlowFormStep(CONFIG_SCHEMA),
}
 
OPTIONS_FLOW = {
    "init": SchemaFlowFormStep(OPTIONS_SCHEMA),
}
 
 
class ConfigFlowHandler(SchemaConfigFlowHandler, domain=DOMAIN):
    """Handle a config or options flow for Times of the Day."""
 
    config_flow = CONFIG_FLOW
    options_flow = OPTIONS_FLOW
 
    def async_config_entry_title(self, options: Mapping[str, Any]) -> str:
        """Return config entry title."""
        return cast(str, options["name"])