80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
import serial
|
|
import sys
|
|
import time
|
|
|
|
PACKET_HEADER = b'\0\0\0\0\0\x01'
|
|
SIGN_TYPE = b'Z'
|
|
SIGN_ADDRESS = b'00'
|
|
VALID_NAMES = b'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
|
|
|
|
def encode_packet(command_code, data):
|
|
return PACKET_HEADER + SIGN_TYPE + SIGN_ADDRESS + b'\x02' + command_code + data + b'\x04'
|
|
|
|
def generate_memory_config(files: list[tuple[bytes, int]]):
|
|
data = b'$'
|
|
for name, size in files:
|
|
if len(name) != 1:
|
|
raise ValueError()
|
|
if name not in VALID_NAMES:
|
|
raise ValueError()
|
|
data += name
|
|
data += b'AU'
|
|
data += hex(size).upper()[2:].encode().rjust(4, b'0')
|
|
# data += b'00FF'
|
|
data += b'FFFF'
|
|
return encode_packet(b'E', data)
|
|
|
|
def generate_run_order(files: list[bytes]):
|
|
data = b'.TU'
|
|
for name in files:
|
|
if len(name) != 1:
|
|
raise ValueError()
|
|
if name not in VALID_NAMES:
|
|
raise ValueError()
|
|
data += name
|
|
return encode_packet(b'E', data)
|
|
|
|
def generate_write_text_file(name: bytes, content: bytes):
|
|
return encode_packet(b'A', name + content)
|
|
|
|
def main(args: list[str]):
|
|
if len(args) < 3:
|
|
print(f'Usage: {args[0]} <file> <serial>')
|
|
exit(1)
|
|
file_name = args[1]
|
|
serial_port = args[2]
|
|
with open(file_name) as f:
|
|
lines = f.readlines()
|
|
if len(lines) > len(VALID_NAMES):
|
|
print('Too many lines! I dont have enough files!')
|
|
exit(1)
|
|
|
|
run_order = []
|
|
files = []
|
|
text_packets = []
|
|
|
|
for file_name, line in zip(VALID_NAMES, lines):
|
|
file_name = bytes([file_name])
|
|
data = line.strip().encode()
|
|
length = len(data)
|
|
run_order.append(file_name)
|
|
files.append((file_name, length))
|
|
text_packets.append(generate_write_text_file(file_name, data))
|
|
|
|
with serial.Serial(serial_port, 9600) as ser:
|
|
print('Config memory')
|
|
ser.write(generate_memory_config(files))
|
|
print('+', generate_memory_config(files).hex())
|
|
time.sleep(1)
|
|
print('Send files')
|
|
for tp in text_packets:
|
|
ser.write(tp)
|
|
print('+', tp.hex())
|
|
time.sleep(1)
|
|
print('Send run order')
|
|
ser.write(generate_run_order(run_order))
|
|
print('+', generate_run_order(run_order).hex())
|
|
print('Done!')
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv)
|