blob: 0ac4ecca2c96bb7331a6cf4478d0bcece7a03f3e [file] [log] [blame]
# COPYRIGHT LINE: FIXME
"""
dbsign.ansi
"""
from __future__ import print_function
import sys
def ANSI(color, msg): # type: (str, str) -> str
if sys.stdout.isatty():
pre, post = _ANSI_CODES[color], _ANSI_CODES['clear']
msg = "{}{}{}".format(pre, msg, post)
return msg
def OK(msg): # type: (str) -> str
return ANSI('green', msg)
def INFO(msg): # type: (str) -> str
return ANSI('blue', msg)
def WARN(msg): # type: (str) -> str
return ANSI('purple', msg)
def ERROR(msg): # type: (str) -> str
return ANSI('red', msg)
_ANSI_CODES = {
'clear': '\033[0m',
'blue': '\033[1;34m',
'green': '\033[1;32m',
'purple': '\033[1;35m',
'red': '\033[1;31m',
}