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)