46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from homeassistant.components.input_number import (
|
|
DOMAIN,
|
|
InputNumber,
|
|
NumberStorageCollection,
|
|
)
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
from ....services import AbstractSpookEntityComponentService
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookEntityComponentService[InputNumber]):
|
|
"""Input number service to delete a helper on the fly."""
|
|
|
|
domain = DOMAIN
|
|
service = "delete"
|
|
schema = {}
|
|
|
|
async def async_handle_service(
|
|
self,
|
|
entity: InputNumber,
|
|
call: ServiceCall, # noqa: ARG002
|
|
) -> None:
|
|
"""Handle the service call."""
|
|
if not entity.editable:
|
|
message = f"This input number is not editable: {entity.entity_id}"
|
|
raise HomeAssistantError(message)
|
|
|
|
collection: NumberStorageCollection
|
|
if DOMAIN in self.hass.data:
|
|
collection = self.hass.data[DOMAIN]
|
|
else:
|
|
# Major hack to get around edge cases. 👻
|
|
collection = self.hass.data["websocket_api"]["input_number/list"][
|
|
0
|
|
].__self__.storage_collection
|
|
await collection.async_delete_item(entity.unique_id)
|