Cast string back to datetime in Sensor Filter (#65396)
This commit is contained in:
parent
14a7ee5d0b
commit
abaf284ef2
2 changed files with 36 additions and 1 deletions
|
@ -3,7 +3,7 @@ from __future__ import annotations
|
||||||
|
|
||||||
from collections import Counter, deque
|
from collections import Counter, deque
|
||||||
from copy import copy
|
from copy import copy
|
||||||
from datetime import timedelta
|
from datetime import datetime, timedelta
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import logging
|
import logging
|
||||||
from numbers import Number
|
from numbers import Number
|
||||||
|
@ -19,6 +19,7 @@ from homeassistant.components.sensor import (
|
||||||
DEVICE_CLASSES as SENSOR_DEVICE_CLASSES,
|
DEVICE_CLASSES as SENSOR_DEVICE_CLASSES,
|
||||||
DOMAIN as SENSOR_DOMAIN,
|
DOMAIN as SENSOR_DOMAIN,
|
||||||
PLATFORM_SCHEMA,
|
PLATFORM_SCHEMA,
|
||||||
|
SensorDeviceClass,
|
||||||
SensorEntity,
|
SensorEntity,
|
||||||
)
|
)
|
||||||
from homeassistant.const import (
|
from homeassistant.const import (
|
||||||
|
@ -346,6 +347,9 @@ class SensorFilter(SensorEntity):
|
||||||
@property
|
@property
|
||||||
def native_value(self):
|
def native_value(self):
|
||||||
"""Return the state of the sensor."""
|
"""Return the state of the sensor."""
|
||||||
|
if self._device_class == SensorDeviceClass.TIMESTAMP:
|
||||||
|
return datetime.fromisoformat(self._state)
|
||||||
|
|
||||||
return self._state
|
return self._state
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|
|
@ -318,6 +318,37 @@ async def test_invalid_state(hass):
|
||||||
assert state.state == STATE_UNAVAILABLE
|
assert state.state == STATE_UNAVAILABLE
|
||||||
|
|
||||||
|
|
||||||
|
async def test_timestamp_state(hass):
|
||||||
|
"""Test if filter state is a datetime."""
|
||||||
|
config = {
|
||||||
|
"sensor": {
|
||||||
|
"platform": "filter",
|
||||||
|
"name": "test",
|
||||||
|
"entity_id": "sensor.test_monitored",
|
||||||
|
"filters": [
|
||||||
|
{"filter": "time_throttle", "window_size": "00:02"},
|
||||||
|
],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await async_init_recorder_component(hass)
|
||||||
|
|
||||||
|
with assert_setup_component(1, "sensor"):
|
||||||
|
assert await async_setup_component(hass, "sensor", config)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
hass.states.async_set(
|
||||||
|
"sensor.test_monitored",
|
||||||
|
"2022-02-01T23:04:05+00:00",
|
||||||
|
{ATTR_DEVICE_CLASS: SensorDeviceClass.TIMESTAMP},
|
||||||
|
)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
state = hass.states.get("sensor.test")
|
||||||
|
assert state.state == "2022-02-01T23:04:05+00:00"
|
||||||
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TIMESTAMP
|
||||||
|
|
||||||
|
|
||||||
async def test_outlier(values):
|
async def test_outlier(values):
|
||||||
"""Test if outlier filter works."""
|
"""Test if outlier filter works."""
|
||||||
filt = OutlierFilter(window_size=3, precision=2, entity=None, radius=4.0)
|
filt = OutlierFilter(window_size=3, precision=2, entity=None, radius=4.0)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue