71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
import platform
|
|
|
|
DEBUG = 36
|
|
INFO = 37
|
|
WARN = 35
|
|
ERROR = 31
|
|
|
|
if platform.system() == "Windows":
|
|
HAS_COLOR_SUPPORT = platform.release() == "10"
|
|
|
|
if HAS_COLOR_SUPPORT:
|
|
# Enable ANSI codes. Otherwise, the ANSI sequences might not be
|
|
# evaluated correctly for the first colored print statement.
|
|
import ctypes
|
|
kernel32 = ctypes.windll.kernel32
|
|
|
|
# -11: stdout
|
|
handle_out = kernel32.GetStdHandle(-11)
|
|
|
|
console_mode = ctypes.c_long()
|
|
kernel32.GetConsoleMode(handle_out, ctypes.byref(console_mode))
|
|
|
|
# 0b100: ENABLE_VIRTUAL_TERMINAL_PROCESSING, enables ANSI codes
|
|
# see https://docs.microsoft.com/en-us/windows/console/setconsolemode
|
|
console_mode.value |= 0b100
|
|
kernel32.SetConsoleMode(handle_out, console_mode)
|
|
else:
|
|
HAS_COLOR_SUPPORT = True
|
|
|
|
info_text = ''
|
|
num_warnings = 0
|
|
num_errors = 0
|
|
|
|
def clear(clear_warnings=False, clear_errors=False):
|
|
global info_text, num_warnings, num_errors
|
|
info_text = ''
|
|
if clear_warnings:
|
|
num_warnings = 0
|
|
if clear_errors:
|
|
num_errors = 0
|
|
|
|
def format_text(text):
|
|
return (text[:80] + '..') if len(text) > 80 else text # Limit str size
|
|
|
|
def log(text, color=None):
|
|
if HAS_COLOR_SUPPORT and color is not None:
|
|
csi = '\033['
|
|
text = csi + str(color) + 'm' + text + csi + '0m'
|
|
print(text)
|
|
|
|
def debug(text):
|
|
log(text, DEBUG)
|
|
|
|
def info(text):
|
|
global info_text
|
|
log(text, INFO)
|
|
info_text = format_text(text)
|
|
|
|
def print_warn(text):
|
|
log('WARNING: ' + text, WARN)
|
|
|
|
def warn(text):
|
|
global num_warnings
|
|
num_warnings += 1
|
|
print_warn(text)
|
|
|
|
def error(text):
|
|
global num_errors
|
|
num_errors += 1
|
|
log('ERROR: ' + text, ERROR)
|