jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 22:57:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add type hints for intent_script integration (#99393)

Browse Source 
* Add type hints for intent_script integration

* Correct 2nd typo

* omit total=False as all options are set
This commit is contained in:
Jan Bouwhuis 2023-09-25 23:01:53 +02:00 committed by GitHub
parent 60b8775f4a
commit 90bf20c6f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
homeassistant/components/intent_script/__init__.py
View File

@ -2,6 +2,7 @@
from __future__ import annotations from __future__ import annotations
import logging import logging
from typing import TypedDict
import voluptuous as vol import voluptuous as vol
@ -62,7 +63,7 @@ CONFIG_SCHEMA = vol.Schema(
) )
async def async_reload(hass: HomeAssistant, servie_call: ServiceCall) -> None: async def async_reload(hass: HomeAssistant, service_call: ServiceCall) -> None:
"""Handle start Intent Script service call.""" """Handle start Intent Script service call."""
new_config = await async_integration_yaml_config(hass, DOMAIN) new_config = await async_integration_yaml_config(hass, DOMAIN)
existing_intents = hass.data[DOMAIN] existing_intents = hass.data[DOMAIN]
@ -79,7 +80,7 @@ async def async_reload(hass: HomeAssistant, servie_call: ServiceCall) -> None:
async_load_intents(hass, new_intents) async_load_intents(hass, new_intents)
def async_load_intents(hass: HomeAssistant, intents: dict): def async_load_intents(hass: HomeAssistant, intents: dict[str, ConfigType]) -> None:
"""Load YAML intents into the intent system.""" """Load YAML intents into the intent system."""
template.attach(hass, intents) template.attach(hass, intents)
hass.data[DOMAIN] = intents hass.data[DOMAIN] = intents
@ -98,8 +99,8 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
async_load_intents(hass, intents) async_load_intents(hass, intents)
async def _handle_reload(servie_call: ServiceCall) -> None: async def _handle_reload(service_call: ServiceCall) -> None:
return await async_reload(hass, servie_call) return await async_reload(hass, service_call)
service.async_register_admin_service( service.async_register_admin_service(
hass, hass,
@ -111,22 +112,41 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return True return True
class _IntentSpeechRepromptData(TypedDict):
"""Intent config data type for speech or reprompt info."""
content: template.Template
title: template.Template
text: template.Template
type: str
class _IntentCardData(TypedDict):
"""Intent config data type for card info."""
type: str
title: template.Template
content: template.Template
class ScriptIntentHandler(intent.IntentHandler): class ScriptIntentHandler(intent.IntentHandler):
"""Respond to an intent with a script.""" """Respond to an intent with a script."""
def __init__(self, intent_type, config): def __init__(self, intent_type: str, config: ConfigType) -> None:
"""Initialize the script intent handler.""" """Initialize the script intent handler."""
self.intent_type = intent_type self.intent_type = intent_type
self.config = config self.config = config
async def async_handle(self, intent_obj: intent.Intent): async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse:
"""Handle the intent.""" """Handle the intent."""
speech = self.config.get(CONF_SPEECH) speech: _IntentSpeechRepromptData | None = self.config.get(CONF_SPEECH)
reprompt = self.config.get(CONF_REPROMPT) reprompt: _IntentSpeechRepromptData | None = self.config.get(CONF_REPROMPT)
card = self.config.get(CONF_CARD) card: _IntentCardData | None = self.config.get(CONF_CARD)
action = self.config.get(CONF_ACTION) action: script.Script | None = self.config.get(CONF_ACTION)
is_async_action = self.config.get(CONF_ASYNC_ACTION) is_async_action: bool = self.config[CONF_ASYNC_ACTION]
slots = {key: value["value"] for key, value in intent_obj.slots.items()} slots: dict[str, str] = {
key: value["value"] for key, value in intent_obj.slots.items()
}
_LOGGER.debug( _LOGGER.debug(
"Intent named %s received with slots: %s", "Intent named %s received with slots: %s",
@ -150,23 +170,23 @@ class ScriptIntentHandler(intent.IntentHandler):
if speech is not None: if speech is not None:
response.async_set_speech( response.async_set_speech(
speech[CONF_TEXT].async_render(slots, parse_result=False), speech["text"].async_render(slots, parse_result=False),
speech[CONF_TYPE], speech["type"],
) )
if reprompt is not None: if reprompt is not None:
text_reprompt = reprompt[CONF_TEXT].async_render(slots, parse_result=False) text_reprompt = reprompt["text"].async_render(slots, parse_result=False)
if text_reprompt: if text_reprompt:
response.async_set_reprompt( response.async_set_reprompt(
text_reprompt, text_reprompt,
reprompt[CONF_TYPE], reprompt["type"],
) )
if card is not None: if card is not None:
response.async_set_card( response.async_set_card(
card[CONF_TITLE].async_render(slots, parse_result=False), card["title"].async_render(slots, parse_result=False),
card[CONF_CONTENT].async_render(slots, parse_result=False), card["content"].async_render(slots, parse_result=False),
card[CONF_TYPE], card["type"],
) )
return response return response