''' Set ANSI Terminal Color and Attributes. ''' from sys import stdout esc = '%s['%chr(27) reset = '%s0m'%esc format = '1;%dm' fgoffset, bgoffset = 30, 40 for k, v in dict( attrs = 'none bold faint italic underline blink fast reverse concealed', colors = 'grey red green yellow blue magenta cyan white' ).items(): globals()[k]=dict((s,i) for i,s in enumerate(v.split())) def echo(arg=None, sep=' ', end='\n'): ''' "arg" is a string or None if "arg" is None : the terminal is reset to his default values. if "arg" is a string it must contain "sep" separated values. if args are found in globals "attrs" or "colors", or start with "@" \ they are interpreted as ANSI commands else they are output as text. colors, if any, must be first (foreground first then background) you can not specify a background color alone ; \ if you specify only one color, it will be the foreground one. @* commands handle the screen and the cursor : @x;y : go to xy @ : go to 1;1 @@ : clear screen and go to 1;1 examples: echo('red') : set red as the foreground color echo('red blue') : red on blue echo('red blink') : blinking red echo() : restore terminal default values echo('reverse') : swap default colors echo('cyan blue reverse') : blue on cyan <=> echo('blue cyan) echo('red reverse') : a way to set up the background only echo('red reverse blink') : you can specify any combinaison of \ attributes in any order with or without colors echo('blink Python') : output a blinking 'Python' echo('@@ hello') : clear the screen and print 'hello' at 1;1 ''' cmd, txt = [reset], [] if arg: arglist=arg.split(sep) for offset in (fgoffset, bgoffset): if arglist and arglist[0] in colors: cmd.append(format % (colors[arglist.pop(0)]+offset)) for a in arglist: c=format % attrs[a] if a in attrs else None if c and c not in cmd: cmd.append(c) else: if a.startswith('@'): a=a[1:] if a=='@': cmd.append('2J') cmd.append('H') else: cmd.append('%sH'%a) else: txt.append(a) if txt and end: txt[-1]+=end stdout.write(esc.join(cmd)+sep.join(txt)) if __name__ == '__main__': echo('@@ reverse blink') print 'reverse blink default colors at 1;1 on a cleared screen' echo('red') print 'red' echo('red blue') print 'red blue' echo('yellow blink') print 'yellow blink' echo('default') echo('cyan blue cyan blue') echo('cyan blue reverse cyan blue reverse') echo('blue cyan blue cyan') echo('red reverse red reverse') echo('yellow red yellow on red 1') echo('yellow,red,yellow on red 2', sep=',') print 'yellow on red 3' print for bg in colors: echo(bg.title().center(8), sep='.', end='') for fg in colors: att=[fg, bg] if fg==bg: att.append('blink') att.append(fg.center(8)) echo(','.join(att), sep=',', end='') print print for att in attrs: echo('%s,%s' % (att, att.title().center(10)), sep=',', end='') print from time import sleep, strftime, gmtime colist='grey blue cyan white cyan blue'.split() while True: try: for c in colist: sleep(.1) echo('%s @28;33 hit ctrl-c to quit' % c) echo('yellow @6;66 %s' % strftime('%H:%M:%S', gmtime())) except KeyboardInterrupt: break except: raise echo('@30;1') print