Refactored DeviceValue into a separate class for each device type.

Author: epdevlab

inelsmqtt/devices/__init__.py

@@ -5,7 +5,7 @@
 
 from typing import Any, Callable
 
-from inelsmqtt.util import DeviceValue
+from inelsmqtt.utils.core import DeviceValue
 from inelsmqtt import InelsMqtt
 from inelsmqtt.const import (
     DEVICE_TYPE_DICT,
@@ -235,6 +235,7 @@ def last_values(self) -> DeviceValue:
         dev_value = DeviceValue(
             self.__device_type,
             self.__inels_type,
+            self.__device_class,
             inels_value=(val.decode() if val is not None else None),
         )
         return dev_value
@@ -254,6 +255,7 @@ def __get_value(self, val: Any) -> DeviceValue:
         dev_value = DeviceValue(
             self.__device_type,
             self.__inels_type,
+            self.__device_class,
             inels_value=(val.decode() if val is not None else None),
             last_value=self.last_values
         )
@@ -284,6 +286,7 @@ def set_ha_value(self, value: Any) -> bool:
         dev = DeviceValue(
             self.__device_type,
             self.__inels_type,
+            self.__device_class,
             ha_value=value,
             last_value=self.__state,
         )