From 1b64dbec67059298a68e49c0a2eaa85995987e1b Mon Sep 17 00:00:00 2001 From: ablack89 <41829275+ablack89@users.noreply.github.com> Date: Mon, 11 Apr 2022 18:37:53 +0100 Subject: [PATCH] EmonCMS - Set device and state class for more feed units (#69872) Co-authored-by: Franck Nijhof --- homeassistant/components/emoncms/sensor.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/emoncms/sensor.py b/homeassistant/components/emoncms/sensor.py index 8879743446e..af684067ec8 100644 --- a/homeassistant/components/emoncms/sensor.py +++ b/homeassistant/components/emoncms/sensor.py @@ -163,12 +163,30 @@ class EmonCmsSensor(SensorEntity): self._sensorid = sensorid self._elem = elem - if unit_of_measurement == "kWh": + if unit_of_measurement in ("kWh", "Wh"): self._attr_device_class = SensorDeviceClass.ENERGY self._attr_state_class = SensorStateClass.TOTAL_INCREASING elif unit_of_measurement == "W": self._attr_device_class = SensorDeviceClass.POWER self._attr_state_class = SensorStateClass.MEASUREMENT + elif unit_of_measurement == "V": + self._attr_device_class = SensorDeviceClass.VOLTAGE + self._attr_state_class = SensorStateClass.MEASUREMENT + elif unit_of_measurement == "A": + self._attr_device_class = SensorDeviceClass.CURRENT + self._attr_state_class = SensorStateClass.MEASUREMENT + elif unit_of_measurement == "VA": + self._attr_device_class = SensorDeviceClass.APPARENT_POWER + self._attr_state_class = SensorStateClass.MEASUREMENT + elif unit_of_measurement in ("°C", "°F", "K"): + self._attr_device_class = SensorDeviceClass.TEMPERATURE + self._attr_state_class = SensorStateClass.MEASUREMENT + elif unit_of_measurement == "Hz": + self._attr_device_class = SensorDeviceClass.FREQUENCY + self._attr_state_class = SensorStateClass.MEASUREMENT + elif unit_of_measurement == "hPa": + self._attr_device_class = SensorDeviceClass.PRESSURE + self._attr_state_class = SensorStateClass.MEASUREMENT if self._value_template is not None: self._state = self._value_template.render_with_possible_json_value(