testsuite/driver/term_color.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

from enum import Enum

enable_color = True

class Color(Enum):
    BLACK   = 30
    RED     = 31
    GREEN   = 32
    YELLOW  = 33
    BLUE    = 34
    MAGENTA = 35
    CYAN    = 36
    WHITE   = 37

def colored(color: Color, s: str) -> str:
    if enable_color:
        return '\033[1m\033[{}m{}\033[0m'.format(color.value, s)
    else:
        return s