# support Python 2.6+

from __future__ import print_function

#: Set disabled = True to disable all colors.

disabled = False

#: Supported colors as enum and string names.

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

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

    """Return ANSI escape code for selected color and font style.

    Set `enable` to true value to enable highlighting, false to disable/end highlighting.

    Set color and style using `fg` and `bg`. These parameters will accept

    string or variables defined in this module (BLACK, ...).

    Usage example::

        highlight(1) -- switch to bold, keep previous color

        highlight(1, 'bold red') -- bold red foreground

        highlight(1, BOLD | YELLOW, BLUE) -- yellow on blue, bold

        highlight(0) -- reset

    """

        fg = sum(names.index(item.lower()) for item in fg.split())

        bg = sum(names.index(item.lower()) for item in bg.split())

def hprint(*args, **kwargs):

    """Highlighted print.

    Usage is same as for print function.

    Whole output is colored as specified by `fg`, `bg` parameters.

    Usage example::

        hprint('Hello in red!', fg='bold red')

    """

    fg = kwargs.pop('fg', None)

    bg = kwargs.pop('bg', None)

    end = kwargs.pop('end', '\n')

    print(highlight(1, fg, bg), end='')

    print(*args, end='', **kwargs)

    print(highlight(0), end=end)