From 916c40747fdfca03bc9bbe27d2bd6bfde61b081b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Noah=20Piqu=C3=A9?= <noah.pique@psi.ch>
Date: Mon, 25 Nov 2024 17:12:27 +0100
Subject: [PATCH] Update coldbox.py

---
 coldbox.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/coldbox.py b/coldbox.py
index 793938e..91be8a4 100644
--- a/coldbox.py
+++ b/coldbox.py
@@ -39,12 +39,13 @@ class Tessie:
             Sends a request for help on the Tessie system.
         """
 
-    def __init__(self):
-        self.broker = 'coldbox01.psi.ch'
+    def __init__(self, broker):
+        print(broker)
+        self.broker = broker
         self.port = 1883
         self.topic = 'ctrlTessie'
         client_id = 'Python Tessie Client'
-        self._client = mqtt_client.Client(client_id)
+        self._client = mqtt_client.Client(mqtt_client.CallbackAPIVersion.VERSION1)
         Tessie.waiting = []
         Tessie.found = []
         self._connect_mqtt()
@@ -181,6 +182,10 @@ class Env:
     def getTempWater(self):
         return float(self._tessie.get("Temp_W", " tec 8"))
 
+    def getVprobe(self, number):
+        data = self._tessie.get(f"vprobe{number}")
+        return [float(x) for x in data.split(",")] # returns [-999.0] if not available
+
 
 class ConfTEC:
     def __init__(self, tessie: Tessie, i):
@@ -284,8 +289,8 @@ class TEC:
 
 
 class Coldbox:
-    def __init__(self):
-        self._tessie = Tessie()
+    def __init__(self, broker):
+        self._tessie = Tessie(broker)
         self.valve0 = Valve(self._tessie, 0)
         self.valve1 = Valve(self._tessie, 1)
         self.tecall = TEC(self._tessie, 0)
@@ -308,7 +313,7 @@ class Coldbox:
 def main():
     print("Hello World!")
 
-    box = Coldbox()
+    box = Coldbox('coldbox02.psi.ch')
     print(box.tec1.getTemp())
     print(box.tec1.getTemp())
     box.tec1.setVoltage(1.5)
@@ -316,6 +321,7 @@ def main():
     box.tec1.setVoltage(2)
 
     print(box.tec1.getTemp())
+    box.tecall.poff()
 
 
 if __name__ == "__main__":