3099d353f9
Add a theme section in the configuration file to allow users to set the colors used by different elements of the ui. Improve the update mechanism so that incremental changes to the configuration file can be incorporated.
93 lines
3.5 KiB
Python
93 lines
3.5 KiB
Python
"""
|
|
Small code to handle colored text
|
|
"""
|
|
import sys
|
|
import re
|
|
|
|
def _color_supported(stream):
|
|
"""Returns True is the stream supports colors"""
|
|
if sys.platform == 'win32' and 'ANSICON' not in os.environ:
|
|
return False
|
|
if hasattr(stream, 'isatty') and stream.isatty(): # we have a tty
|
|
try:
|
|
import curses
|
|
curses.setupterm()
|
|
return curses.tigetnum('colors') >= 8
|
|
except Exception: # not picky.
|
|
return False
|
|
return False
|
|
|
|
COLOR_LIST = [u'black', u'red', u'green', u'yellow', u'blue', u'purple', u'cyan', u'grey']
|
|
|
|
def generate_colors(stream, color=True, bold=True, italic=True):
|
|
colors = {name: u'' for name in COLOR_LIST}
|
|
colors.update({u'b' +name: u'' for name in COLOR_LIST})
|
|
colors.update({u'i' +name: u'' for name in COLOR_LIST})
|
|
colors.update({u'bi'+name: u'' for name in COLOR_LIST})
|
|
colors[u'bold'] = u''
|
|
colors[u'italic'] = u''
|
|
colors[u'end'] = u''
|
|
colors[u''] = u''
|
|
|
|
if (color or bold or italic) and _color_supported(stream):
|
|
bold_flag, italic_flag = '', ''
|
|
if bold:
|
|
colors['bold'] = u'\033[1m'
|
|
bold_flag = '1;'
|
|
if italic:
|
|
colors['italic'] = u'\033[3m'
|
|
italic_flag = '3;'
|
|
if bold and italic:
|
|
colors['bolditalic'] = u'\033[1;3m'
|
|
|
|
for i, name in enumerate(COLOR_LIST):
|
|
if color:
|
|
colors[name] = u'\x1b[3{}m'.format(i)
|
|
colors.update({u'b'+name: u'\033[{}3{}m'.format(bold_flag, i) for i, name in enumerate(COLOR_LIST)})
|
|
colors.update({u'i'+name: u'\033[{}3{}m'.format(italic_flag, i) for i, name in enumerate(COLOR_LIST)})
|
|
colors.update({u'bi'+name: u'\033[{}{}3{}m'.format(bold_flag, italic_flag, i) for i, name in enumerate(COLOR_LIST)})
|
|
else:
|
|
if bold:
|
|
colors.update({u'b'+name: u'\033[1m' for i, name in enumerate(COLOR_LIST)})
|
|
if italic:
|
|
colors.update({u'i'+name: u'\033[3m' for i, name in enumerate(COLOR_LIST)})
|
|
if bold or italic:
|
|
colors.update({u'bi'+name: u'\033[{}{}m'.format(bold_flag, italic_flag) for i, name in enumerate(COLOR_LIST)})
|
|
|
|
if color or bold or italic:
|
|
colors['end'] = u'\033[0m'
|
|
|
|
return colors
|
|
|
|
|
|
COLORS_OUT = generate_colors(sys.stdout, color=False, bold=False, italic=False)
|
|
COLORS_ERR = generate_colors(sys.stderr, color=False, bold=False, italic=False)
|
|
|
|
def dye_out(s, color='end'):
|
|
return u'{}{}{}'.format(COLORS_OUT[color], s, COLORS_OUT['end'])
|
|
|
|
def dye_err(s, color='end'):
|
|
return u'{}{}{}'.format(COLORS_ERR[color], s, COLORS_OUT['end'])
|
|
|
|
def _nodye(s, *args, **kwargs):
|
|
return s
|
|
|
|
def setup(conf):
|
|
global COLORS_OUT, COLORS_ERR
|
|
COLORS_OUT = generate_colors(sys.stdout, color=conf['formating']['color'],
|
|
bold=conf['formating']['bold'],
|
|
italic=conf['formating']['italics'])
|
|
COLORS_ERR = generate_colors(sys.stderr, color=conf['formating']['color'],
|
|
bold=conf['formating']['bold'],
|
|
italic=conf['formating']['italics'])
|
|
for key, value in conf['theme'].items():
|
|
COLORS_OUT[key] = COLORS_OUT.get(value, '')
|
|
COLORS_ERR[key] = COLORS_ERR.get(value, '')
|
|
|
|
# undye
|
|
undye_re = re.compile('\x1b\[[;\d]*[A-Za-z]')
|
|
|
|
def undye(s):
|
|
"""Purge string s of color"""
|
|
return undye_re.sub('', s)
|