From fd2e640c7418d6b76e12ef006e1e7147e4b1cc53 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Hjelseth=20H=C3=B8yer?= Date: Thu, 20 May 2021 18:23:00 +0200 Subject: [PATCH] Use sensor constants in recorder (#50906) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Daniel Hjelseth Høyer --- homeassistant/components/sensor/recorder.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/sensor/recorder.py b/homeassistant/components/sensor/recorder.py index 34e373dae50..e3da50d9738 100644 --- a/homeassistant/components/sensor/recorder.py +++ b/homeassistant/components/sensor/recorder.py @@ -6,14 +6,22 @@ import itertools from statistics import fmean from homeassistant.components.recorder import history, statistics -from homeassistant.components.sensor import ATTR_STATE_CLASS, STATE_CLASS_MEASUREMENT +from homeassistant.components.sensor import ( + ATTR_STATE_CLASS, + DEVICE_CLASS_ENERGY, + DEVICE_CLASS_TEMPERATURE, + STATE_CLASS_MEASUREMENT, +) from homeassistant.const import ATTR_DEVICE_CLASS from homeassistant.core import HomeAssistant import homeassistant.util.dt as dt_util from . import DOMAIN -DEVICE_CLASS_STATISTICS = {"temperature": {"mean", "min", "max"}, "energy": {"sum"}} +DEVICE_CLASS_STATISTICS = { + DEVICE_CLASS_TEMPERATURE: {"mean", "min", "max"}, + DEVICE_CLASS_ENERGY: {"sum"}, +} def _get_entities(hass: HomeAssistant) -> list[tuple[str, str]]: