From 0fdea8ec8fdc0f1ce4be4e85b9cb259900356bc2 Mon Sep 17 00:00:00 2001
From: Andre Richter <andre-richter@users.noreply.github.com>
Date: Tue, 24 Aug 2021 12:16:31 +0200
Subject: [PATCH] SMA: Add statistics support for power sensors (#54422)

---
 homeassistant/components/sma/sensor.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/homeassistant/components/sma/sensor.py b/homeassistant/components/sma/sensor.py
index 8808272ad75..f0a10a5d5e1 100644
--- a/homeassistant/components/sma/sensor.py
+++ b/homeassistant/components/sma/sensor.py
@@ -9,6 +9,7 @@ import voluptuous as vol
 
 from homeassistant.components.sensor import (
     PLATFORM_SCHEMA,
+    STATE_CLASS_MEASUREMENT,
     STATE_CLASS_TOTAL_INCREASING,
     SensorEntity,
 )
@@ -21,7 +22,9 @@ from homeassistant.const import (
     CONF_SSL,
     CONF_VERIFY_SSL,
     DEVICE_CLASS_ENERGY,
+    DEVICE_CLASS_POWER,
     ENERGY_KILO_WATT_HOUR,
+    POWER_WATT,
 )
 from homeassistant.core import HomeAssistant
 import homeassistant.helpers.config_validation as cv
@@ -166,6 +169,9 @@ class SMAsensor(CoordinatorEntity, SensorEntity):
         if self.unit_of_measurement == ENERGY_KILO_WATT_HOUR:
             self._attr_state_class = STATE_CLASS_TOTAL_INCREASING
             self._attr_device_class = DEVICE_CLASS_ENERGY
+        if self.unit_of_measurement == POWER_WATT:
+            self._attr_state_class = STATE_CLASS_MEASUREMENT
+            self._attr_device_class = DEVICE_CLASS_POWER
 
         # Set sensor enabled to False.
         # Will be enabled by async_added_to_hass if actually used.