64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# pylint: disable=duplicate-code # Mirrors remove_device_tracker by design.
|
|
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.person import DOMAIN, Person, PersonStorageCollection
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
from ....const import DOMAIN as SPOOK_DOMAIN
|
|
from ....errors import entity_not_found
|
|
from ....services import AbstractSpookAdminService
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
from homeassistant.helpers.entity_component import EntityComponent
|
|
|
|
|
|
class SpookService(AbstractSpookAdminService):
|
|
"""Home Assistant service to add a device tracker to a person."""
|
|
|
|
domain = DOMAIN
|
|
service = "add_device_tracker"
|
|
schema = {
|
|
vol.Required("entity_id"): cv.entity_domain(DOMAIN),
|
|
vol.Required("device_tracker"): vol.All(
|
|
cv.ensure_list,
|
|
[cv.entity_domain("device_tracker")],
|
|
),
|
|
}
|
|
|
|
async def async_handle_service(self, call: ServiceCall) -> None:
|
|
"""Handle the service call."""
|
|
collection: PersonStorageCollection
|
|
entity_component: EntityComponent[Person]
|
|
_, collection, entity_component = self.hass.data[DOMAIN]
|
|
|
|
if not (entity := entity_component.get_entity(call.data["entity_id"])):
|
|
raise entity_not_found(call.data["entity_id"])
|
|
|
|
# pylint: disable-next=protected-access
|
|
if not entity.editable or "id" not in entity._config: # noqa: SLF001
|
|
raise HomeAssistantError(
|
|
translation_domain=SPOOK_DOMAIN,
|
|
translation_key="person_not_editable",
|
|
translation_placeholders={
|
|
"entity_id": call.data["entity_id"],
|
|
},
|
|
)
|
|
|
|
await collection.async_update_item(
|
|
# pylint: disable-next=protected-access
|
|
entity._config["id"], # noqa: SLF001
|
|
{
|
|
"device_trackers": list(
|
|
set(entity.device_trackers + call.data["device_tracker"]),
|
|
),
|
|
},
|
|
)
|