luau/bench/color.py

38 lines
993 B
Python
Raw Normal View History

# This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
from enum import Enum
import sys
class Color(Enum):
DEFAULT = 0
RED = 1
GREEN = 2
BLUE = 3
YELLOW = 4
WHITE = 5
def colored_on(color:Color, message:str):
from colorama import Fore, Style
color_mappings = {
Color.DEFAULT: (Fore.WHITE, Style.NORMAL),
Color.RED: (Fore.RED, Style.NORMAL),
Color.GREEN: (Fore.GREEN, Style.NORMAL),
Color.BLUE: (Fore.BLUE, Style.BRIGHT),
Color.YELLOW: (Fore.YELLOW, Style.NORMAL),
Color.WHITE: (Fore.WHITE, Style.BRIGHT)
}
fore, style = color_mappings[color]
return fore + style + message + Style.RESET_ALL
def colored_off(color:Color, message:str):
return message
try:
if sys.stdout.isatty():
import colorama
colorama.init()
colored = colored_on
else:
colored = colored_off
except:
colored = colored_off