mirror of https://github.com/home-assistant/core
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
"""Intents for the todo integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import intent
|
|
|
|
from . import TodoItem, TodoItemStatus, TodoListEntity
|
|
from .const import DATA_COMPONENT, DOMAIN
|
|
|
|
INTENT_LIST_ADD_ITEM = "HassListAddItem"
|
|
|
|
|
|
async def async_setup_intents(hass: HomeAssistant) -> None:
|
|
"""Set up the todo intents."""
|
|
intent.async_register(hass, ListAddItemIntent())
|
|
|
|
|
|
class ListAddItemIntent(intent.IntentHandler):
|
|
"""Handle ListAddItem intents."""
|
|
|
|
intent_type = INTENT_LIST_ADD_ITEM
|
|
description = "Add item to a todo list"
|
|
slot_schema = {
|
|
vol.Required("item"): intent.non_empty_string,
|
|
vol.Required("name"): intent.non_empty_string,
|
|
}
|
|
platforms = {DOMAIN}
|
|
|
|
async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse:
|
|
"""Handle the intent."""
|
|
hass = intent_obj.hass
|
|
|
|
slots = self.async_validate_slots(intent_obj.slots)
|
|
item = slots["item"]["value"].strip()
|
|
list_name = slots["name"]["value"]
|
|
|
|
target_list: TodoListEntity | None = None
|
|
|
|
# Find matching list
|
|
match_constraints = intent.MatchTargetsConstraints(
|
|
name=list_name, domains=[DOMAIN], assistant=intent_obj.assistant
|
|
)
|
|
match_result = intent.async_match_targets(hass, match_constraints)
|
|
if not match_result.is_match:
|
|
raise intent.MatchFailedError(
|
|
result=match_result, constraints=match_constraints
|
|
)
|
|
|
|
target_list = hass.data[DATA_COMPONENT].get_entity(
|
|
match_result.states[0].entity_id
|
|
)
|
|
if target_list is None:
|
|
raise intent.IntentHandleError(f"No to-do list: {list_name}")
|
|
|
|
# Add to list
|
|
await target_list.async_create_todo_item(
|
|
TodoItem(summary=item, status=TodoItemStatus.NEEDS_ACTION)
|
|
)
|
|
|
|
response = intent_obj.create_response()
|
|
response.response_type = intent.IntentResponseType.ACTION_DONE
|
|
response.async_set_results(
|
|
[
|
|
intent.IntentResponseTarget(
|
|
type=intent.IntentResponseTargetType.ENTITY,
|
|
name=list_name,
|
|
id=match_result.states[0].entity_id,
|
|
)
|
|
]
|
|
)
|
|
return response
|