46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
import logging
|
|
|
|
from homeassistant.components.number import NumberEntity
|
|
from homeassistant.const import UnitOfTemperature
|
|
|
|
from .core.entity import YandexCustomEntity, extract_instance
|
|
from .hass import hass_utils
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
INCLUDE_CAPABILITIES = ("devices.capabilities.range",)
|
|
|
|
UNITS = {"unit.temperature.celsius": UnitOfTemperature.CELSIUS}
|
|
|
|
|
|
async def async_setup_entry(hass, entry, async_add_entities):
|
|
entities = []
|
|
|
|
for quasar, device, config in hass_utils.incluce_devices(hass, entry):
|
|
if instances := config.get("capabilities"):
|
|
for instance in device["capabilities"]:
|
|
if instance["type"] not in INCLUDE_CAPABILITIES:
|
|
continue
|
|
if extract_instance(instance) in instances:
|
|
entities.append(YandexCustomNumber(quasar, device, instance))
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
# noinspection PyAbstractClass
|
|
class YandexCustomNumber(NumberEntity, YandexCustomEntity):
|
|
def internal_init(self, capabilities: dict, properties: dict):
|
|
if item := capabilities.get(self.instance):
|
|
if range_ := item.get("range"):
|
|
self._attr_native_max_value = range_["max"]
|
|
self._attr_native_min_value = range_["min"]
|
|
self._attr_native_step = range_["precision"]
|
|
self._attr_native_unit_of_measurement = UNITS.get(item["unit"])
|
|
|
|
def internal_update(self, capabilities: dict, properties: dict):
|
|
if self.instance in capabilities:
|
|
self._attr_native_value = capabilities[self.instance]
|
|
|
|
async def async_set_native_value(self, value: float) -> None:
|
|
await self.device_action(self.instance, value)
|