Add set variables to scripts

Author: arturpragacz

homeassistant/const.py

@@ -268,6 +268,7 @@ class Platform(StrEnum):
 CONF_SERVICE_DATA_TEMPLATE: Final = "data_template"
 CONF_SERVICE_TEMPLATE: Final = "service_template"
 CONF_SET_CONVERSATION_RESPONSE: Final = "set_conversation_response"
+CONF_SET_VARIABLES: Final = "set_variables"
 CONF_SHOW_ON_MAP: Final = "show_on_map"
 CONF_SLAVE: Final = "slave"
 CONF_SOURCE: Final = "source"

homeassistant/helpers/config_validation.py

@@ -76,6 +76,7 @@
     CONF_SERVICE_DATA_TEMPLATE,
     CONF_SERVICE_TEMPLATE,
     CONF_SET_CONVERSATION_RESPONSE,
+    CONF_SET_VARIABLES,
     CONF_STATE,
     CONF_STOP,
     CONF_TARGET,
@@ -1926,13 +1927,20 @@ def _base_trigger_validator(value: Any) -> Any:
     }
 )
 
-_SCRIPT_SET_SCHEMA = vol.Schema(
+_SCRIPT_VARIABLES_SCHEMA = vol.Schema(
     {
         **SCRIPT_ACTION_BASE_SCHEMA,
         vol.Required(CONF_VARIABLES): SCRIPT_VARIABLES_SCHEMA,
     }
 )
 
+_SCRIPT_SET_VARIABLES_SCHEMA = vol.Schema(
+    {
+        **SCRIPT_ACTION_BASE_SCHEMA,
+        vol.Required(CONF_SET_VARIABLES): SCRIPT_VARIABLES_SCHEMA,
+    }
+)
+
 _SCRIPT_SET_CONVERSATION_RESPONSE_SCHEMA = vol.Schema(
     {
         **SCRIPT_ACTION_BASE_SCHEMA,
@@ -1992,6 +2000,7 @@ def _base_trigger_validator(value: Any) -> Any:
 SCRIPT_ACTION_REPEAT = "repeat"
 SCRIPT_ACTION_SEQUENCE = "sequence"
 SCRIPT_ACTION_SET_CONVERSATION_RESPONSE = "set_conversation_response"
+SCRIPT_ACTION_SET_VARIABLES = "set_variables"
 SCRIPT_ACTION_STOP = "stop"
 SCRIPT_ACTION_VARIABLES = "variables"
 SCRIPT_ACTION_WAIT_FOR_TRIGGER = "wait_for_trigger"
@@ -2012,6 +2021,7 @@ def _base_trigger_validator(value: Any) -> Any:
     CONF_CHOOSE: SCRIPT_ACTION_CHOOSE,
     CONF_WAIT_FOR_TRIGGER: SCRIPT_ACTION_WAIT_FOR_TRIGGER,
     CONF_VARIABLES: SCRIPT_ACTION_VARIABLES,
+    CONF_SET_VARIABLES: SCRIPT_ACTION_SET_VARIABLES,
     CONF_IF: SCRIPT_ACTION_IF,
     CONF_ACTION: SCRIPT_ACTION_CALL_SERVICE,
     CONF_SERVICE: SCRIPT_ACTION_CALL_SERVICE,
@@ -2051,8 +2061,9 @@ def determine_script_action(action: dict[str, Any]) -> str:
     SCRIPT_ACTION_REPEAT: _SCRIPT_REPEAT_SCHEMA,
     SCRIPT_ACTION_SEQUENCE: _SCRIPT_SEQUENCE_SCHEMA,
     SCRIPT_ACTION_SET_CONVERSATION_RESPONSE: _SCRIPT_SET_CONVERSATION_RESPONSE_SCHEMA,
+    SCRIPT_ACTION_SET_VARIABLES: _SCRIPT_SET_VARIABLES_SCHEMA,
     SCRIPT_ACTION_STOP: _SCRIPT_STOP_SCHEMA,
-    SCRIPT_ACTION_VARIABLES: _SCRIPT_SET_SCHEMA,
+    SCRIPT_ACTION_VARIABLES: _SCRIPT_VARIABLES_SCHEMA,
     SCRIPT_ACTION_WAIT_FOR_TRIGGER: _SCRIPT_WAIT_FOR_TRIGGER_SCHEMA,
     SCRIPT_ACTION_WAIT_TEMPLATE: _SCRIPT_WAIT_TEMPLATE_SCHEMA,
 }