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

Preserves config flow information in case of error for SQL (#91142)

pull/91299/head
dougiteixeira 2023-04-12 16:03:14 -03:00 committed by GitHub
parent 6aa1460143
commit 1ef6391e9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
homeassistant/components/sql/config_flow.py
View File

@ -22,19 +22,32 @@ from .util import resolve_db_url
_LOGGER = logging.getLogger(__name__)
DATA_SCHEMA = vol.Schema(
OPTIONS_SCHEMA: vol.Schema = vol.Schema(
{
vol.Required(CONF_NAME, default="Select SQL Query"): selector.TextSelector(),
vol.Optional(CONF_DB_URL): selector.TextSelector(),
vol.Required(CONF_COLUMN_NAME): selector.TextSelector(),
vol.Required(CONF_QUERY): selector.TextSelector(
selector.TextSelectorConfig(multiline=True)
),
vol.Optional(CONF_UNIT_OF_MEASUREMENT): selector.TextSelector(),
vol.Optional(CONF_VALUE_TEMPLATE): selector.TemplateSelector(),
vol.Optional(
CONF_DB_URL,
): selector.TextSelector(),
vol.Required(
CONF_COLUMN_NAME,
): selector.TextSelector(),
vol.Required(
CONF_QUERY,
): selector.TextSelector(selector.TextSelectorConfig(multiline=True)),
vol.Optional(
CONF_UNIT_OF_MEASUREMENT,
): selector.TextSelector(),
vol.Optional(
CONF_VALUE_TEMPLATE,
): selector.TemplateSelector(),
}
)
CONFIG_SCHEMA: vol.Schema = vol.Schema(
{
vol.Required(CONF_NAME, default="Select SQL Query"): selector.TextSelector(),
}
).extend(OPTIONS_SCHEMA.schema)
def validate_sql_select(value: str) -> str | None:
"""Validate that value is a SQL SELECT query."""
@ -127,7 +140,7 @@ class SQLConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
return self.async_show_form(
step_id="user",
data_schema=DATA_SCHEMA,
data_schema=self.add_suggested_values_to_schema(CONFIG_SCHEMA, user_input),
errors=errors,
)
@ -175,43 +188,8 @@ class SQLOptionsFlowHandler(config_entries.OptionsFlow):
return self.async_show_form(
step_id="init",
data_schema=vol.Schema(
{
vol.Optional(
CONF_DB_URL,
description={
"suggested_value": self.entry.options.get(CONF_DB_URL)
},
): selector.TextSelector(),
vol.Required(
CONF_QUERY,
description={"suggested_value": self.entry.options[CONF_QUERY]},
): selector.TextSelector(
selector.TextSelectorConfig(multiline=True)
),
vol.Required(
CONF_COLUMN_NAME,
description={
"suggested_value": self.entry.options[CONF_COLUMN_NAME]
},
): selector.TextSelector(),
vol.Optional(
CONF_UNIT_OF_MEASUREMENT,
description={
"suggested_value": self.entry.options.get(
CONF_UNIT_OF_MEASUREMENT
)
},
): selector.TextSelector(),
vol.Optional(
CONF_VALUE_TEMPLATE,
description={
"suggested_value": self.entry.options.get(
CONF_VALUE_TEMPLATE
)
},
): selector.TemplateSelector(),
}
data_schema=self.add_suggested_values_to_schema(
OPTIONS_SCHEMA, user_input or self.entry.options
),
errors=errors,
)