From 9d82ce8ab49a43e943cb579502a53945ee37eb44 Mon Sep 17 00:00:00 2001
From: Philip Allgaier <mail@spacegaier.de>
Date: Sat, 30 Jan 2021 21:32:25 +0100
Subject: [PATCH] Add missing device_classes to sensor (#8288)

---
 src/util/hass-attributes-util.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/util/hass-attributes-util.ts b/src/util/hass-attributes-util.ts
index f26384786a..1b9c521e34 100644
--- a/src/util/hass-attributes-util.ts
+++ b/src/util/hass-attributes-util.ts
@@ -40,13 +40,17 @@ const hassAttributeUtil = {
     humidifier: ["dehumidifier", "humidifier"],
     sensor: [
       "battery",
+      "current",
+      "energy",
       "humidity",
       "illuminance",
-      "temperature",
-      "pressure",
       "power",
+      "power_factor",
+      "pressure",
       "signal_strength",
+      "temperature",
       "timestamp",
+      "voltage",
     ],
     switch: ["switch", "outlet"],
   },