Initial commit

custom_components/oasis_mini/number.py

@@ -0,0 +1,58 @@
+"""Oasis Mini number entity."""
+
+from __future__ import annotations
+
+from homeassistant.components.number import NumberEntity, NumberEntityDescription
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import DOMAIN
+from .coordinator import OasisMiniCoordinator
+from .entity import OasisMiniEntity
+
+
+class OasisMiniNumberEntity(OasisMiniEntity, NumberEntity):
+    """Oasis Mini number entity."""
+
+    @property
+    def native_value(self) -> str | None:
+        """Return the value reported by the number."""
+        return getattr(self.device, self.entity_description.key)
+
+    async def async_set_native_value(self, value: float) -> None:
+        """Set new value."""
+        if self.entity_description.key == "ball_speed":
+            await self.device.async_set_ball_speed(value)
+        elif self.entity_description.key == "led_speed":
+            await self.device.async_set_led(led_speed=value)
+        await self.coordinator.async_request_refresh()
+
+
+DESCRIPTORS = {
+    NumberEntityDescription(
+        key="ball_speed",
+        name="Ball speed",
+        native_max_value=800,
+        native_min_value=200,
+    ),
+    NumberEntityDescription(
+        key="led_speed",
+        name="LED speed",
+        native_max_value=90,
+        native_min_value=-90,
+    ),
+}
+
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+) -> None:
+    """Set up Oasis Mini numbers using config entry."""
+    coordinator: OasisMiniCoordinator = hass.data[DOMAIN][entry.entry_id]
+    async_add_entities(
+        [
+            OasisMiniNumberEntity(coordinator, entry, descriptor)
+            for descriptor in DESCRIPTORS
+        ]
+    )