hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add ability to force color mode even if terminal is NOT a tty

Browse source
This commit is contained in:
Todd Gamblin 2015-05-29 17:19:03 -07:00
parent ceecd2ce41
commit 9b40d93fc2
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lib/spack/llnl/util/tty/color.py
View file

@ -99,6 +99,10 @@ def __init__(self, message):
color_re = r'@(?:@|\.|([*_])?([a-zA-Z])?(?:{((?:[^}]|}})*)})?)' color_re = r'@(?:@|\.|([*_])?([a-zA-Z])?(?:{((?:[^}]|}})*)})?)'
# Force color even if stdout is not a tty.
_force_color = False
class match_to_ansi(object): class match_to_ansi(object):
def __init__(self, color=True): def __init__(self, color=True):
self.color = color self.color = color
@ -162,7 +166,7 @@ def cwrite(string, stream=sys.stdout, color=None):
then it will be set based on stream.isatty(). then it will be set based on stream.isatty().
""" """
if color is None: if color is None:
color = stream.isatty() color = stream.isatty() or _force_color
stream.write(colorize(string, color=color)) stream.write(colorize(string, color=color))
@ -189,7 +193,7 @@ def write(self, string, **kwargs):
if raw: if raw:
color=True color=True
else: else:
color = self._stream.isatty() color = self._stream.isatty() or _force_color
raw_write(colorize(string, color=color)) raw_write(colorize(string, color=color))
def writelines(self, sequence, **kwargs): def writelines(self, sequence, **kwargs):