Updated apps

custom_components/spook/ectoplasms/input_number/services/increment.py

@@ -0,0 +1,50 @@
+"""Spook - Your homie."""
+
+from __future__ import annotations
+
+import math
+from typing import TYPE_CHECKING
+
+import voluptuous as vol
+
+from homeassistant.components.input_number import DOMAIN, InputNumber
+
+from ....services import AbstractSpookEntityComponentService, ReplaceExistingService
+
+if TYPE_CHECKING:
+    from homeassistant.core import ServiceCall
+
+
+class SpookService(
+    AbstractSpookEntityComponentService[InputNumber], ReplaceExistingService
+):
+    """Input number entity service, increase value by a single step.
+
+    It override the built-in increment service to allow for a custom amount.
+    """
+
+    domain = DOMAIN
+    service = "increment"
+    schema = {vol.Optional("amount"): vol.Coerce(float)}
+
+    async def async_handle_service(
+        self,
+        entity: InputNumber,
+        call: ServiceCall,
+    ) -> None:
+        """Handle the service call."""
+        # pylint: disable=protected-access
+        amount = call.data.get("amount", entity._step)  # noqa: SLF001
+        if not math.isclose(amount % entity._step, 0, abs_tol=1e-9):  # noqa: SLF001
+            msg = (
+                f"Amount {amount} not valid for {entity.entity_id}, "
+                f"it needs to be a multiple of {entity._step}",  # noqa: SLF001
+            )
+            raise ValueError(msg)
+
+        await entity.async_set_value(
+            min(
+                entity._current_value + amount,  # noqa: SLF001
+                entity._maximum,  # noqa: SLF001
+            ),
+        )