# -*- coding: utf-8 -*-
"""
celery.utils.term
~~~~~~~~~~~~~~~~~
Terminals and colors.
"""
from __future__ import absolute_import, unicode_literals
import platform
from six import text_type
from thriftpool.utils.encoding import smart_str, smart_unicode
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)
OP_SEQ = '\033[%dm'
RESET_SEQ = '\033[0m'
COLOR_SEQ = '\033[1;%dm'
fg = lambda s: COLOR_SEQ % s
SYSTEM = platform.system()
IS_WINDOWS = SYSTEM == 'Windows'
[docs]def isatty(stream):
return hasattr(stream, 'isatty') and stream.isatty()
[docs]class colored(object):
"""Terminal colored text.
Example::
>>> c = colored(enabled=True)
>>> print(str(c.red('the quick '), c.blue('brown ', c.bold('fox ')),
... c.magenta(c.underline('jumps over')),
... c.yellow(' the lazy '),
... c.green('dog ')))
"""
def __init__(self, *s, **kwargs):
self.s = s
self.enabled = not IS_WINDOWS and kwargs.get('enabled', True)
self.op = kwargs.get('op', '')
self.names = {'black': self.black,
'red': self.red,
'green': self.green,
'yellow': self.yellow,
'blue': self.blue,
'magenta': self.magenta,
'cyan': self.cyan,
'white': self.white}
def _add(self, a, b):
return text_type(a) + text_type(b)
def _fold_no_color(self, a, b):
try:
A = a.no_color()
except AttributeError:
A = text_type(a)
try:
B = b.no_color()
except AttributeError:
B = text_type(b)
return smart_str(A) + smart_str(B)
[docs] def no_color(self):
if self.s:
return reduce(self._fold_no_color, self.s)
return ''
[docs] def embed(self):
prefix = ''
if self.enabled:
prefix = self.op
return smart_str(prefix) + smart_str(reduce(self._add, self.s))
def __unicode__(self):
suffix = ''
if self.enabled:
suffix = RESET_SEQ
return smart_unicode(self.embed()) + suffix
def __str__(self):
return smart_str(self.__unicode__())
def __len__(self):
return len(str(self))
[docs] def node(self, s, op):
return self.__class__(enabled=self.enabled, op=op, *s)
[docs] def black(self, *s):
return self.node(s, fg(30 + BLACK))
[docs] def red(self, *s):
return self.node(s, fg(30 + RED))
[docs] def green(self, *s):
return self.node(s, fg(30 + GREEN))
[docs] def yellow(self, *s):
return self.node(s, fg(30 + YELLOW))
[docs] def blue(self, *s):
return self.node(s, fg(30 + BLUE))
[docs] def magenta(self, *s):
return self.node(s, fg(30 + MAGENTA))
[docs] def cyan(self, *s):
return self.node(s, fg(30 + CYAN))
[docs] def white(self, *s):
return self.node(s, fg(30 + WHITE))
def __repr__(self):
return repr(self.no_color())
[docs] def bold(self, *s):
return self.node(s, OP_SEQ % 1)
[docs] def underline(self, *s):
return self.node(s, OP_SEQ % 4)
[docs] def blink(self, *s):
return self.node(s, OP_SEQ % 5)
[docs] def reverse(self, *s):
return self.node(s, OP_SEQ % 7)
[docs] def bright(self, *s):
return self.node(s, OP_SEQ % 8)
[docs] def ired(self, *s):
return self.node(s, fg(40 + RED))
[docs] def igreen(self, *s):
return self.node(s, fg(40 + GREEN))
[docs] def iyellow(self, *s):
return self.node(s, fg(40 + YELLOW))
[docs] def iblue(self, *s):
return self.node(s, fg(40 + BLUE))
[docs] def imagenta(self, *s):
return self.node(s, fg(40 + MAGENTA))
[docs] def icyan(self, *s):
return self.node(s, fg(40 + CYAN))
[docs] def iwhite(self, *s):
return self.node(s, fg(40 + WHITE))
[docs] def reset(self, *s):
return self.node(s or [''], RESET_SEQ)
def __add__(self, other):
return text_type(self) + text_type(other)