62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
class CLI:
|
||
|
|
NO_COLOR = os.environ.get("NO_COLOR", "")
|
||
|
|
|
||
|
|
class style:
|
||
|
|
reset = "\033[0m"
|
||
|
|
bold = "\033[01m"
|
||
|
|
disable = "\033[02m"
|
||
|
|
underline = "\033[04m"
|
||
|
|
reverse = "\033[07m"
|
||
|
|
strikethrough = "\033[09m"
|
||
|
|
invisible = "\033[08m"
|
||
|
|
|
||
|
|
class esc:
|
||
|
|
up = "\033[A"
|
||
|
|
down = "\033[B"
|
||
|
|
right = "\033[C"
|
||
|
|
left = "\033[D"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def clear_line(opt=0):
|
||
|
|
# 0=to end, 1=from start, 2=all
|
||
|
|
return "\033[" + str(opt) + "K"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def write(*text):
|
||
|
|
for t in text:
|
||
|
|
sys.stdout.write(str(t))
|
||
|
|
sys.stdout.flush()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def printline(*text):
|
||
|
|
CLI.write(*text, CLI.esc.clear_line(), "\n")
|
||
|
|
|
||
|
|
# 4bit system colors
|
||
|
|
@staticmethod
|
||
|
|
def fg4(col):
|
||
|
|
# black=0,red=1,green=2,orange=3,blue=4,purple=5,cyan=6,lightgrey=7
|
||
|
|
# darkgrey=8,lightred=9,lightgreen=10,yellow=11,lightblue=12,pink=13,lightcyan=14
|
||
|
|
if CLI.NO_COLOR:
|
||
|
|
return ""
|
||
|
|
else:
|
||
|
|
return f"\033[{(30+col) if col<8 else (90-8+col)}m"
|
||
|
|
|
||
|
|
# 8bit xterm colors
|
||
|
|
@staticmethod
|
||
|
|
def fg8(col):
|
||
|
|
if CLI.NO_COLOR:
|
||
|
|
return ""
|
||
|
|
else:
|
||
|
|
return f"\033[38;5;{col}m"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def bg8(col):
|
||
|
|
if CLI.NO_COLOR:
|
||
|
|
return ""
|
||
|
|
else:
|
||
|
|
return f"\033[48;5;{col}m"
|