[models] added control functions

pyheatpump/models/heatpump.py

@@ -40,3 +40,15 @@ class Heatpump:
         for label, var_type in self.var_types.items():
             r[label] = var_type.get_variables_values_since(self.last_update)
         return r
+
+    def control(self, data):
+        if data['mac_address'] != self.mac_address:
+            logger.warn(f'Orders are not for the current heatpump ({self.mac_address} != {data["mac_address"]})')
+            raise Exception('Wrong mac_address')
+
+        for label, var_type in self.var_types.items():
+            if var_type.label not in data.keys():
+                continue
+
+            var_type.control(data[var_type.label])
+        return True

pyheatpump/models/variable.py

@@ -73,7 +73,7 @@ class Variable(RowClass):
 
     @staticmethod
     def getall_values_of_type_since(type: VariableType, since: int) -> dict:
-        floatcast = lambda x: round(float(x)/10, 2)
+        floatcast = type
         cast_fct = floatcast if type.slabel == 'F' else lambda x: x
 
         offset = 10000 if (type.slabel == 'D') else 0

pyheatpump/models/variable_type.py

@@ -63,6 +63,38 @@ class VariableType(RowClass):
         from .variable import Variable
         return Variable.getall_values_of_type_since(self, since)
 
+    def control(self, data):
+        for s_address, s_value in data.items():
+            address = int(s_address)
+
+            # All values are stored in strings that represent floats
+            # We have to convert them
+            f_value = float(s_value)
+
+            if self.slabel == 'A':
+                f_value = f_value * 10
+
+            # Value is stored as integers in database
+            value = int(f_value)
+
+            if address not in range(self.start_address, self.end_address + 1):
+                continue
+
+            if not Variable(self, address).exists():
+                continue
+
+            new_var_value = VariableValue({
+                'type': self,
+                'address': address,
+                'value': value})
+
+            old_var_value = VariableValue.get(
+                self, address)
+
+            if old_var_value.equals(new_var_value):
+                continue
+
+            new_var_value.set()
 
     @staticmethod
     def getall():

pyheatpump/models/variable_value.py

@@ -5,6 +5,7 @@ from pprint import pprint
 
 from .variable_type import VariableType
 from pyheatpump.logger import logger_init
+from pyheatpump.modbus import write_coil, write_holding_register
 
 logger = logger_init()
 
@@ -56,6 +57,17 @@ class VariableValue(RowClass):
     def get_value(self):
         return self.type.cast()(self.value)
 
+    def equals(self, var_value):
+        return self.get_value() == var_value.get_value()
+
+    def set(self):
+        if self.type == 'D':
+            write_coil(self)
+        else:
+            write_holding_register(self)
+
+        return self.insert()
+
     @staticmethod
     def get(type, address, time=datetime.now()):
         try: