/
term.py
159 lines (115 loc) · 3.82 KB
/
term.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
# -*- coding: utf-8 -*-
"""
celery.utils.term
~~~~~~~~~~~~~~~~~
Terminals and colors.
"""
from __future__ import absolute_import, unicode_literals
import platform
from functools import reduce
from kombu.utils.encoding import safe_str
from celery.five import string
__all__ = ['colored']
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
IS_WINDOWS = platform.system() == 'Windows'
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 string(a) + string(b)
def _fold_no_color(self, a, b):
try:
A = a.no_color()
except AttributeError:
A = string(a)
try:
B = b.no_color()
except AttributeError:
B = string(b)
return safe_str(safe_str(A) + safe_str(B))
def no_color(self):
if self.s:
return safe_str(reduce(self._fold_no_color, self.s))
return ''
def embed(self):
prefix = ''
if self.enabled:
prefix = self.op
return safe_str(prefix) + safe_str(reduce(self._add, self.s))
def __unicode__(self):
suffix = ''
if self.enabled:
suffix = RESET_SEQ
return safe_str(self.embed() + safe_str(suffix))
def __str__(self):
return safe_str(self.__unicode__())
def node(self, s, op):
return self.__class__(enabled=self.enabled, op=op, *s)
def black(self, *s):
return self.node(s, fg(30 + BLACK))
def red(self, *s):
return self.node(s, fg(30 + RED))
def green(self, *s):
return self.node(s, fg(30 + GREEN))
def yellow(self, *s):
return self.node(s, fg(30 + YELLOW))
def blue(self, *s):
return self.node(s, fg(30 + BLUE))
def magenta(self, *s):
return self.node(s, fg(30 + MAGENTA))
def cyan(self, *s):
return self.node(s, fg(30 + CYAN))
def white(self, *s):
return self.node(s, fg(30 + WHITE))
def __repr__(self):
return repr(self.no_color())
def bold(self, *s):
return self.node(s, OP_SEQ % 1)
def underline(self, *s):
return self.node(s, OP_SEQ % 4)
def blink(self, *s):
return self.node(s, OP_SEQ % 5)
def reverse(self, *s):
return self.node(s, OP_SEQ % 7)
def bright(self, *s):
return self.node(s, OP_SEQ % 8)
def ired(self, *s):
return self.node(s, fg(40 + RED))
def igreen(self, *s):
return self.node(s, fg(40 + GREEN))
def iyellow(self, *s):
return self.node(s, fg(40 + YELLOW))
def iblue(self, *s):
return self.node(s, fg(40 + BLUE))
def imagenta(self, *s):
return self.node(s, fg(40 + MAGENTA))
def icyan(self, *s):
return self.node(s, fg(40 + CYAN))
def iwhite(self, *s):
return self.node(s, fg(40 + WHITE))
def reset(self, *s):
return self.node(s or [''], RESET_SEQ)
def __add__(self, other):
return string(self) + string(other)