dotfiles/zsh/welcome/colors.py

22 lines
623 B
Python
Raw Normal View History

2018-08-09 10:35:09 +00:00
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])