device_class & state_class first try

Author: yozik04

nibe_mqtt/mqtt.py

@@ -106,7 +106,17 @@ def publish_discovery(self, coil: Coil, device_info: dict):
         uom = coil.unit
         if uom is not None:
             config["unit_of_measurement"] = uom
-
+            if uom == "°C":
+                config["device_class"] = "temperature"
+                config["state_class"] = "measurement"
+            elif uom in ["h", "min"]:
+                config["device_class"] = "duration"
+            elif uom in ["kW", "W"]:
+                config["device_class"] = "power"
+            elif uom == "kWh":
+                config["device_class"] = "energy"
+            elif uom == "Hz":
+                config["device_class"] = "frequency"
         if coil.is_boolean:
             if coil.is_writable:  # switch
                 component = "switch"