fly316/PX4-Autopilot
7
0
Fork 0
mirror of https://gitee.com/mirrors_PX4/PX4-Autopilot.git synced 2026-10-31 04:41:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Tools/HIL: color output (ERROR, WARN, etc)

Browse Source
This commit is contained in:
Daniel Agar
2021-07-16 19:33:27 -04:00
parent 7f4ffcef93
commit 8c6ade0fc8
4 changed files with 120 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Tools/HIL/monitor_firmware_upload.py
+25 -1
View File
@@ -7,6 +7,30 @@ from argparse import ArgumentParser
import re
import sys
COLOR_YELLOW = "\x1b[31m"
COLOR_GREEN = "\x1b[32m"
COLOR_RED = "\x1b[33m"
COLOR_WHITE = "\x1b[37m"
COLOR_RESET = "\x1b[0m"
def print_line(line):
if "WARNING" in line:
line = line.replace("WARNING", f"{COLOR_YELLOW}WARNING{COLOR_RESET}", 1)
elif "WARN" in line:
line = line.replace("WARN", f"{COLOR_YELLOW}WARN{COLOR_RESET}", 1)
elif "ERROR" in line:
line = line.replace("ERROR", f"{COLOR_RED}ERROR{COLOR_RESET}", 1)
elif "INFO" in line:
line = line.replace("INFO", f"{COLOR_WHITE}INFO{COLOR_RESET}", 1)
if "PASSED" in line:
line = line.replace("PASSED", f"{COLOR_GREEN}PASSED{COLOR_RESET}", 1)
if "FAILED" in line:
line = line.replace("FAILED", f"{COLOR_RED}FAILED{COLOR_RESET}", 1)
print(line, end='')
def monitor_firmware_upload(port, baudrate):
ser = serial.Serial(port, baudrate, bytesize=serial.EIGHTBITS, parity=serial.PARITY_NONE, stopbits=serial.STOPBITS_ONE, timeout=1, xonxoff=True, rtscts=False, dsrdtr=False)
@@ -23,7 +47,7 @@ def monitor_firmware_upload(port, baudrate):
sys.exit(0)
else:
if len(serial_line) > 0:
print(serial_line, end='')
print_line(serial_line)
if time.time() > timeout_start + timeout:
print("Error, timeout")