← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to terminal.py

  • Committer: Aaron Bentley
  • Date: 2013-08-20 03:02:43 UTC
  • Revision ID: aaron@aaronbentley.com-20130820030243-r8v1xfbcnd8f10p4
Fix zap command for 2.6/7

Show diffs side-by-side

added added

removed removed

Lines of Context:
terminal.py
1
1
# Copyright (C) 2004 Aaron Bentley
2
 
# <aaron.bentley@utoronto.ca>
 
2
# <aaron@aaronbentley.com>
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
19
20
 
20
21
__docformat__ = "restructuredtext"
21
22
__doc__ = "Terminal control functionality"
24
25
    # XXX The whole color handling should be rewritten to use terminfo
25
26
    # XXX before we get there, checking for setaf capability should do.
26
27
    # 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
27
32
    import curses
28
 
    curses.setupterm()
 
33
    try:
 
34
        curses.setupterm()
 
35
    except curses.error:
 
36
        return False
29
37
    return bool(curses.tigetstr('setaf'))
30
38
 
31
39
colors = {
39
47
"white": "7"
40
48
}
41
49
 
42
 
def colorstring(text, fgcolor=None, bright=False, bgcolor=None):
 
50
def colorstring(text, fgcolor=None, bgcolor=None):
43
51
    """
44
52
    Returns a string using ANSI control codes to set the text color.
45
53
 
47
55
    :type text: string
48
56
    :param fgcolor: The foreground color to use
49
57
    :type fgcolor: string
50
 
    :param bright: If true, make the foreground color bright
51
 
    :type bright: bool
52
58
    :param bgcolor: The background color to use
53
59
    :type bgcolor: string
54
60
    """
55
61
    code = []
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])
 
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
 
62
75
    return "".join(("\033[", ';'.join(code), "m", text, "\033[0m"))
63
76
 
 
77
 
64
78
def term_title(title):
65
79
    term = os.environ.get('TERM', '')
66
80
    if term.startswith('xterm') or term == 'dtterm':