Added Icons for deltaEnergy and powerEnergy

veista · Jan 6, 2022 · 73a34cd · 73a34cd
1 parent 1e11115
commit 73a34cd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/custom_components/smartthings/sensor.py b/custom_components/smartthings/sensor.py
@@ -751,4 +751,10 @@ def native_unit_of_measurement(self):
         if self.report_name == "power":
             return POWER_WATT
         return ENERGY_KILO_WATT_HOUR
+
+    @property
+    def icon(self) -> str | None:
+        if self.report_name in ("deltaEnergy", "powerEnergy"):
+            return "mdi:current-ac"
+        return None