File size: 908 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import re
_colour_codes = {
"black": "\x1b[30m",
"red": "\x1b[31m",
"green": "\x1b[32m",
"yellow": "\x1b[33m",
"blue": "\x1b[34m",
"magenta": "\x1b[35m",
"cyan": "\x1b[36m",
"white": "\x1b[37m",
}
def ansi_format(text: str, fg_colour: str, bold: bool) -> str:
"""Formats `text` with a foreground colour `fg_colour`, and optionally mark it
`bold`, using ANSI colour codes.
"""
try:
colour_code = _colour_codes[fg_colour]
except KeyError as e:
raise ValueError(
f"Colour not recognised. Valid colours are {tuple(_colour_codes.keys())}"
) from e
out = f"{colour_code}{text}\x1b[0m"
if bold:
out = "\x1b[1m" + out
return out
_ansi_regex = re.compile(r"\x1b\[[;?0-9]*[a-zA-Z]")
def ansi_strip(text: str) -> str:
"""Removes all ANSI codes from a string."""
return _ansi_regex.sub("", text)
|