Initial commit

terminal.py

@@ -0,0 +1,112 @@
+
+# Copyright (C) 2015 Andrew Hamilton. All rights reserved.
+# Licensed under the Artistic License 2.0.
+
+
+import contextlib
+import curses
+import os
+import select
+import sys
+import termios
+import tty
+
+import urwid
+import urwid.raw_display
+
+
+curses.setupterm(os.environ.get("TERM", "unknown"), sys.stdout.fileno())
+
+
+def _get_code(capability):
+    code = curses.tigetstr(capability)
+    return code.decode("latin1") if code is not None else code
+
+
+normal = _get_code("sgr0")
+bold = _get_code("bold")
+italic = _get_code("sitm")
+shadow = _get_code("sshm")
+standout = _get_code("smso")
+subscript = _get_code("ssubm")
+superscript = _get_code("ssupm")
+underline = _get_code("smul")
+enter_fullscreen = _get_code("smcup")
+exit_fullscreen = _get_code("rmcup")
+hide_cursor = _get_code("civis")
+normal_cursor = _get_code("cnorm")
+clear = _get_code("clear")
+save = _get_code("sc")
+restore = _get_code("rc")
+# reverse:rev, blink:blink, dim:dim, flash:flash
+
+
+_fg_color = curses.tigetstr("setaf")
+_bg_color = curses.tigetstr("setab")
+_move = curses.tigetstr("cup")
+
+
+def fg_color(color_number):
+    return curses.tparm(_fg_color, color_number).decode("latin1")
+
+
+def bg_color(color_number):
+    return curses.tparm(_bg_color, color_number).decode("latin1")
+
+
+def fg_rgb_color(rgb):
+    # Is there a better way?
+    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):
+    return curses.tparm(_move, y, x).decode("latin1")
+
+
+@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)
+
+
+@contextlib.contextmanager
+def urwid_screen():
+    screen = urwid.raw_display.Screen()
+    screen.set_mouse_tracking(True)
+    screen.start()
+    try:
+        yield screen
+    finally:
+        screen.stop()