root/cleverbox/trunk/cleverbox/utils/styles.py

# This file is part of the "Cleverbox" program.
#
# Cleverbox is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Cleverbox is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Cleverbox.  If not, see <http://www.gnu.org/licenses/>.
#
# Copyright 2008, Tristan Rivoallan

import sys
from cleverbox.utils import termcolors

# Set up the terminal color scheme.
class dummy: pass
style = dummy()
style.ERROR = termcolors.make_style(fg='red', opts=('bold',))
style.WARNING = termcolors.make_style(fg='yellow', opts=('bold',))
style.SUCCESS = termcolors.make_style(fg='green', opts=('bold',))
style.H1 = termcolors.make_style(fg='white', opts=('bold', 'underline'))
style.ENABLED = termcolors.make_style(fg='green', opts=('bold',))
style.DISABLED = termcolors.make_style(fg='red')

style.SUCCESSNORESET = termcolors.make_style(fg='green', opts=('bold', 'noreset',))
style.RESET = termcolors.make_style()
del dummy

def disable_termcolors():
    class dummy:
        def __getattr__(self, attr):
            return lambda x: x
    global style
    style = dummy()

# Disable terminal coloring on Windows, Pocket PC, or if somebody's piping the output.
if sys.platform == 'win32' or sys.platform == 'Pocket PC' or not sys.stdout.isatty():
    disable_termcolors()