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
|
# -*- coding: utf-8 -*-
import logging
import sys
import click
_ctx = click.get_current_context
LOGGER_KEY = __name__ + '.logger'
DEFAULT_LEVEL = logging.INFO
PY2 = sys.version_info[0] == 2
if PY2:
text_type = unicode # noqa
else:
text_type = str
def _meta():
return _ctx().meta.setdefault(LOGGER_KEY, {})
class ColorFormatter(logging.Formatter):
colors = {
'error': dict(fg='red'),
'exception': dict(fg='red'),
'critical': dict(fg='red'),
'debug': dict(fg='blue'),
'warning': dict(fg='yellow')
}
def format(self, record):
if not record.exc_info:
level = record.levelname.lower()
msg = record.getMessage()
if level in self.colors:
prefix = click.style('{}: '.format(level),
**self.colors[level])
msg = '\n'.join(prefix + x for x in msg.splitlines())
return msg
return logging.Formatter.format(self, record)
class ClickHandler(logging.Handler):
_use_stderr = True
def emit(self, record):
try:
msg = self.format(record)
level = record.levelname.lower()
click.echo(msg, err=self._use_stderr)
except Exception:
self.handleError(record)
_default_handler = ClickHandler()
_default_handler.formatter = ColorFormatter()
def _normalize_logger(logger):
if not isinstance(logger, logging.Logger):
logger = logging.getLogger(logger)
return logger
def basic_config(logger=None):
'''Set up the default handler (:py:class:`ClickHandler`) and formatter
(:py:class:`ColorFormatter`) on the given logger.'''
logger = _normalize_logger(logger)
logger.handlers = [_default_handler]
logger.propagate = False
return logger
|
