mirror of
https://github.com/openSUSE/osc.git
synced 2024-11-14 16:26:13 +01:00
39 lines
746 B
Python
39 lines
746 B
Python
import os
|
|
import sys
|
|
|
|
|
|
IS_INTERACTIVE = os.isatty(sys.stdout.fileno())
|
|
|
|
|
|
ESCAPE_CODES = {
|
|
"reset": "\033[0m",
|
|
"bold": "\033[1m",
|
|
"underline": "\033[4m",
|
|
"black": "\033[30m",
|
|
"red": "\033[31m",
|
|
"green": "\033[32m",
|
|
"yellow": "\033[33m",
|
|
"blue": "\033[34m",
|
|
"magenta": "\033[35m",
|
|
"cyan": "\033[36m",
|
|
"white": "\033[37m",
|
|
}
|
|
|
|
|
|
def colorize(text, color):
|
|
"""
|
|
Colorize `text` if the `color` is specified and we're running in an interactive terminal.
|
|
"""
|
|
if not IS_INTERACTIVE:
|
|
return text
|
|
|
|
if not color:
|
|
return text
|
|
|
|
result = ""
|
|
for i in color.split(","):
|
|
result += ESCAPE_CODES[i]
|
|
result += text
|
|
result += ESCAPE_CODES["reset"]
|
|
return result
|