Now Number includes device_class and unit_of_measurement

alexbarcelo · Jul 3, 2023 · c363dfc · c363dfc
1 parent 4913f2e
commit c363dfc
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/examples/simple_number.py b/examples/simple_number.py
@@ -21,6 +21,8 @@
 class SimpleNumber(OptimisticNumber):
     name = "Awesomest Number"
     short_id = "moreawesome"
+    device_class = "data_size"
+    unit_of_measurement = "MiB"
 
     def callback(self, state: float):
         super().callback(state)

diff --git a/src/ham/__init__.py b/src/ham/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.5.5"
+__version__ = "0.5.6"
 
 from .manager import MqttManager
 

diff --git a/src/ham/number.py b/src/ham/number.py
@@ -9,8 +9,11 @@ class Number(Thing):
     min: float
     max: float
     step: float
+    device_class: str
+    unit_of_measurement: str
 
-    config_fields = ["min", "max", "step"]
+    config_fields = ["min", "max", "step",
+                     "device_class", "unit_of_measurement"]
 
     _state: float = 1.0