k80-linux-cooling/Demos/Python/pyserialtransfer_demo.py
2022-10-25 21:28:47 -05:00

76 lines
2.3 KiB
Python

import time
from subprocess import getoutput
from pySerialTransfer import pySerialTransfer as txfer
try:
# ---
# You will need to customize port to your own computer; to find your port:
# With your Arduino plugged in, open Arduino IDE, and click "Select Board". Then, you can see the port.
# On Linux, it will be "/dev/ttyXXX#", like "/dev/ttyACM0" or "/dev/ttyUSB0". I think macOS is the same.
# On Windows, it will be "COM#", like "COM3"
# ---
port = 'ttyACM'
while True:
ports = getoutput(f'ls /dev | grep {port}').split('\n')
if ports != ['']:
ports.sort()
port = ports[len(ports) - 1]
break
link = txfer.SerialTransfer(port, 115200, timeout=.1)
link.open()
time.sleep(2)
temp = 0
while temp < 100:
time.sleep(1)
################################################################################################################
# How to send data to the Arduino
################################################################################################################
# send_size will be increased when data is added to payload
send_size = 0
# Adds data to payload
temp += 1
int_size = link.tx_obj(temp, send_size) - send_size
send_size += int_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))
# Parse response from Arduino
rec_int = link.rx_obj(obj_type=int, obj_byte_size=int_size, start_pos=(send_size - int_size))
# Print sent and received data
print('Send:', temp, 'Received:', rec_int)
except KeyboardInterrupt:
try:
link.close()
except:
pass
except:
import traceback
traceback.print_exc()
try:
link.close()
except:
pass