198 lines
4.9 KiB
Python
198 lines
4.9 KiB
Python
# Python library for ansi colorization
|
|
import re, sys
|
|
|
|
|
|
class code:
|
|
K = "\033[1;30m"
|
|
R = "\033[1;31m"
|
|
G = "\033[1;32m"
|
|
B = "\033[1;34m"
|
|
Y = "\033[1;33m"
|
|
M = "\033[1;35m"
|
|
C = "\033[1;36m"
|
|
W = "\033[1;37m"
|
|
|
|
k = "\033[0;30m"
|
|
r = "\033[0;31m"
|
|
g = "\033[0;32m"
|
|
b = "\033[0;34m"
|
|
y = "\033[0;33m"
|
|
m = "\033[0;35m"
|
|
c = "\033[0;36m"
|
|
w = "\033[0;37m"
|
|
|
|
bk = "\033[40m"
|
|
br = "\033[41m"
|
|
bg = "\033[42m"
|
|
by = "\033[43m"
|
|
bb = "\033[44m"
|
|
bm = "\033[45m"
|
|
bc = "\033[46m"
|
|
bw = "\033[47m"
|
|
|
|
S = "\033[1m" # strong
|
|
s = "\033[2m" # strong off
|
|
U = "\033[4m" # underline
|
|
u = "\033[24m" # underline off
|
|
Z = "\033[0m" # zero colors
|
|
ic = "\033[7m" # inverse colors
|
|
io = "\033[27m" # inverse off
|
|
st = "\033[9m" # strike on
|
|
so = "\033[29m" # strike off
|
|
CLR = "\033[2J"
|
|
CLREND = "\033[K"
|
|
CLRBEG = "\033[1K"
|
|
CLRSCR = CLR + "\033[0;0H"
|
|
|
|
color_keys = "K,R,G,B,Y,M,C,W,k,r,g,b,y,m,c,w,S,s,U,u,Z,ic,io,st,so,bk,br,bg,by,bb,bm,bc,bw,CLR,CLREND,CLRBEG,CLRSCR".split(
|
|
","
|
|
)
|
|
color_list = [
|
|
K,
|
|
R,
|
|
G,
|
|
B,
|
|
Y,
|
|
M,
|
|
C,
|
|
W,
|
|
k,
|
|
r,
|
|
g,
|
|
b,
|
|
y,
|
|
m,
|
|
c,
|
|
w,
|
|
S,
|
|
s,
|
|
U,
|
|
u,
|
|
Z,
|
|
ic,
|
|
io,
|
|
st,
|
|
so,
|
|
bk,
|
|
br,
|
|
bg,
|
|
by,
|
|
bb,
|
|
bm,
|
|
bc,
|
|
bw,
|
|
CLR,
|
|
CLREND,
|
|
CLRBEG,
|
|
CLRSCR,
|
|
]
|
|
custom_match = re.compile(r"(\${)([0-9;]*[ABCDEFGHJKSTfminsu]+)(})")
|
|
|
|
def pos(self, y, x):
|
|
""" Go to absolute position """
|
|
return "\033[" + str(y) + ";" + str(x) + "H"
|
|
|
|
def column(self, x):
|
|
""" Go to absolute column """
|
|
return "\033[" + str(x) + "G"
|
|
|
|
def posprint(self, y, x, s):
|
|
""" Print string at a location """
|
|
sys.stdout.write(self.pos(y, x) + str(s))
|
|
self.flush()
|
|
|
|
def clear(self):
|
|
sys.stdout.write(self.CLRSCR + self.pos(0, 0))
|
|
|
|
def clear_to_end(self):
|
|
sys.stdout.write(self.CLREND)
|
|
|
|
def clear_to_beginning(self):
|
|
sys.stdout.write(self.CLRBEG)
|
|
|
|
def up(self, n=1):
|
|
sys.stdout.write("\033[" + str(n) + "A")
|
|
|
|
def down(self, n=1):
|
|
sys.stdout.write("\033[" + str(n) + "B")
|
|
|
|
def right(self, n=1):
|
|
sys.stdout.write("\033[" + str(n) + "C")
|
|
|
|
def left(self, n=1):
|
|
sys.stdout.write("\033[" + str(n) + "D")
|
|
|
|
def up_line(self, n=1):
|
|
sys.stdout.write("\033[" + str(n) + "F")
|
|
|
|
def down_line(self, n=1):
|
|
sys.stdout.write("\033[" + str(n) + "E")
|
|
|
|
def save(self):
|
|
""" Save cursor position """
|
|
sys.stdout.write("\033[s")
|
|
|
|
def restore(self):
|
|
""" Restore cursor position """
|
|
sys.stdout.write("\033[u")
|
|
|
|
def color_string(self, s):
|
|
for i, c in enumerate(self.color_keys):
|
|
s = s.replace("${" + c + "}", self.color_list[i])
|
|
return self.custom_color(s)
|
|
|
|
def nocolor_string(self, s):
|
|
for i, c in enumerate(self.color_keys):
|
|
s = s.replace("${" + c + "}", "")
|
|
return self.custom_nocolor(s)
|
|
|
|
def custom_color(self, s):
|
|
return self.custom_match.sub("\033[\\2", s)
|
|
|
|
def custom_nocolor(self, s):
|
|
return self.custom_match.sub("", s)
|
|
|
|
def get_keys(self):
|
|
return self.color_keys
|
|
|
|
def flush(self):
|
|
sys.stdout.flush()
|
|
|
|
|
|
def demo():
|
|
""" Print all control sequences """
|
|
c = code()
|
|
unformatted = """${S}ANSI CODES
|
|
==========${Z}
|
|
${S}Fo${U}rm${st}at${u}ti${ic}ng${Z}
|
|
${S}==========${Z}
|
|
0 Z Clear format
|
|
1 S ${S}Strong ${Z} 2 s ${s}Off${Z}
|
|
4 U ${U}Underline${Z} 24 u Off
|
|
7 ic ${ic}Inverse${Z} 27 io Off
|
|
9 st ${st}Strike${Z} 29 so Off
|
|
${R}Co${G}lo${B}rs${Z}
|
|
${S}======${Z}
|
|
30 k ${k}Black ${Z}1 K ${K}Strong ${Z}40 bk ${bk}Background${Z}
|
|
31 r ${r}Red ${Z}1 R ${R}Strong ${Z}41 br ${br}Background${Z}
|
|
32 g ${g}Green ${Z}1 G ${G}Strong ${Z}42 bg ${bg}Background${Z}
|
|
33 y ${y}Yellow ${Z}1 Y ${Y}Strong ${Z}43 by ${by}Background${Z}
|
|
34 b ${b}Blue ${Z}1 B ${B}Strong ${Z}44 bb ${bb}Background${Z}
|
|
35 m ${m}Magenta ${Z}1 M ${M}Strong ${Z}45 bm ${bm}Background${Z}
|
|
36 c ${c}Cyan ${Z}1 C ${C}Strong ${Z}46 bc ${bc}Background${Z}
|
|
37 w ${w}White ${Z}1 W ${W}Strong ${Z}47 bw ${bw}Background${Z}
|
|
${S}Clearing and movement
|
|
${S}=====================${Z}
|
|
2J CLR Clear screen
|
|
2J ;H CLRSCR .clear() Clear screen and cursor to upper left
|
|
K CLREND Clear to end of line
|
|
1K CLRBEG Clear to beginning of line
|
|
|
|
s .save() Save location u .restore() Restore location
|
|
A .up() Up E .up_line() Up line
|
|
B .down() Down F .down_line() Down line
|
|
C .right() Right y;xH .pos() Absolute Position
|
|
D .left() Left """
|
|
|
|
return c.color_string(unformatted)
|