Leons lost commit

src/helmholtz_cage_device.py

@@ -80,6 +80,17 @@ class HelmholtzCageDevice:
             # The axes talks to the HW objects (Arduino, PSU) referenced in this object
             self.axes.append(Axis(i, self))
 
+        # --- Preliminary PSU type info ---
+        # Required post-initialization but before hardware has been connected
+        psu_type_string = config_handling.read_from_config("Supplies", "supply_model",
+                                                           config_handling.CONFIG_OBJECT)
+        if psu_type_string == "ps2000b":
+            self.psu_type = PSUDevicePS2000B
+        elif psu_type_string == "ql355tp":
+            self.psu_type = PSUDeviceQL355TP
+        else:
+            raise Exception("Invalid psu model:\n{}".format(self.psu_type))
+
         # --- HW COMMUNICATION THREAD ---
         self._cmd_exec_thread = Thread(target=self._cmd_exec_thread_method)
         self._cmd_exec_thread.start()