Added pump management task for ePBR (#30)

Author: xtrojak

custom/tasks/PBR.py

@@ -301,10 +301,31 @@ def __init__(self, config):
         }
 
 
-class ePBRGeneralPump(PBRGeneralPump):
+class ePBRpump(PBRGeneralPump):
     def __init__(self, config):
-        super(ePBRGeneralPump, self).__init__(config)
+        super(ePBRpump, self).__init__(config)
 
-        # od should be available somehow
-        # because is being used for OD bounds
-        # so we need to use it also for decision to keep on pump
+    def stabilize(self, od):
+        if self.is_od_value_too_high(od):
+            self.is_pump_on = True
+        if self.is_od_value_too_low(od):
+            self.is_pump_on = False
+        if self.is_pump_on:
+            self.change_pump_state(True)
+
+    def change_pump_state(self, state: bool):
+        for try_n in range(5):
+            command = Command(self.device_id, "8", [self.pump_id, 1], self.task_id)
+            self.device.post_command(command, 1)
+            command.await_cmd()
+
+            if isinstance(command.response, bool) and command.response:
+                sleep(2)
+
+                command = Command(self.device_id, "8", [self.pump_id, 0], self.task_id)
+                self.device.post_command(command, 1)
+                command.await_cmd()
+
+                if isinstance(command.response, bool) and command.response:
+                    return
+        raise ConnectionError

custom/tasks/__init__.py

@@ -11,5 +11,5 @@
     "PSI_GMS_measure_all": GMS.GMSMeasureAll,
     # Phenometrics
     "ePBR_measure_all": PBR.ePBRMeasureAll,
-    "ePBR_pump": PBR.ePBRGeneralPump,
+    "ePBR_pump": PBR.ePBRpump,
 }