blob: 9a345800832baa8c37da0d6839e5c4e327aaeed4 [file] [log] [blame]
import sys
class Terminal(object):
COLOR = {
'red': '31',
'green': '32',
'blue': '34',
'gray': '37'
}
NORMAL_INTENSITY = '1'
BRIGHT_INTENSITY = '2'
ESCAPE = '\x1b['
RESET = '0'
SEPARATOR = ';'
COLOR_CODE = 'm'
CLEAR_RIGHT_CODE = 'K'
@classmethod
def set_color(cls, color):
"""
color: str - color definition string
"""
mod = Terminal.NORMAL_INTENSITY
if color.startswith('bright'):
mod = Terminal.BRIGHT_INTENSITY
color = color[len('bright'):]
color_code = Terminal.COLOR[color]
sys.stdout.write(cls.ESCAPE + color_code + cls.SEPARATOR + mod
+ cls.COLOR_CODE)
@classmethod
def reset_color(cls):
sys.stdout.write(cls.ESCAPE + cls.RESET + cls.COLOR_CODE)
@classmethod
def clear_right(cls):
sys.stdout.write(cls.ESCAPE + cls.CLEAR_RIGHT_CODE)