mirror of
https://github.com/keanuplayz/dotfiles.git
synced 2024-08-15 02:33:12 +00:00
22 lines
623 B
Python
22 lines
623 B
Python
|
from colorama import Fore, Style, ansi
|
||
|
|
||
|
COLORS = [ansi.code_to_chars(30 + color_index) for color_index in range(0, 8)]
|
||
|
|
||
|
|
||
|
def colored(string, *colors):
|
||
|
return "".join(colors + (string, Style.RESET_ALL))
|
||
|
|
||
|
|
||
|
def bright_colored(string, *colors):
|
||
|
return "".join(colors + (Style.BRIGHT, string, Style.RESET_ALL))
|
||
|
|
||
|
|
||
|
def colorize_percent(percent, warning, critical, inverse=False):
|
||
|
COLORS = [Fore.GREEN, Fore.YELLOW, Fore.RED]
|
||
|
|
||
|
color_index = 0 if percent < warning else 1 if percent < critical else 2
|
||
|
if inverse:
|
||
|
color_index = 2 - color_index
|
||
|
|
||
|
return colored("%.2f%%" % percent, COLORS[color_index])
|