evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add optional limits to compensation sensors (#85886)

Browse source 
Co-authored-by: Tom Harris <tomharris@harrisnj.net>
Co-authored-by: J. Nick Koston <nick@koston.org>
This commit is contained in:
Petro31 2023-06-26 15:36:59 -04:00 • committed by GitHub
parent b71e0302d6
commit f9707cc87b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 86 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
homeassistant/components/compensation/sensor.py
View file

@ -10,6 +10,8 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT,
CONF_ATTRIBUTE,
CONF_MAXIMUM,
CONF_MINIMUM,
CONF_SOURCE,
CONF_UNIQUE_ID,
CONF_UNIT_OF_MEASUREMENT,
@ -64,6 +66,8 @@ async def async_setup_platform(
conf[CONF_PRECISION],
conf[CONF_POLYNOMIAL],
conf.get(CONF_UNIT_OF_MEASUREMENT),
conf[CONF_MINIMUM],
conf[CONF_MAXIMUM],
)
]
)
@ -83,6 +87,8 @@ class CompensationSensor(SensorEntity):
precision: int,
polynomial: np.poly1d,
unit_of_measurement: str | None,
minimum: tuple[float, float] | None,
maximum: tuple[float, float] | None,
) -> None:
"""Initialize the Compensation sensor."""
self._source_entity_id = source
@ -93,6 +99,8 @@ class CompensationSensor(SensorEntity):
self._coefficients = polynomial.coefficients.tolist()
self._attr_unique_id = unique_id
self._attr_name = name
self._minimum = minimum
self._maximum = maximum
async def async_added_to_hass(self) -> None:
"""Handle added to Hass."""
@ -132,7 +140,14 @@ class CompensationSensor(SensorEntity):
else:
value = None if new_state.state == STATE_UNKNOWN else new_state.state
try:
self._attr_native_value = round(self._poly(float(value)), self._precision)
x_value = float(value)
if self._minimum is not None and x_value <= self._minimum[0]:
y_value = self._minimum[1]
elif self._maximum is not None and x_value >= self._maximum[0]:
y_value = self._maximum[1]
else:
y_value = self._poly(x_value)
self._attr_native_value = round(y_value, self._precision)
except (ValueError, TypeError):
self._attr_native_value = None