From ff20725b0fb3fe4cb22b72bd5ccc649b3ab89863 Mon Sep 17 00:00:00 2001
From: askiiart <benzimm07@gmail.com>
Date: Wed, 2 Nov 2022 19:26:03 -0500
Subject: [PATCH] Add message in case of Arduino disconnect at any time

---
 Python/tx_speed.py | 125 +++++++++++++++++++++++++--------------------
 1 file changed, 69 insertions(+), 56 deletions(-)

diff --git a/Python/tx_speed.py b/Python/tx_speed.py
index 16317c7..ca5792b 100644
--- a/Python/tx_speed.py
+++ b/Python/tx_speed.py
@@ -1,8 +1,11 @@
-import os
-import GPUtil
 import time
-from pySerialTransfer import pySerialTransfer as txfer
 from subprocess import getoutput
+try:
+    import GPUtil
+    from pySerialTransfer import pySerialTransfer as txfer
+    from serial import SerialException
+except ImportError:
+    print('Make sure you have installed the required libraries: GPUtil, pySerialTransfer')
 
 try:
     # ---
@@ -11,71 +14,81 @@ try:
     # The port of the Uno will be "/dev/ttyXXX#", like "/dev/ttyACM0" or "/dev/ttyUSB0". I think macOS is the same.
     # Then, just remove "/dev/", and the number, from the port, and type it into the variable `port`
     # ---
-
-    port = 'ttyACM'
-    while True:
-        ports = getoutput(f'ls /dev | grep {port}').split('\n')
-        if ports != ['']:
-            ports.sort()
-            port = ports[len(ports) - 1]
-            break
-        print('Waiting for Arduino...')
-        time.sleep(3)
-
-    link = txfer.SerialTransfer(port, 115200, timeout=.1)
-
+    
+    def get_port():
+        port = 'ttyACM'
+        while True:
+            ports = getoutput(f'ls /dev | grep {port}').split('\n')
+            if ports != ['']:
+                ports.sort()
+                port = ports[len(ports) - 1]
+                break
+            print('Waiting for Arduino...')
+            time.sleep(3)
+    
+        return port
+    
+    link = txfer.SerialTransfer(get_port(), 115200, timeout=.1)
     link.open()
-    time.sleep(2)
-
 
     while True:
-        time.sleep(1)
-        gpus = GPUtil.getGPUs()
-        temp = max(gpu.temperature for gpu in gpus)
+        
+        try:
+            time.sleep(1)
+            gpus = GPUtil.getGPUs()
+            temp = max(gpu.temperature for gpu in gpus)
 
-        speed = int(((temp - 40) / 40) * 100)
-        min_speed = 0
-        max_speed = 100
-        if speed < 0:
-            speed = min_speed
-        elif speed > 100:
-            speed = max_speed
+            speed = int(((temp - 40) / 40) * 100)
+            min_speed = 34
+            max_speed = 100
+            if speed < 0:
+                speed = min_speed
+            elif speed > 100:
+                speed = max_speed
 
-        print(f'Temp: {temp}')
-        print(f'Speed: {speed}')
+            print(f'Temp: {temp}')
+            print(f'Speed: {speed}')
 
-        ################################################################################################################
-        # Send data to the Arduino
-        ################################################################################################################
+            ################################################################################################################
+            # Send data to the Arduino
+            ################################################################################################################
 
-        # send_size will be increased when data is added to payload
-        send_size = 0
+            # send_size will be increased when data is added to payload
+            send_size = 0
 
-        # Adds data to payload
-        int_size = link.tx_obj(speed, send_size) - send_size
-        send_size += int_size
+            # Adds data to payload
+            int_size = link.tx_obj(speed, send_size) - send_size
+            send_size += int_size
 
-        # Sends data to Arduino
-        link.send(send_size)
+            # Sends data to Arduino
+            link.send(send_size)
 
-        # Waits for response from Arduino, and reports errors while receiving packets
-        while not link.available():
-            if link.status < 0:
-                if link.status == txfer.CRC_ERROR:
-                    print('Error: CRC_ERROR')
-                elif link.status == txfer.PAYLOAD_ERROR:
-                    print('Error: PAYLOAD_ERROR')
-                elif link.status == txfer.STOP_BYTE_ERROR:
-                    print('Error: STOP_BYTE_ERROR')
-                else:
-                    print('Error: {}'.format(link.status))
+            # Waits for response from Arduino, and reports errors while receiving packets
+            while not link.available():
+                if link.status < 0:
+                    if link.status == txfer.CRC_ERROR:
+                        print('Error: CRC_ERROR')
+                    elif link.status == txfer.PAYLOAD_ERROR:
+                        print('Error: PAYLOAD_ERROR')
+                    elif link.status == txfer.STOP_BYTE_ERROR:
+                        print('Error: STOP_BYTE_ERROR')
+                    else:
+                        print('Error: {}'.format(link.status))
+            
+            # Parse response from Arduino
+            rec_int = link.rx_obj(obj_type=int, obj_byte_size=int_size, start_pos=(send_size - int_size))
 
-        # Parse response from Arduino
-        rec_int = link.rx_obj(obj_type=int, obj_byte_size=int_size, start_pos=(send_size - int_size))
+            # Evaluate for comm. errors
+            if speed != rec_int:
+                print(f'Error: sent {speed}, received {rec_int}')
+        except OSError or SerialException:
+            try:
+                link.close()
+            except:
+                pass
+            link = txfer.SerialTransfer(get_port(), 115200, timeout=.1)
+            link.open()
 
-        # Evaluate for comm. errors
-        if speed != rec_int:
-            print(f'Error: sent {speed}, received {rec_int}')
 
 except KeyboardInterrupt:
     try: