40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.homeassistant import DOMAIN
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import area_registry as ar, config_validation as cv
|
|
|
|
from ....services import AbstractSpookAdminService
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookAdminService):
|
|
"""Home Assistant service to set the aliases of an area."""
|
|
|
|
domain = DOMAIN
|
|
service = "set_area_aliases"
|
|
schema = {
|
|
vol.Required("area_id"): cv.string,
|
|
vol.Required("aliases"): vol.All(cv.ensure_list, [cv.string]),
|
|
}
|
|
|
|
async def async_handle_service(self, call: ServiceCall) -> None:
|
|
"""Handle the service call."""
|
|
area_registry = ar.async_get(self.hass)
|
|
if not area_registry.async_get_area(call.data["area_id"]):
|
|
msg = f"Area {call.data['area_id']} not found"
|
|
raise HomeAssistantError(msg)
|
|
|
|
area_registry.async_update(
|
|
call.data["area_id"],
|
|
aliases=set(call.data["aliases"]),
|
|
)
|