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