Fix typo, add device class for plugged in

cgarwood · May 19, 2020 · c7342c7 · c7342c7
1 parent 50ef0b9
commit c7342c7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/fullykiosk/binary_sensor.py b/custom_components/fullykiosk/binary_sensor.py
@@ -2,7 +2,7 @@
 import logging
 
 from homeassistant.components.binary_sensor import BinarySensorDevice
-from homeassistant.const import DEVICE_CLASS_BATTERY
+from homeassistant.const import DEVICE_CLASS_PLUG
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
 
 from .const import DOMAIN, COORDINATOR, CONTROLLER
@@ -12,7 +12,7 @@
 SENSOR_TYPES = {
     "kioskMode": "Kiosk Mode",
     "kioskLocked": "Kiosk Locked",
-    "plugged": "Pluggin In",
+    "plugged": "Plugged In",
     "isDeviceAdmin": "Device Admin",
 }
 
@@ -46,8 +46,8 @@ def is_on(self):
 
     @property
     def device_class(self):
-        if self._sensor == "batteryLevel":
-            return DEVICE_CLASS_BATTERY
+        if self._sensor == "plugged":
+            return DEVICE_CLASS_PLUG
         return None
 
     @property