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     )