57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import voluptuous as vol
|
|
|
|
# Home Assistant keeps the exposure settings, and the list of assistants that
|
|
# have them, in this module. There is no public helper for either.
|
|
from homeassistant.components.homeassistant.exposed_entities import (
|
|
KNOWN_ASSISTANTS,
|
|
async_expose_entity,
|
|
)
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import config_validation as cv, entity_registry as er
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import HomeAssistant, ServiceCall
|
|
|
|
CONF_ASSISTANTS = "assistants"
|
|
|
|
EXPOSURE_SERVICE_SCHEMA = {
|
|
vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
|
|
vol.Required(CONF_ASSISTANTS): vol.All(cv.ensure_list, [vol.In(KNOWN_ASSISTANTS)]),
|
|
}
|
|
|
|
|
|
def async_set_voice_assistant_exposure(
|
|
hass: HomeAssistant,
|
|
call: ServiceCall,
|
|
*,
|
|
should_expose: bool,
|
|
) -> None:
|
|
"""Set whether entities are exposed to the given voice assistants.
|
|
|
|
Shared by the expose and unexpose actions, which differ only in what they
|
|
set it to.
|
|
"""
|
|
entity_registry = er.async_get(hass)
|
|
|
|
# Checked up front, so an unknown entity halfway down the list does not
|
|
# leave the ones before it already changed, with nothing saying which.
|
|
entity_ids = call.data[ATTR_ENTITY_ID]
|
|
for entity_id in entity_ids:
|
|
if (
|
|
hass.states.get(entity_id) is None
|
|
and entity_registry.async_get(entity_id) is None
|
|
):
|
|
msg = f"Unknown entity: {entity_id}"
|
|
raise HomeAssistantError(msg)
|
|
|
|
for entity_id in entity_ids:
|
|
for assistant in call.data[CONF_ASSISTANTS]:
|
|
async_expose_entity(hass, assistant, entity_id, should_expose=should_expose)
|