101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Cross-platform color text printing
|
|
|
|
Based on colorama (see pyqtgraph/util/colorama/README.txt)
|
|
"""
|
|
import sys
|
|
|
|
from .colorama.winterm import WinTerm, WinColor, WinStyle
|
|
from .colorama.win32 import windll
|
|
|
|
_WIN = sys.platform.startswith('win')
|
|
if windll is not None:
|
|
winterm = WinTerm()
|
|
else:
|
|
_WIN = False
|
|
|
|
def winset(reset=False, fore=None, back=None, style=None, stderr=False):
|
|
if reset:
|
|
winterm.reset_all()
|
|
if fore is not None:
|
|
winterm.fore(fore, stderr)
|
|
if back is not None:
|
|
winterm.back(back, stderr)
|
|
if style is not None:
|
|
winterm.style(style, stderr)
|
|
|
|
ANSI = {}
|
|
WIN = {}
|
|
for i,color in enumerate(['BLACK', 'RED', 'GREEN', 'YELLOW', 'BLUE', 'MAGENTA', 'CYAN', 'WHITE']):
|
|
globals()[color] = i
|
|
globals()['BR_' + color] = i + 8
|
|
globals()['BACK_' + color] = i + 40
|
|
ANSI[i] = "\033[%dm" % (30+i)
|
|
ANSI[i+8] = "\033[2;%dm" % (30+i)
|
|
ANSI[i+40] = "\033[%dm" % (40+i)
|
|
color = 'GREY' if color == 'WHITE' else color
|
|
WIN[i] = {'fore': getattr(WinColor, color), 'style': WinStyle.NORMAL}
|
|
WIN[i+8] = {'fore': getattr(WinColor, color), 'style': WinStyle.BRIGHT}
|
|
WIN[i+40] = {'back': getattr(WinColor, color)}
|
|
|
|
RESET = -1
|
|
ANSI[RESET] = "\033[0m"
|
|
WIN[RESET] = {'reset': True}
|
|
|
|
|
|
def cprint(stream, *args, **kwds):
|
|
"""
|
|
Print with color. Examples::
|
|
|
|
# colors are BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE
|
|
cprint('stdout', RED, 'This is in red. ', RESET, 'and this is normal\n')
|
|
|
|
# Adding BR_ before the color manes it bright
|
|
cprint('stdout', BR_GREEN, 'This is bright green.\n', RESET)
|
|
|
|
# Adding BACK_ changes background color
|
|
cprint('stderr', BACK_BLUE, WHITE, 'This is white-on-blue.', -1)
|
|
|
|
# Integers 0-7 for normal, 8-15 for bright, and 40-47 for background.
|
|
# -1 to reset.
|
|
cprint('stderr', 1, 'This is in red.', -1)
|
|
|
|
"""
|
|
if isinstance(stream, str):
|
|
stream = kwds.get('stream', 'stdout')
|
|
err = stream == 'stderr'
|
|
stream = getattr(sys, stream)
|
|
else:
|
|
err = kwds.get('stderr', False)
|
|
|
|
if hasattr(stream, 'isatty') and stream.isatty():
|
|
if _WIN:
|
|
# convert to win32 calls
|
|
for arg in args:
|
|
if isinstance(arg, str):
|
|
stream.write(arg)
|
|
else:
|
|
kwds = WIN[arg]
|
|
winset(stderr=err, **kwds)
|
|
else:
|
|
# convert to ANSI
|
|
for arg in args:
|
|
if isinstance(arg, str):
|
|
stream.write(arg)
|
|
else:
|
|
stream.write(ANSI[arg])
|
|
else:
|
|
# ignore colors
|
|
for arg in args:
|
|
if isinstance(arg, str):
|
|
stream.write(arg)
|
|
|
|
def cout(*args):
|
|
"""Shorthand for cprint('stdout', ...)"""
|
|
cprint('stdout', *args)
|
|
|
|
def cerr(*args):
|
|
"""Shorthand for cprint('stderr', ...)"""
|
|
cprint('stderr', *args)
|