From 075be8c67ab28c13d5afa613e50ad81a13e9c316 Mon Sep 17 00:00:00 2001
From: Noel <75380532+Eagleshot@users.noreply.github.com>
Date: Wed, 15 Nov 2023 18:06:45 +0100
Subject: [PATCH] Set low and recovery voltage witty pi 4

---
 main.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/main.py b/main.py
index d3bbeb2..b71f9a4 100644
--- a/main.py
+++ b/main.py
@@ -549,8 +549,34 @@ def get_recovery_voltage_treshold_witty_pi_4():
         # error += f"Could not get recovery voltage treshold: {str(e)}" # TODO Return error value
         print(f"Could not get recovery voltage treshold: {str(e)}")
         return "-"
+
+# Set low voltage treshold
+def set_low_voltage_treshold_witty_pi_4(voltage: float):
+    '''Sets the low voltage treshold from the Witty Pi 4'''
+    try:
+        low_voltage_treshold = run_witty_pi_4_command(f"set_low_voltage_threshold {voltage}")
+        print(f"Set low voltage treshold: {low_voltage_treshold} V")
+        return low_voltage_treshold
+    except Exception as e:
+        # error += f"Could not set low voltage treshold: {str(e)}" # TODO Return error value
+        print(f"Could not set low voltage treshold: {str(e)}")
+        return "-"
+    
+# Set recovery voltage treshold
+def set_recovery_voltage_treshold_witty_pi_4(voltage: float):
+    '''Sets the recovery voltage treshold from the Witty Pi 4'''
+    try:
+        recovery_voltage_treshold = run_witty_pi_4_command(f"set_recovery_voltage_threshold {voltage}")
+        print(f"Set recovery voltage treshold: {recovery_voltage_treshold} V")
+        return recovery_voltage_treshold
+    except Exception as e:
+        # error += f"Could not set recovery voltage treshold: {str(e)}" # TODO Return error value
+        print(f"Could not set recovery voltage treshold: {str(e)}")
+        return "-"
     
 # TODO
+set_low_voltage_treshold_witty_pi_4(2.5)
+set_recovery_voltage_treshold_witty_pi_4(3.0)
 get_low_voltage_treshold_witty_pi_4()
 get_recovery_voltage_treshold_witty_pi_4()