Add unit of measurement for battery sensor (#18)

* Add unit of measurement for battery sensor * Update sensor.py
cgarwood · Oct 13, 2020 · 81fcb23 · 81fcb23
1 parent d52efc5
commit 81fcb23
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/custom_components/fullykiosk/sensor.py b/custom_components/fullykiosk/sensor.py
@@ -1,7 +1,7 @@
 """Fully Kiosk Browser sensor."""
 import logging
 
-from homeassistant.const import DEVICE_CLASS_BATTERY
+from homeassistant.const import DEVICE_CLASS_BATTERY, PERCENTAGE
 from homeassistant.helpers.entity import Entity
 
 from .const import DOMAIN, COORDINATOR
@@ -53,6 +53,12 @@ def device_class(self):
             return DEVICE_CLASS_BATTERY
         return None
 
+    @property
+    def unit_of_measurement(self):
+        if self._sensor == "batteryLevel":
+            return PERCENTAGE
+        return None
+
     @property
     def device_info(self):
         return {