48 lines
1.5 KiB
Python
48 lines
1.5 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 config_validation as cv, label_registry as lr
|
|
|
|
from ....services import AbstractSpookAdminService
|
|
from ..labels import SUPPORTED_LABEL_THEME_COLORS
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookAdminService):
|
|
"""Home Assistant service to create labels on the fly."""
|
|
|
|
domain = DOMAIN
|
|
service = "create_label"
|
|
schema = {
|
|
vol.Required("name"): cv.string,
|
|
vol.Optional("color"): vol.Any(
|
|
cv.color_hex, vol.In(SUPPORTED_LABEL_THEME_COLORS)
|
|
),
|
|
vol.Optional("description"): cv.string,
|
|
vol.Optional("icon"): cv.icon,
|
|
}
|
|
|
|
async def async_handle_service(self, call: ServiceCall) -> None:
|
|
"""Handle the service call."""
|
|
label_registry = lr.async_get(self.hass)
|
|
try:
|
|
label_registry.async_create(
|
|
name=call.data["name"],
|
|
color=call.data.get("color"),
|
|
description=call.data.get("description"),
|
|
icon=call.data.get("icon"),
|
|
)
|
|
except ValueError as err:
|
|
# A name another label already has. Left alone this comes back as
|
|
# an unknown error with the registry's own wording.
|
|
raise HomeAssistantError(str(err)) from err
|