Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactor schema generation in Template integration (#120889) 

Co-authored-by: Robert Resch <robert@resch.dev>
pull/121472/head
dougiteixeira 2024-07-08 04:06:15 -03:00 committed by GitHub
parent e30f315565
commit cf4bd7fd1c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
homeassistant/components/template/config_flow.py
View File

@ -3,6 +3,7 @@
from __future__ import annotations
from collections.abc import Callable, Coroutine, Mapping
from functools import partial
from typing import Any, cast
import voluptuous as vol
@ -39,25 +40,34 @@ from .const import DOMAIN
from .sensor import async_create_preview_sensor
from .template_entity import TemplateEntity
_SCHEMA_STATE: dict[vol.Marker, Any] = {
vol.Required(CONF_STATE): selector.TemplateSelector(),
}
def generate_schema(domain: str, flow_type: str) -> dict[vol.Marker, Any]:
def generate_schema(domain: str, flow_type: str) -> vol.Schema:
"""Generate schema."""
schema: dict[vol.Marker, Any] = {}
if domain == Platform.BINARY_SENSOR and flow_type == "config":
schema = {
vol.Optional(CONF_DEVICE_CLASS): selector.SelectSelector(
selector.SelectSelectorConfig(
options=[cls.value for cls in BinarySensorDeviceClass],
mode=selector.SelectSelectorMode.DROPDOWN,
translation_key="binary_sensor_device_class",
sort=True,
if flow_type == "config":
schema = {vol.Required(CONF_NAME): selector.TextSelector()}
if domain == Platform.BINARY_SENSOR:
schema |= _SCHEMA_STATE
if flow_type == "config":
schema |= {
vol.Optional(CONF_DEVICE_CLASS): selector.SelectSelector(
selector.SelectSelectorConfig(
options=[cls.value for cls in BinarySensorDeviceClass],
mode=selector.SelectSelectorMode.DROPDOWN,
translation_key="binary_sensor_device_class",
sort=True,
),
),
)
}
}
if domain == Platform.SENSOR:
schema = {
schema |= _SCHEMA_STATE | {
vol.Optional(CONF_UNIT_OF_MEASUREMENT): selector.SelectSelector(
selector.SelectSelectorConfig(
options=list(
@ -98,26 +108,12 @@ def generate_schema(domain: str, flow_type: str) -> dict[vol.Marker, Any]:
schema[vol.Optional(CONF_DEVICE_ID)] = selector.DeviceSelector()
return schema
return vol.Schema(schema)
def options_schema(domain: str) -> vol.Schema:
"""Generate options schema."""
return vol.Schema(
{vol.Required(CONF_STATE): selector.TemplateSelector()}
| generate_schema(domain, "option"),
)
options_schema = partial(generate_schema, flow_type="options")
def config_schema(domain: str) -> vol.Schema:
"""Generate config schema."""
return vol.Schema(
{
vol.Required(CONF_NAME): selector.TextSelector(),
vol.Required(CONF_STATE): selector.TemplateSelector(),
}
| generate_schema(domain, "config"),
)
config_schema = partial(generate_schema, flow_type="config")
async def choose_options_step(options: dict[str, Any]) -> str: