From e050c71104c86744fa404b33ddaedc58889dc27c Mon Sep 17 00:00:00 2001
From: Ethan Lewis <lewisec@g.cofc.edu>
Date: Tue, 12 Nov 2024 21:33:17 -0500
Subject: [PATCH] Created main water script

---
 .DS_Store            | Bin 10244 -> 10244 bytes
 Engine/water_main.py |  93 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 Engine/water_main.py

diff --git a/.DS_Store b/.DS_Store
index 0c7ec6b3c7a48faa857712b6f897a5d32d77f424..3d6d7856999ed1ed566ef828e5d56b65d023a061 100644
GIT binary patch
delta 173
zcmZn(XbIS$CJ^g(fPsO5g+Y%YogtH<Bsbs1B`GIA2`I+#P*(M=QqeI-RQVLR{I0l~
zH|X*O8HT~h`MCu^Jq!$j8#Xr!RI)MUIZXa1Cd0aIO3r%a$@1dzP?oc}BIAP1sp8Z5
L@v4}dDn1_oS!y$^

delta 173
zcmZn(XbIS$CJ?K2g@J*Ag+Y%YogtH<Bsbs1B`GIA2`I(^#MgUsjya;rr{I+@$S@2}
z&d)6X>S16IIIy`{ppuO#A#d_OF&S2%$WpG!^5XJPmb17b<Alwr;?w!@s+gQAJ|6&F
CSSnWl

diff --git a/Engine/water_main.py b/Engine/water_main.py
new file mode 100644
index 0000000..9cef370
--- /dev/null
+++ b/Engine/water_main.py
@@ -0,0 +1,93 @@
+import serial
+from datetime import datetime
+from influxdb_client import InfluxDBClient, Point
+from influxdb_client.client.write_api import SYNCHRONOUS
+import certifi
+import time
+
+# InfluxDB configuration
+token = '8Kph_oFbnNqTvgUfERhTJveJpw1A08-HDsnhftEIYj8HRSUJSmKIdoPZdaIbNeIkG4likhDrOtr5FgyF5W03hg=='
+org = "Ethan"
+bucket = "Test"  # Use "bucket" instead of "database" in influxdb-client
+url = "https://us-east-1-1.aws.cloud2.influxdata.com"
+
+# Create InfluxDB client
+client = InfluxDBClient(url=url, token=token, org=org, ssl_ca_cert=certifi.where())
+write_api = client.write_api(write_options=SYNCHRONOUS)
+
+# Log file paths
+sensor_log_file_path = '/home/openstem/Desktop/openstem_ethan/water_sensor_log.txt'
+pump_log_file_path = '/home/openstem/Desktop/openstem_ethan/pump_control_log.txt'
+
+# Initialize serial connection to Arduino
+ser = serial.Serial('/dev/ttyUSB0', 9600, timeout=1)
+time.sleep(2)  # Wait for the serial connection to stabilize
+
+# Function to log messages to a text file
+def log_message(message, log_file_path):
+    with open(log_file_path, 'a') as log_file:
+        log_file.write(f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - {message}\n")
+
+# Function to fetch sensor data from Arduino
+def fetch_sensor_data():
+    ser.write(b'R')  # Send 'R' command to request sensor readings
+    time.sleep(5)  # Wait for Arduino to respond
+
+    data = {"pH": None, "EC": None, "Temperature": None}
+    while ser.in_waiting > 0:
+        line = ser.readline().decode('utf-8').strip()
+        if "pH" in line:
+            data["pH"] = float(line.split(":")[1].strip())
+        elif "EC" in line:
+            data["EC"] = float(line.split(":")[1].strip())
+        elif "Temperature" in line:
+            data["Temperature"] = float(line.split(":")[1].strip())
+    return data
+
+# Function to upload water sensor data to InfluxDB
+def upload_water_data(data):
+    if all(value is not None for value in data.values()):
+        point = (
+            Point("water_sensors")  # Measurement for water sensors
+            .field("RTD", data["Temperature"])  # RTD value
+            .field("pH", data["pH"])  # pH value
+            .field("EC", data["EC"])  # Electrical conductivity
+        )
+        write_api.write(bucket=bucket, org=org, record=point)
+        log_message("Data uploaded to InfluxDB", sensor_log_file_path)
+    else:
+        log_message("Incomplete data. Upload aborted.", sensor_log_file_path)
+
+# Function to trigger pump control on Arduino
+def trigger_pump_control():
+    try:
+        ser.write(b'P')  # Send 'P' command to trigger pump control
+        time.sleep(5)  # Wait for Arduino to process and respond
+
+        # Read and log Arduino responses
+        while ser.in_waiting > 0:
+            line = ser.readline().decode('utf-8').strip()
+            log_message(line, pump_log_file_path)
+            print(line)
+        log_message("Pump control command executed successfully.", pump_log_file_path)
+        
+    except Exception as e:
+        log_message(f"Error: {str(e)}", pump_log_file_path)
+
+# Main execution
+try:
+    # Check the current time
+    current_time = datetime.now()
+    minute = current_time.minute
+
+    # Perform sensor data upload every half hour
+    sensor_data = fetch_sensor_data()
+    upload_water_data(sensor_data)
+
+    # Perform pump control at the top of every hour
+    if minute == 0:
+        time.sleep(180)  # Delay 3 minutes to allow for sensor reading and uploading
+        trigger_pump_control()
+
+except Exception as e:
+    log_message(f"Error: {str(e)}", sensor_log_file_path)
\ No newline at end of file