38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from homeassistant.components.number import DOMAIN, NumberEntity
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
from ....const import DOMAIN as SPOOK_DOMAIN
|
|
from ....services import AbstractSpookEntityComponentService
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookEntityComponentService[NumberEntity]):
|
|
"""Number entity service, set the max value."""
|
|
|
|
domain = DOMAIN
|
|
service = "max"
|
|
|
|
async def async_handle_service(
|
|
self,
|
|
entity: NumberEntity,
|
|
call: ServiceCall, # noqa: ARG002
|
|
) -> None:
|
|
"""Handle the service call."""
|
|
if entity.max_value is None:
|
|
raise HomeAssistantError(
|
|
translation_domain=SPOOK_DOMAIN,
|
|
translation_key="no_max_value",
|
|
translation_placeholders={
|
|
"entity_id": entity.entity_id,
|
|
},
|
|
)
|
|
await entity.async_set_native_value(entity.native_max_value)
|