task.mk/src/color.py

54 lines
1.2 KiB
Python
Raw Normal View History

2022-09-05 12:23:17 -05:00
#% extends "py-script.mk" %#
#% block name %#color#% endblock %#
#% block script %#
import os
import sys
color2byte = dict(
black=0,
red=1,
green=2,
yellow=3,
blue=4,
magenta=5,
cyan=6,
white=7,
)
state2byte = dict(
bold=1, faint=2, italic=3, underline=4, blink=5, fast_blink=6, crossed=9
)
def fg(byte):
return 30 + byte
def bg(byte):
return 40 + byte
class Colors:
"""ANSI color codes"""
def setcolor(self, name, escape_code):
if not sys.stdout.isatty() or os.getenv("NO_COLOR", False):
setattr(self, name, "")
else:
setattr(self, name, escape_code)
def __init__(self):
self.setcolor("end", "\033[0m")
for name, byte in color2byte.items():
self.setcolor(name, f"\033[{fg(byte)}m")
self.setcolor(f"b_{name}", f"\033[1;{fg(byte)}m")
self.setcolor(f"d_{name}", f"\033[2;{fg(byte)}m")
for bgname, bgbyte in color2byte.items():
self.setcolor(f"{name}_on_{bgname}", f"\033[{bg(bgbyte)};{fg(byte)}m")
for name, byte in state2byte.items():
self.setcolor(name, f"\033[{byte}m")
2022-09-05 13:03:53 -05:00
c = color = Colors()
2022-09-05 12:23:17 -05:00
#% endblock %#