← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to terminal.py

  • Committer: Aaron Bentley
  • Date: 2006-02-01 04:28:35 UTC
  • Revision ID: aaron.bentley@utoronto.ca-20060201042835-0ff0d7af60662f20
Fixed graph-ancestry

Show diffs side-by-side

added added

removed removed

Lines of Context:
terminal.py
1
1
# Copyright (C) 2004 Aaron Bentley
2
 
# <aaron@aaronbentley.com>
 
2
# <aaron.bentley@utoronto.ca>
3
3
#
4
4
#    This program is free software; you can redistribute it and/or modify
5
5
#    it under the terms of the GNU General Public License as published by
16
16
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
17
 
18
18
import os
19
 
import sys
20
19
 
21
20
__docformat__ = "restructuredtext"
22
21
__doc__ = "Terminal control functionality"
25
24
    # XXX The whole color handling should be rewritten to use terminfo
26
25
    # XXX before we get there, checking for setaf capability should do.
27
26
    # XXX See terminfo(5) for all the gory details.
28
 
    if sys.platform == "win32":
29
 
        return False
30
 
    if not sys.stdout.isatty():
31
 
        return False
32
27
    import curses
33
 
    try:
34
 
        curses.setupterm()
35
 
    except curses.error:
36
 
        return False
 
28
    curses.setupterm()
37
29
    return bool(curses.tigetstr('setaf'))
38
30
 
39
31
colors = {
47
39
"white": "7"
48
40
}
49
41
 
50
 
def colorstring(text, fgcolor=None, bgcolor=None):
 
42
def colorstring(text, fgcolor=None, bright=False, bgcolor=None):
51
43
    """
52
44
    Returns a string using ANSI control codes to set the text color.
53
45
 
55
47
    :type text: string
56
48
    :param fgcolor: The foreground color to use
57
49
    :type fgcolor: string
 
50
    :param bright: If true, make the foreground color bright
 
51
    :type bright: bool
58
52
    :param bgcolor: The background color to use
59
53
    :type bgcolor: string
60
54
    """
61
55
    code = []
62
 
 
63
 
    if fgcolor:
64
 
        if fgcolor.startswith('dark'):
65
 
            code.append('0')
66
 
            fgcolor = fgcolor[4:]
67
 
        else:
68
 
            code.append('1')
69
 
 
70
 
        code.append('3' + colors[fgcolor])
71
 
 
72
 
    if bgcolor:
73
 
        code.append('4' + colors[bgcolor])
74
 
 
 
56
    if (bright): 
 
57
        code.append("1")
 
58
    if (fgcolor): 
 
59
        code.append('3'+colors[fgcolor])
 
60
    if (bgcolor): 
 
61
        code.append("4"+colors[bgcolor])
75
62
    return "".join(("\033[", ';'.join(code), "m", text, "\033[0m"))
76
63
 
77
 
 
78
64
def term_title(title):
79
65
    term = os.environ.get('TERM', '')
80
66
    if term.startswith('xterm') or term == 'dtterm':