53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import math
|
|
from typing import TYPE_CHECKING
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.number import DOMAIN, NumberEntity
|
|
|
|
from ....services import AbstractSpookEntityComponentService
|
|
from . import native_value_as_float
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookEntityComponentService[NumberEntity]):
|
|
"""Number entity service, increase value by a single step."""
|
|
|
|
domain = DOMAIN
|
|
service = "increment"
|
|
schema = {vol.Optional("amount"): vol.Coerce(float)}
|
|
|
|
async def async_handle_service(
|
|
self,
|
|
entity: NumberEntity,
|
|
call: ServiceCall,
|
|
) -> None:
|
|
"""Handle the service call."""
|
|
step = entity.step or 1
|
|
amount = call.data.get("amount", step)
|
|
remainder = amount % step
|
|
# Float modulo wraps around just short of the step, 0.3 % 0.1 is
|
|
# 0.09999999999999998, so a remainder against either end is a clean multiple.
|
|
if not (
|
|
math.isclose(remainder, 0, rel_tol=0, abs_tol=1e-9)
|
|
or math.isclose(remainder, step, rel_tol=0, abs_tol=1e-9)
|
|
):
|
|
msg = (
|
|
f"Amount {amount} not valid for {entity.entity_id}, "
|
|
f"it needs to be a multiple of {step}"
|
|
)
|
|
raise ValueError(msg)
|
|
|
|
value = native_value_as_float(entity) + amount
|
|
|
|
if entity.max_value is not None:
|
|
value = min(value, entity.max_value)
|
|
|
|
await entity.async_set_native_value(value)
|