fix table_def
[rainbowstream.git] / rainbowstream / colors.py
CommitLineData
2a6238f5
O
1import random
2import itertools
8c840a83 3from functools import wraps
8c840a83 4from pyfiglet import figlet_format
632c6fa5 5from .config import *
c075e6dc 6from functools import reduce
8c840a83 7
6fa09c14
VNM
8
9def basic_color(code):
10 """
11 16 colors supported
12 """
13 def inner(text, bold=True):
533edcdb
VNM
14 c = code
15 if bold:
16 c = "1;%s" % c
17 return "\033[%sm%s\033[0m" % (c, text)
18 return inner
8c840a83 19
6fa09c14
VNM
20
21def RGB(code):
22 """
23 256 colors supported
24 """
93384849 25 def inner(text):
6fa09c14 26 c = code
6fa09c14
VNM
27 return "\033[38;5;%sm%s\033[0m" % (c, text)
28 return inner
29
30
c075e6dc
O
31default = basic_color('39')
32black = basic_color('30')
33red = basic_color('31')
34green = basic_color('32')
35yellow = basic_color('33')
36blue = basic_color('34')
37magenta = basic_color('35')
38cyan = basic_color('36')
39grey = basic_color('90')
40light_red = basic_color('91')
41light_green = basic_color('92')
42light_yellow = basic_color('93')
43light_blue = basic_color('94')
44light_magenta = basic_color('95')
45light_cyan = basic_color('96')
46white = basic_color('97')
6fa09c14 47
c075e6dc
O
48on_default = basic_color('49')
49on_black = basic_color('40')
50on_red = basic_color('41')
51on_green = basic_color('42')
52on_yellow = basic_color('43')
53on_blue = basic_color('44')
54on_magenta = basic_color('45')
55on_cyan = basic_color('46')
56on_grey = basic_color('100')
57on_light_red = basic_color('101')
58on_light_green = basic_color('102')
59on_light_yellow = basic_color('103')
60on_light_blue = basic_color('104')
6fa09c14 61on_light_magenta = basic_color('105')
c075e6dc
O
62on_light_cyan = basic_color('106')
63on_white = basic_color('107')
533edcdb 64
c075e6dc
O
65colors_shuffle = [locals()[i.encode('utf8')] if not i.startswith(
66 'RGB_') else RGB(int(i[4:])) for i in c['CYCLE_COLOR']]
632c6fa5 67
2a6238f5
O
68background_shuffle = [
69 on_grey,
533edcdb
VNM
70 on_light_red,
71 on_light_green,
72 on_light_yellow,
73 on_light_blue,
74 on_light_magenta,
75 on_light_cyan]
668db21c 76cyc = itertools.cycle(colors_shuffle[1:])
8c840a83
O
77
78
79def order_rainbow(s):
80 """
81 Print a string with ordered color with each character
82 """
2a6238f5
O
83 c = [colors_shuffle[i % 7](s[i]) for i in xrange(len(s))]
84 return reduce(lambda x, y: x + y, c)
85
8c840a83
O
86
87def random_rainbow(s):
88 """
89 Print a string with random color with each character
90 """
668db21c 91 c = [random.choice(colors_shuffle)(i) for i in s]
2a6238f5
O
92 return reduce(lambda x, y: x + y, c)
93
8c840a83
O
94
95def Memoize(func):
96 """
97 Memoize decorator
98 """
99 cache = {}
2a6238f5 100
8c840a83
O
101 @wraps(func)
102 def wrapper(*args):
103 if args not in cache:
104 cache[args] = func(*args)
105 return cache[args]
106 return wrapper
107
2a6238f5 108
8c840a83
O
109@Memoize
110def cycle_color(s):
111 """
668db21c 112 Cycle the colors_shuffle
8c840a83
O
113 """
114 return next(cyc)(s)
115
2a6238f5 116
42fde775 117def ascii_art(text):
8c840a83
O
118 """
119 Draw the Ascii Art
120 """
42fde775 121 fi = figlet_format(text, font='doom')
8c840a83 122 print('\n'.join(
2a6238f5
O
123 [next(cyc)(i) for i in fi.split('\n')]
124 )
8c840a83 125 )