1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
# vim: set fileencoding=utf-8 :
#
# (C) 2010 Guido Guenther <agx@sigxcpu.org>
# This program 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 2 of the License, or
# (at your option) any later version.
#
# This program 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 this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
"""Simple colored logging classes"""
import os
import sys
import logging
from logging import DEBUG, INFO, WARNING, ERROR, CRITICAL
import gbp.tristate
# Initialize default logger
LOGGER = logging.getLogger(__name__)
COLORS = dict([('none', 0)] + zip(['black', 'red', 'green', 'yellow', 'blue',
'magenta', 'cyan', 'white'], range(30, 38)))
DEFAULT_COLOR_SCHEME = {DEBUG: COLORS['green'],
INFO: COLORS['green'],
WARNING: COLORS['red'],
ERROR: COLORS['red'],
CRITICAL: COLORS['red']}
class GbpFilter(object):
"""Filter for enabling selective output"""
def __init__(self, levels):
self._levels = levels
def filter(self, record):
"""Do we show the record"""
if record.levelno in self._levels:
return True
return False
class GbpStreamHandler(logging.StreamHandler):
"""Special stream handler for enabling colored output"""
COLOR_SEQ = "\033[%dm"
OFF_SEQ = "\033[0m"
def __init__(self, stream=None, color='auto'):
logging.StreamHandler.__init__(self, stream)
self._color = gbp.tristate.Tristate(color)
self._color_scheme = DEFAULT_COLOR_SCHEME.copy()
msg_fmt = "%(color)s%(name)s:%(levelname)s: %(message)s%(coloroff)s"
self.setFormatter(logging.Formatter(fmt=msg_fmt))
def set_color(self, color):
"""Set/unset colorized output"""
self._color = gbp.tristate.Tristate(color)
def set_color_scheme(self, color_scheme={}):
"""Set logging colors"""
self._color_scheme = DEFAULT_COLOR_SCHEME.copy()
self._color_scheme.update(color_scheme)
def set_format(self, fmt):
"""Set logging format"""
self.setFormatter(logging.Formatter(fmt=fmt))
def _use_color(self):
"""Check if to print in color or not"""
if self._color.is_on():
return True
elif self._color.is_auto() and hasattr(self.stream, 'isatty'):
in_emacs = (os.getenv("EMACS") and
os.getenv("INSIDE_EMACS", "").endswith(",comint"))
return self.stream.isatty() and not in_emacs
return False
def format(self, record):
"""Colorizing formatter"""
record.color = record.coloroff = ""
if self._use_color():
record.color = self.COLOR_SEQ % self._color_scheme[record.levelno]
record.coloroff = self.OFF_SEQ
record.levelname = record.levelname.lower()
return logging.StreamHandler.format(self, record)
class GbpLogger(logging.Logger):
"""Logger class for git-buildpackage"""
def __init__(self, name, *args, **kwargs):
logging.Logger.__init__(self, name, *args, **kwargs)
self.default_handlers = []
def init_default_handlers(self, color='auto'):
"""Initialize and set default handlers to logger"""
self.default_handlers = [GbpStreamHandler(sys.stdout, color),
GbpStreamHandler(sys.stderr, color)]
self.default_handlers[0].addFilter(GbpFilter([DEBUG, INFO]))
self.default_handlers[1].addFilter(GbpFilter([WARNING, ERROR,
CRITICAL]))
for hdlr in self.default_handlers:
self.addHandler(hdlr)
# We don't want to propagate as we add our own handlers
self.propagate = False
def set_color(self, color):
"""Set/unset colorized output of the default handlers"""
for hdlr in self.default_handlers:
hdlr.set_color(color)
def set_color_scheme(self, color_scheme={}):
"""Set the color scheme of the default handlers"""
for hdlr in self.default_handlers:
hdlr.set_color_scheme(color_scheme)
def set_format(self, fmt):
"""Set the format of the default handlers"""
for hdlr in self.default_handlers:
hdlr.set_format(fmt)
def err(msg):
"""Logs a message with level ERROR on the GBP logger"""
LOGGER.error(msg)
def error(msg):
err(msg)
def warn(msg):
"""Logs a message with level WARNING on the GBP logger"""
LOGGER.warning(msg)
def warning(msg):
warn(msg)
def info(msg):
"""Logs a message with level INFO on the GBP logger"""
LOGGER.info(msg)
def debug(msg):
"""Logs a message with level DEBUG on the GBP logger"""
LOGGER.debug(msg)
def _parse_color_scheme(color_scheme=""):
"""Set logging colors"""
scheme = {}
colors = color_scheme.split(':')
levels = (DEBUG, INFO, WARNING, ERROR)
if color_scheme and len(colors) != len(levels):
raise ValueError("Number color fields in color scheme not %d'"
% len(levels))
for field, color in enumerate(colors):
level = levels[field]
try:
scheme[level] = int(color)
except ValueError:
try:
scheme[level] = COLORS[color.lower()]
except KeyError: pass
return scheme
def getLogger(*args, **kwargs):
"""Gbp-specific function"""
if not issubclass(logging.getLoggerClass(), GbpLogger):
logging.setLoggerClass(GbpLogger)
color = kwargs.pop('color') if 'color' in kwargs else 'auto'
logger = logging.getLogger(*args, **kwargs)
if not logger.default_handlers:
logger.init_default_handlers(color)
return logger
def setup(color, verbose, color_scheme=""):
"""Basic logger setup"""
# Initialize, if not done yet
if not isinstance(LOGGER, GbpLogger):
initialize()
LOGGER.set_color(color)
LOGGER.set_color_scheme(_parse_color_scheme(color_scheme))
if verbose:
LOGGER.setLevel(DEBUG)
else:
LOGGER.setLevel(INFO)
def initialize():
"""Initialize the logger module"""
global LOGGER
LOGGER = getLogger("gbp")
|
