Fix update button (#10)

* Fix update button
fmtr · May 25, 2022 · ef7fcd7 · ef7fcd7
1 parent f956aa2
commit ef7fcd7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/amniotic/mqtt/control.py b/amniotic/mqtt/control.py
@@ -422,7 +422,7 @@ class ButtonUpdate(Entity):
 
     """
     HA_PLATFORM = 'button'
-    NAME = 'Update Check'
+    NAME = 'Update'
 
     def get_value(self) -> Any:
         pass

diff --git a/amniotic/mqtt/loop.py b/amniotic/mqtt/loop.py
@@ -118,6 +118,7 @@ def entity_classes(self):
         from amniotic.mqtt import control, sensor
         entity_classes = [
             control.SelectTheme, control.VolumeMaster, control.VolumeTheme, control.ToggleTheme, control.DeviceTheme, control.ButtonUpdateCheck,
+            control.ButtonUpdate,
             sensor.Title, sensor.Album, sensor.Date, sensor.By, sensor.Duration, sensor.Elapsed, sensor.UpdateStatus
         ]
         return entity_classes

diff --git a/amniotic/mqtt/sensor.py b/amniotic/mqtt/sensor.py
@@ -129,7 +129,7 @@ class Elapsed(Duration):
 class UpdateStatus(Sensor):
     NAME = 'Update Status'
     ICON_SUFFIX = 'semantic-web'
-    message = None
+    message = Sensor.NA_VALUE
 
     def set_value(self, value):
         """