from os import linesep
# we can also do print(colored('green', msg), end='') for example...
# printc helps if You have a global color_enabled flag in Your program
def printc(color, msg, no_newline=False, enable_color=False):
"""like print but with color support (if enabled!)"""
ret = str()
if enable_color:
ret += colored(color, msg)
else:
ret += msg
if no_newline:
end = ''
else:
end = linesep
print(ret, end=end, flush=True)
return ret
def colored(color, msg):
"""returns msg in ANSI color"""
ret = str()
if color.startswith("#"):
ret += '\033[38;2;'
ret += str(int(color[1:3], 16)) + ';' \
+ str(int(color[3:5], 16)) + ';' \
+ str(int(color[5:7], 16))
ret += 'm'
else:
c = 'default'
if color == 'red':
c = '31'
elif color == 'green':
c = '32'
elif color == 'yellow':
c = '33'
elif color == 'blue':
c = '34'
elif color == 'magenta':
c = '35'
elif color == 'cyan':
c = '36'
if c is not 'default':
# set color
ret += '\033[0;' + c + 'm'
ret += msg
# reset color
ret += '\033[0m'
return ret
# When data contains ANSI color escape codes convert them to BBcode format
# e.g.: Foo bar baz [\033[0;31m + text + [\033[0m
# where 31 can be 31..36 31=red, 32=green,
# 33=yellow, 34=blue, 35=magenta, 36-cyan
# 0 resets back to standard
# we can map those three parts to [color=#abcdef] + text + [/color]
def convert_ansi_to_bbcode_colors(instr):
"""convert ansi escape sequences in a string to bbcode tags"""
ansi_red = '\033[0;31m'
ansi_green = '\033[0;32m'
ansi_yellow = '\033[0;33m'
ansi_blue = '\033[0;34m'
ansi_magenta = '\033[0;35m'
ansi_cyan = '\033[0;36m'
ansi_end = '\033[0m'
bb_red = '[color=#FF0000]'
bb_green = '[color=#00FF00]'
bb_yellow = '[color=#FFFF00]'
bb_blue = '[color=#0000FF]'
bb_magenta = '[color=#FF00FF]'
bb_cyan = '[color=#00FFFF]'
bb_end = '[/color]'
return instr.replace(ansi_red, bb_red) \
.replace(ansi_green, bb_green) \
.replace(ansi_yellow, bb_yellow) \
.replace(ansi_blue, bb_blue) \
.replace(ansi_magenta, bb_magenta) \
.replace(ansi_cyan, bb_cyan) \
.replace(ansi_end, bb_end)
