refactoring and add various comments
[rainbowstream.git] / rainbowstream / colors.py
index 909a85af3e5fc0bcc9fc32dad41d0935ff4c70c7..bdbc4b44c14d6af7c4009af898b13ed3e86bd44f 100644 (file)
@@ -1,11 +1,3 @@
-import random
-import itertools
-from functools import wraps
-from pyfiglet import figlet_format
-from .config import *
-from functools import reduce
-
-
 def basic_color(code):
     """
     16 colors supported
 def basic_color(code):
     """
     16 colors supported
@@ -18,7 +10,7 @@ def basic_color(code):
     return inner
 
 
     return inner
 
 
-def RGB(code):
+def term_color(code):
     """
     256 colors supported
     """
     """
     256 colors supported
     """
@@ -28,6 +20,9 @@ def RGB(code):
     return inner
 
 
     return inner
 
 
+"""
+16 basic colors
+"""
 default = basic_color('39')
 black = basic_color('30')
 red = basic_color('31')
 default = basic_color('39')
 black = basic_color('30')
 red = basic_color('31')
@@ -45,6 +40,9 @@ light_magenta = basic_color('95')
 light_cyan = basic_color('96')
 white = basic_color('97')
 
 light_cyan = basic_color('96')
 white = basic_color('97')
 
+"""
+16 basic colors on background
+"""
 on_default = basic_color('49')
 on_black = basic_color('40')
 on_red = basic_color('41')
 on_default = basic_color('49')
 on_black = basic_color('40')
 on_red = basic_color('41')
@@ -61,65 +59,3 @@ on_light_blue = basic_color('104')
 on_light_magenta = basic_color('105')
 on_light_cyan = basic_color('106')
 on_white = basic_color('107')
 on_light_magenta = basic_color('105')
 on_light_cyan = basic_color('106')
 on_white = basic_color('107')
-
-colors_shuffle = [locals()[i.encode('utf8')] if not i.startswith(
-    'RGB_') else RGB(int(i[4:])) for i in c['CYCLE_COLOR']]
-
-background_shuffle = [
-    on_grey,
-    on_light_red,
-    on_light_green,
-    on_light_yellow,
-    on_light_blue,
-    on_light_magenta,
-    on_light_cyan]
-cyc = itertools.cycle(colors_shuffle[1:])
-
-
-def order_rainbow(s):
-    """
-    Print a string with ordered color with each character
-    """
-    c = [colors_shuffle[i % 7](s[i]) for i in xrange(len(s))]
-    return reduce(lambda x, y: x + y, c)
-
-
-def random_rainbow(s):
-    """
-    Print a string with random color with each character
-    """
-    c = [random.choice(colors_shuffle)(i) for i in s]
-    return reduce(lambda x, y: x + y, c)
-
-
-def Memoize(func):
-    """
-    Memoize decorator
-    """
-    cache = {}
-
-    @wraps(func)
-    def wrapper(*args):
-        if args not in cache:
-            cache[args] = func(*args)
-        return cache[args]
-    return wrapper
-
-
-@Memoize
-def cycle_color(s):
-    """
-    Cycle the colors_shuffle
-    """
-    return next(cyc)(s)
-
-
-def ascii_art(text):
-    """
-    Draw the Ascii Art
-    """
-    fi = figlet_format(text, font='doom')
-    print('\n'.join(
-        [next(cyc)(i) for i in fi.split('\n')]
-    )
-    )