mirror of
https://github.com/jdejaegh/irm-kmi-ha.git
synced 2025-06-27 11:39:26 +02:00
99 lines
4.5 KiB
Python
99 lines
4.5 KiB
Python
"""Config flow to set up IRM KMI integration via the UI."""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
from homeassistant.components.zone import DOMAIN as ZONE_DOMAIN
|
|
from homeassistant.config_entries import ConfigFlow, OptionsFlow, ConfigEntry
|
|
from homeassistant.const import CONF_ZONE
|
|
from homeassistant.core import callback
|
|
from homeassistant.data_entry_flow import FlowResult
|
|
from homeassistant.helpers.selector import (EntitySelector,
|
|
EntitySelectorConfig,
|
|
SelectSelector,
|
|
SelectSelectorConfig,
|
|
SelectSelectorMode)
|
|
|
|
from .utils import get_config_value
|
|
from .const import (CONF_DARK_MODE, CONF_STYLE, CONF_STYLE_OPTIONS,
|
|
OPTION_STYLE_STD, DOMAIN, CONF_USE_DEPRECATED_FORECAST, OPTION_DEPRECATED_FORECAST_NOT_USED,
|
|
CONF_USE_DEPRECATED_FORECAST_OPTIONS)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class IrmKmiConfigFlow(ConfigFlow, domain=DOMAIN):
|
|
VERSION = 3
|
|
|
|
@staticmethod
|
|
@callback
|
|
def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlow:
|
|
"""Create the options flow."""
|
|
return IrmKmiOptionFlow(config_entry)
|
|
|
|
async def async_step_user(self, user_input: dict | None = None) -> FlowResult:
|
|
"""Define the user step of the configuration flow."""
|
|
if user_input is not None:
|
|
_LOGGER.debug(f"Provided config user is: {user_input}")
|
|
|
|
await self.async_set_unique_id(user_input[CONF_ZONE])
|
|
self._abort_if_unique_id_configured()
|
|
|
|
state = self.hass.states.get(user_input[CONF_ZONE])
|
|
return self.async_create_entry(
|
|
title=state.name if state else "IRM KMI",
|
|
data={CONF_ZONE: user_input[CONF_ZONE],
|
|
CONF_STYLE: user_input[CONF_STYLE],
|
|
CONF_DARK_MODE: user_input[CONF_DARK_MODE],
|
|
CONF_USE_DEPRECATED_FORECAST: user_input[CONF_USE_DEPRECATED_FORECAST]},
|
|
)
|
|
|
|
return self.async_show_form(
|
|
step_id="user",
|
|
data_schema=vol.Schema({
|
|
vol.Required(CONF_ZONE):
|
|
EntitySelector(EntitySelectorConfig(domain=ZONE_DOMAIN)),
|
|
|
|
vol.Optional(CONF_STYLE, default=OPTION_STYLE_STD):
|
|
SelectSelector(SelectSelectorConfig(options=CONF_STYLE_OPTIONS,
|
|
mode=SelectSelectorMode.DROPDOWN,
|
|
translation_key=CONF_STYLE)),
|
|
|
|
vol.Optional(CONF_DARK_MODE, default=False): bool,
|
|
|
|
vol.Optional(CONF_USE_DEPRECATED_FORECAST, default=OPTION_DEPRECATED_FORECAST_NOT_USED):
|
|
SelectSelector(SelectSelectorConfig(options=CONF_USE_DEPRECATED_FORECAST_OPTIONS,
|
|
mode=SelectSelectorMode.DROPDOWN,
|
|
translation_key=CONF_USE_DEPRECATED_FORECAST))
|
|
|
|
}))
|
|
|
|
|
|
class IrmKmiOptionFlow(OptionsFlow):
|
|
def __init__(self, config_entry: ConfigEntry) -> None:
|
|
"""Initialize options flow."""
|
|
self.config_entry = config_entry
|
|
|
|
async def async_step_init(self, user_input: dict | None = None) -> FlowResult:
|
|
"""Manage the options."""
|
|
if user_input is not None:
|
|
return self.async_create_entry(data=user_input)
|
|
|
|
return self.async_show_form(
|
|
step_id="init",
|
|
data_schema=vol.Schema(
|
|
{
|
|
vol.Optional(CONF_STYLE, default=get_config_value(self.config_entry, CONF_STYLE)):
|
|
SelectSelector(SelectSelectorConfig(options=CONF_STYLE_OPTIONS,
|
|
mode=SelectSelectorMode.DROPDOWN,
|
|
translation_key=CONF_STYLE)),
|
|
|
|
vol.Optional(CONF_DARK_MODE, default=get_config_value(self.config_entry, CONF_DARK_MODE)): bool,
|
|
|
|
vol.Optional(CONF_USE_DEPRECATED_FORECAST,
|
|
default=get_config_value(self.config_entry, CONF_USE_DEPRECATED_FORECAST)):
|
|
SelectSelector(SelectSelectorConfig(options=CONF_USE_DEPRECATED_FORECAST_OPTIONS,
|
|
mode=SelectSelectorMode.DROPDOWN,
|
|
translation_key=CONF_USE_DEPRECATED_FORECAST))
|
|
}
|
|
),
|
|
)
|