mirror of
https://github.com/home-assistant/core.git
synced 2026-09-24 23:41:48 -05:00
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
"""Number platform for Prana integration."""
|
|
|
|
from collections.abc import Callable
|
|
from dataclasses import dataclass
|
|
from enum import StrEnum
|
|
from typing import Any, override
|
|
|
|
from homeassistant.components.number import (
|
|
NumberEntity,
|
|
NumberEntityDescription,
|
|
NumberMode,
|
|
)
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .coordinator import PranaConfigEntry, PranaCoordinator
|
|
from .entity import PranaBaseEntity
|
|
|
|
PARALLEL_UPDATES = 1
|
|
|
|
|
|
class PranaNumberType(StrEnum):
|
|
"""Enumerates Prana number types exposed by the device API."""
|
|
|
|
DISPLAY_BRIGHTNESS = "display_brightness"
|
|
|
|
|
|
@dataclass(frozen=True, kw_only=True)
|
|
class PranaNumberEntityDescription(NumberEntityDescription):
|
|
"""Description of a Prana number entity."""
|
|
|
|
key: PranaNumberType
|
|
value_fn: Callable[[PranaCoordinator], float | None]
|
|
set_value_fn: Callable[[Any, float], Any]
|
|
|
|
|
|
ENTITIES: tuple[PranaNumberEntityDescription, ...] = (
|
|
PranaNumberEntityDescription(
|
|
key=PranaNumberType.DISPLAY_BRIGHTNESS,
|
|
translation_key="display_brightness",
|
|
native_min_value=0,
|
|
native_max_value=6,
|
|
native_step=1,
|
|
mode=NumberMode.SLIDER,
|
|
entity_category=EntityCategory.CONFIG,
|
|
value_fn=lambda coord: coord.data.brightness,
|
|
set_value_fn=lambda api, val: api.set_brightness(
|
|
0 if val == 0 else 2 ** (int(val) - 1)
|
|
),
|
|
),
|
|
)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: PranaConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Prana number entities from a config entry."""
|
|
async_add_entities(
|
|
PranaNumber(entry.runtime_data, entity_description)
|
|
for entity_description in ENTITIES
|
|
)
|
|
|
|
|
|
class PranaNumber(PranaBaseEntity, NumberEntity):
|
|
"""Representation of a Prana number entity."""
|
|
|
|
entity_description: PranaNumberEntityDescription
|
|
|
|
@property
|
|
@override
|
|
def native_value(self) -> float | None:
|
|
"""Return the entity value."""
|
|
return self.entity_description.value_fn(self.coordinator)
|
|
|
|
@override
|
|
async def async_set_native_value(self, value: float) -> None:
|
|
"""Set new value."""
|
|
await self.entity_description.set_value_fn(self.coordinator.api_client, value)
|
|
await self.coordinator.async_refresh()
|