mirror of https://github.com/home-assistant/core
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
"""Provide configuration end points for Automations."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
import uuid
|
|
|
|
from homeassistant.components.automation import DOMAIN as AUTOMATION_DOMAIN
|
|
from homeassistant.components.automation.config import async_validate_config_item
|
|
from homeassistant.config import AUTOMATION_CONFIG_PATH
|
|
from homeassistant.const import CONF_ID, SERVICE_RELOAD
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers import config_validation as cv, entity_registry as er
|
|
|
|
from .const import ACTION_DELETE
|
|
from .view import EditIdBasedConfigView
|
|
|
|
|
|
@callback
|
|
def async_setup(hass: HomeAssistant) -> bool:
|
|
"""Set up the Automation config API."""
|
|
|
|
async def hook(action: str, config_key: str) -> None:
|
|
"""post_write_hook for Config View that reloads automations."""
|
|
if action != ACTION_DELETE:
|
|
await hass.services.async_call(
|
|
AUTOMATION_DOMAIN, SERVICE_RELOAD, {CONF_ID: config_key}
|
|
)
|
|
return
|
|
|
|
ent_reg = er.async_get(hass)
|
|
|
|
entity_id = ent_reg.async_get_entity_id(
|
|
AUTOMATION_DOMAIN, AUTOMATION_DOMAIN, config_key
|
|
)
|
|
|
|
if entity_id is None:
|
|
return
|
|
|
|
ent_reg.async_remove(entity_id)
|
|
|
|
hass.http.register_view(
|
|
EditAutomationConfigView(
|
|
AUTOMATION_DOMAIN,
|
|
"config",
|
|
AUTOMATION_CONFIG_PATH,
|
|
cv.string,
|
|
post_write_hook=hook,
|
|
data_validator=async_validate_config_item,
|
|
)
|
|
)
|
|
return True
|
|
|
|
|
|
class EditAutomationConfigView(EditIdBasedConfigView):
|
|
"""Edit automation config."""
|
|
|
|
def _write_value(
|
|
self,
|
|
hass: HomeAssistant,
|
|
data: list[dict[str, Any]],
|
|
config_key: str,
|
|
new_value: dict[str, Any],
|
|
) -> None:
|
|
"""Set value."""
|
|
updated_value = {CONF_ID: config_key}
|
|
|
|
# Iterate through some keys that we want to have ordered in the output
|
|
for key in (
|
|
"alias",
|
|
"description",
|
|
"triggers",
|
|
"trigger",
|
|
"conditions",
|
|
"condition",
|
|
"actions",
|
|
"action",
|
|
):
|
|
if key in new_value:
|
|
updated_value[key] = new_value[key]
|
|
|
|
# We cover all current fields above, but just in case we start
|
|
# supporting more fields in the future.
|
|
updated_value.update(new_value)
|
|
|
|
updated = False
|
|
for index, cur_value in enumerate(data):
|
|
# When people copy paste their automations to the config file,
|
|
# they sometimes forget to add IDs. Fix it here.
|
|
if CONF_ID not in cur_value:
|
|
cur_value[CONF_ID] = uuid.uuid4().hex
|
|
|
|
elif cur_value[CONF_ID] == config_key:
|
|
data[index] = updated_value
|
|
updated = True
|
|
|
|
if not updated:
|
|
data.append(updated_value)
|