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