Marc Billow
daf7e3787f
Adds [tool.ruff] and [tool.ty] config to pyproject.toml with a curated ruff rule set (E, F, W, I, UP, B, C4, SIM, RUF, ASYNC, LOG, G, PIE, RET, PERF, N), pins ruff/ty in requirements-dev.txt, reformats the whole tree with `ruff format`, and fixes the pre-existing lint and type-check debt those tools surfaced so both run clean. Production-code type fixes include: HA's ConfigFlowResult vs. the generic FlowResult in config_flow.py, narrowing BoundEntity.desc to its platform-specific subclass (SelectDesc/NumberDesc/SensorDesc/etc.) via cast() instead of an unchecked annotation, converting HA device_class strings to their proper enum types, a resolve_registry callback typed as `object` instead of `DeviceRegistry | None`, and a couple of other narrow correctness fixes (CA key type validation, an index-out-of-bounds false positive from an empty-tuple fallback, a bool/dict argument swap). Test-file fixes are mechanical: narrowing SamsungEntityDescription to the correct subclass via isinstance()/cast() before accessing subclass-only fields, and asserting Optional write_fn/unit_fn fields are set before calling them.
101 lines
3.7 KiB
Python
101 lines
3.7 KiB
Python
"""Number platform for Local Things."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import cast
|
|
|
|
from homeassistant.components.number import NumberDeviceClass, NumberEntity, NumberMode
|
|
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 LocalThingsCoordinator
|
|
from .entity import LocalThingsEntity, _is_included
|
|
from .registry.entities import NumberDesc
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
coordinator: LocalThingsCoordinator = hass.data[DOMAIN][entry.entry_id]
|
|
async_add_entities(
|
|
LocalThingsNumber(coordinator, b)
|
|
for b in coordinator.bound
|
|
if isinstance(b.desc, NumberDesc) and _is_included(b, coordinator)
|
|
)
|
|
|
|
|
|
class LocalThingsNumber(LocalThingsEntity, NumberEntity):
|
|
_attr_mode = NumberMode.SLIDER
|
|
|
|
def __init__(self, coordinator: LocalThingsCoordinator, bound) -> None:
|
|
super().__init__(coordinator, bound)
|
|
desc = cast(NumberDesc, bound.desc)
|
|
self._attr_native_unit_of_measurement = desc.unit
|
|
self._attr_device_class = (
|
|
NumberDeviceClass(desc.device_class) if desc.device_class else None
|
|
)
|
|
if desc.native_min is not None:
|
|
self._attr_native_min_value = desc.native_min
|
|
if desc.native_max is not None:
|
|
self._attr_native_max_value = desc.native_max
|
|
if desc.step is not None:
|
|
self._attr_native_step = desc.step
|
|
|
|
@property
|
|
def native_unit_of_measurement(self):
|
|
desc = cast(NumberDesc, self._bound.desc)
|
|
if desc.unit_fn is not None:
|
|
return desc.unit_fn(self.coordinator.resource(self._bound.href))
|
|
return self._attr_native_unit_of_measurement
|
|
|
|
def _range_from_resource(self) -> list | None:
|
|
desc = cast(NumberDesc, self._bound.desc)
|
|
if not desc.range_field:
|
|
return None
|
|
r = self.coordinator.resource(self._bound.href).get(desc.range_field)
|
|
return r if (isinstance(r, (list, tuple)) and len(r) == 2) else None
|
|
|
|
@property
|
|
def native_min_value(self) -> float:
|
|
desc = cast(NumberDesc, self._bound.desc)
|
|
if desc.native_min_fn is not None:
|
|
return desc.native_min_fn(self.coordinator.resource(self._bound.href))
|
|
r = self._range_from_resource()
|
|
if r is not None:
|
|
return float(r[0])
|
|
if hasattr(self, "_attr_native_min_value"):
|
|
return self._attr_native_min_value
|
|
return super().native_min_value
|
|
|
|
@property
|
|
def native_max_value(self) -> float:
|
|
desc = cast(NumberDesc, self._bound.desc)
|
|
if desc.native_max_fn is not None:
|
|
return desc.native_max_fn(self.coordinator.resource(self._bound.href))
|
|
r = self._range_from_resource()
|
|
if r is not None:
|
|
return float(r[1])
|
|
if hasattr(self, "_attr_native_max_value"):
|
|
return self._attr_native_max_value
|
|
return super().native_max_value
|
|
|
|
@property
|
|
def native_step(self) -> float | None:
|
|
desc = cast(NumberDesc, self._bound.desc)
|
|
if desc.step_fn is not None:
|
|
return desc.step_fn(self.coordinator.resource(self._bound.href))
|
|
if hasattr(self, "_attr_native_step"):
|
|
return self._attr_native_step
|
|
return super().native_step
|
|
|
|
@property
|
|
def native_value(self):
|
|
return (self.coordinator.data or {}).get(self._state_key)
|
|
|
|
async def async_set_native_value(self, value: float) -> None:
|
|
await self.coordinator.async_send_command(self._bound, value)
|