Andrew Hamilton
e1e7bf8054
May as well hard code them since only supporting gnome-terminal and stterm atm. Only one code differed between stterm & gnome-terminal, "terminal.normal", and have taken the gnome-terminal version for now. stterm seems to work fine with gnome-terminal's version.
88 lines
2 KiB
Python
88 lines
2 KiB
Python
|
|
# Copyright (C) 2015-2016 Andrew Hamilton. All rights reserved.
|
|
# Licensed under the Artistic License 2.0.
|
|
|
|
|
|
import contextlib
|
|
import sys
|
|
|
|
|
|
_PREFIX = "\x1b"
|
|
|
|
|
|
normal = _PREFIX + "(B\x1b[m" # sgr0 "[0m" ?
|
|
bold = _PREFIX + "[1m" # bold
|
|
italic = _PREFIX + "[3m" # sitm
|
|
standout = _PREFIX + "[7m" # smso
|
|
underline = _PREFIX + "[4m" # smul
|
|
enter_fullscreen = _PREFIX + "[?1049h" # smcup
|
|
exit_fullscreen = _PREFIX + "[?1049l" # rmcup
|
|
hide_cursor = _PREFIX + "[?25l" # civis
|
|
normal_cursor = _PREFIX + "[?25l\x1b[?25h" # cnorm
|
|
clear = _PREFIX + "[H\x1b[2J" # clear
|
|
save = _PREFIX + "7" # sc
|
|
restore = _PREFIX + "8" # rc
|
|
|
|
|
|
_FG_CODES = ["30", "31", "32", "33", "34", "35", "36", "37",
|
|
"90", "91", "92", "93", "94", "95", "96", "97"]
|
|
|
|
|
|
def fg_color(color_number): # setaf
|
|
return ("\x1b[38;5;%im" % color_number if color_number > 15
|
|
else "\x1b[%sm" % _FG_CODES[color_number])
|
|
|
|
|
|
_BG_CODES = ["40", "41", "42", "43", "44", "45", "46", "47",
|
|
"100", "101", "102", "103", "104", "105", "106", "107"]
|
|
|
|
|
|
def bg_color(color_number): # setab
|
|
return ("\x1b[48;5;%im" % color_number if color_number > 15
|
|
else "\x1b[%sm" % _BG_CODES[color_number])
|
|
|
|
|
|
def fg_rgb_color(rgb):
|
|
return "\x1b[38;2;%i;%i;%im" % rgb
|
|
|
|
|
|
def bg_rgb_color(rgb):
|
|
return "\x1b[48;2;%i;%i;%im" % rgb
|
|
|
|
|
|
def move(x, y): # cup
|
|
return "\x1b[%i;%iH" % (y + 1, x + 1)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def fullscreen():
|
|
if enter_fullscreen is None:
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(clear)
|
|
else:
|
|
sys.stdout.write(enter_fullscreen)
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(exit_fullscreen)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def hidden_cursor():
|
|
sys.stdout.write(hide_cursor)
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(normal_cursor)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def console_title(title):
|
|
sys.stdout.write(save)
|
|
sys.stdout.write("\033]0;%s\007" % title)
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(restore)
|