display.py

   1 import sys
   2 import curses
   3
   4 curses.setupterm()
   5
   6 cursor_pos = 0
   7 n_cols = curses.tigetnum('cols') or 80
   8
   9 COLOURS = {
  10         'BLACK'         : 0,
  11         'BLUE'          : 1,
  12         'GREEN'         : 2,
  13         'CYAN'          : 3,
  14         'RED'           : 4,
  15         'MAGENTA'       : 5,
  16         'YELLOW'        : 6,
  17         'WHITE'         : 7,
  18 }
  19
  20 set_fg = curses.tigetstr('setf') or ''
  21 normal = curses.tigetstr('sgr0') or ''
  22
  23 def checking(msg, indent = 2):
  24         global cursor_pos
  25         if cursor_pos:
  26                 result('!', 'RED')
  27         msg = (' ' * indent) + msg
  28         cursor_pos = len(msg)
  29         sys.stdout.write(msg)
  30         sys.stdout.flush()
  31
  32 def result(msg, colour = 'GREEN'):
  33         global cursor_pos
  34         result_col = n_cols - max(15, len(msg) + 5)
  35         if cursor_pos > result_col:
  36                 print
  37                 cursor_pos = 0
  38         if colour:
  39                 msg = highlight(msg, colour)
  40         print " " * (result_col - cursor_pos), "[ %s ]" % msg
  41         cursor_pos = 0
  42
  43 def error(msg):
  44         result(msg, 'RED')
  45
  46 def error_new_line(msg, colour = 'RED'):
  47         if cursor_pos:
  48                 error('ERROR')
  49         print highlight(msg, colour)
  50
  51 def highlight(msg, colour):
  52         return curses.tparm(set_fg, COLOURS[colour]) + msg + curses.tparm(normal)