wip

python-test/flow.py

@@ -79,7 +79,7 @@ def flow(client):
 
 if __name__ == "__main__":
     def main(address):
-        device = ble_conn(address)
+        client = ble_conn(address)
 
         try:
             if get_unit_is_fahrenheit(client):
@@ -90,9 +90,14 @@ if __name__ == "__main__":
         except:
             print("\nTurning heater off")
             set_state(client, (False, False)) # turn off heater and pump
+
+            print("Disconnecting")
+            client.disconnect()
+
             raise
 
-        print("Disconnecting...")
+        print("Disconnecting")
+        client.disconnect()
 
     arg = None
     if len(sys.argv) > 1:

python-test/poll.py

@@ -83,14 +83,23 @@ if __name__ == "__main__":
         if device == None:
             return
 
-        print("Writing...")
-        set_target_temp(device, 190.0)
-
-        print("Reading...")
-        for i in range(0, 5):
-            test_poll(device)
-            print()
-            time.sleep(2.0)
+        try:
+            print("Writing...")
+            set_target_temp(device, 190.0)
+
+            print("Reading...")
+            for i in range(0, 5):
+                test_poll(device)
+                print()
+                time.sleep(2.0)
+        except:
+            print("Disconnecting")
+            client.disconnect()
+
+            raise
+
+        print("Disconnecting")
+        client.disconnect()
 
     import sys