# -*- coding: utf-8 -*-

(BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE, BOLD) = range(0,9)

names = ['black','red','green','yellow','blue','magenta','cyan','white']

disabled = False  # set True to disable all colors

def color(enable, fg=None, bg=None):

    '''

    color(1) -- switch to bold

    color(1,1) -- red foreground

    color(1,3,4) -- red on blue

    color(0) -- reset

    '''

    global disabled

    if disabled:

        return ''

    if isinstance(fg, str):

        fg = names.index(fg)

    if isinstance(bg, str):

        bg = names.index(bg)

    if enable:

        code = '1'

        if fg is not None:

            code = ''

            if fg >= 8:

                fg -= 8

                code += '1;'

            code += str(30 + fg)

        if bg is not None:

            code += ';'

            if bg >= 8:

                bg -= 8

                code += '1;'

            code += str(40 + bg)

        return "\033[" + code + "m"

    else:

        return "\033[0m"

def setcolor(enable, fg=None, bg=None):

    print(color(enable, fg, bg), end='')

if __name__ == '__main__':

    for c in range(0,8):

        print(

            color(1,c), names[c].ljust(20),

            color(1,8+c), names[c].ljust(20),

            color(0), sep='')