Add Ke of PID at restore and to set_pid_gain service.

Author: ScratMan

custom_components/smart_thermostat/climate.py

@@ -417,8 +417,9 @@ def _async_startup(event):
             if old_state.attributes.get('Kd') is not None and self._pidController is not None:
                 self._kd = float(old_state.attributes.get('Kd'))
                 self._pidController.set_pid_param(kd=self._kd)
-            if old_state.attributes.get('Ke') is not None:
+            if old_state.attributes.get('Ke') is not None and self._pidController is not None:
                 self._ke = float(old_state.attributes.get('Ke'))
+                self._pidController.set_pid_param(ke=self._ke)
             if old_state.attributes.get('pid_mode') is not None and self._pidController is not None:
                 self._pidController.mode = old_state.attributes.get('pid_mode')
 
@@ -668,7 +669,7 @@ async def async_set_pid(self, **kwargs):
             self._kd = float(kd)
         if ke is not None:
             self._ke = float(ke)
-        self._pidController.set_pid_param(self._kp, self._ki, self._kd)
+        self._pidController.set_pid_param(self._kp, self._ki, self._kd, self._ke)
         await self._async_control_heating(calc_pid=True)
 
     async def async_set_pid_mode(self, **kwargs):

custom_components/smart_thermostat/pid_controller/__init__.py

@@ -89,14 +89,16 @@ def integral(self, i):
         self._integral = i
         self.I = i
 
-    def set_pid_param(self, kp=None, ki=None, kd=None):
+    def set_pid_param(self, kp=None, ki=None, kd=None, ke=None):
         """Set PID parameters."""
         if kp is not None and isinstance(kp, (int, float)):
             self._Kp = kp
         if ki is not None and isinstance(ki, (int, float)):
             self._Ki = ki
         if kd is not None and isinstance(kd, (int, float)):
             self._Kd = kd
+        if ke is not None and isinstance(ke, (int, float)):
+            self._Ke = ke
 
     def clear_samples(self):
         """Clear the samples values and timestamp to restart PID from clean state after