mirror of
https://github.com/luau-lang/luau.git
synced 2024-11-15 14:25:44 +08:00
38 lines
993 B
Python
38 lines
993 B
Python
|
# 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
|