comparison src/tools/common/ansi.py @ 2154:7cbffd754b4a

core (constants), tools (common/ansi): moved ANSI escape codes to a dedicated class, with helper methods
author Goffi <goffi@goffi.org>
date Thu, 16 Feb 2017 00:44:54 +0100
parents
children a0fbb2b11fd0
comparison
equal deleted inserted replaced
2153:f67434fd88d2 2154:7cbffd754b4a
1 #!/usr/bin/env python2
2 # -*- coding: utf-8 -*-
3
4 # SàT: a XMPP client
5 # Copyright (C) 2009-2016 Jérôme Poisson (goffi@goffi.org)
6
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
11
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU Affero General Public License for more details.
16
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
19
20 import sys
21
22 class ANSI(object):
23
24 ## ANSI escape sequences ##
25 RESET = '\033[0m'
26 NORMAL_WEIGHT = '\033[22m'
27 FG_BLACK, FG_RED, FG_GREEN, FG_YELLOW, FG_BLUE, FG_MAGENTA, FG_CYAN, FG_WHITE = ('\033[3%dm' % nb for nb in xrange(8))
28 BOLD = '\033[1m'
29 BLINK = '\033[5m'
30 BLINK_OFF = '\033[25m'
31
32 @classmethod
33 def color(cls, *args):
34 """output text using ANSI codes
35
36 this method simply merge arguments, and add RESET if is not the last arguments
37 """
38 # XXX: we expect to have at least on argument
39 if args[-1] != cls.RESET:
40 args = list(args)
41 args.append(cls.RESET)
42 return u''.join(args)
43
44
45 if not sys.stdout.isatty():
46 # we don't want ANSI escape codes if we are not outputing to a tty!
47 for attr in dir(ANSI):
48 if isinstance(getattr(ANSI, attr), basestring):
49 setattr(ANSI, attr, u'')