Source code for pyro.util.msg

"""
support output in highlighted colors
"""

import sys
import traceback


# inspiration from
# http://stackoverflow.com/questions/287871/print-in-terminal-with-colors-using-python
# which in-turn cites the blender build scripts
class _TermColors:
    WARNING = '\033[33m'
    SUCCESS = '\033[32m'
    FAIL = '\033[31m'
    BOLD = '\033[1m'
    ENDC = '\033[0m'




def fail(string):
    """
    Output a string to the terminal and abort if we are running
    non-interactively.  The string is colored red to indicate a
    failure
    """
    print(_TermColors.FAIL + string + _TermColors.ENDC)
    traceback.print_stack()
    # only exit if we are not running in interactive mode.  sys.ps1 is
    # only defined in interactive mode.
    if not hasattr(sys, 'ps1'):
        sys.exit(1)





def warning(string):
    """
    Output a string to the terminal colored orange to indicate a
    warning
    """
    print(_TermColors.WARNING + string + _TermColors.ENDC)





def success(string):
    """
    Output a string to the terminal colored green to indicate
    success
    """
    print(_TermColors.SUCCESS + string + _TermColors.ENDC)





def bold(string):
    """ Output a string in a bold weight"""
    print(_TermColors.BOLD + string + _TermColors.ENDC)