diff --git a/tools/tiny-test-fw/Utility/__init__.py b/tools/tiny-test-fw/Utility/__init__.py index b851d5e14..9a67f5a69 100644 --- a/tools/tiny-test-fw/Utility/__init__.py +++ b/tools/tiny-test-fw/Utility/__init__.py @@ -3,18 +3,18 @@ import sys _COLOR_CODES = { - "white": '\033[0m', - "red": '\033[31m', - "green": '\033[32m', - "orange": '\033[33m', - "blue": '\033[34m', - "purple": '\033[35m', - "W": '\033[0m', - "R": '\033[31m', - "G": '\033[32m', - "O": '\033[33m', - "B": '\033[34m', - "P": '\033[35m' + "white": u'\033[0m', + "red": u'\033[31m', + "green": u'\033[32m', + "orange": u'\033[33m', + "blue": u'\033[34m', + "purple": u'\033[35m', + "W": u'\033[0m', + "R": u'\033[31m', + "G": u'\033[32m', + "O": u'\033[33m', + "B": u'\033[34m', + "P": u'\033[35m' } @@ -29,8 +29,10 @@ def console_log(data, color="white", end="\n"): if color not in _COLOR_CODES: color = "white" color_codes = _COLOR_CODES[color] + if type(data) is type(b''): + data = data.decode('utf-8', 'replace') print(color_codes + data, end=end) if color not in ["white", "W"]: # reset color to white for later logs - print(_COLOR_CODES["white"] + "\r") + print(_COLOR_CODES["white"] + u"\r") sys.stdout.flush()