-
Notifications
You must be signed in to change notification settings - Fork 1
/
app.py
94 lines (78 loc) · 2.63 KB
/
app.py
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
import logging
import os
import sys
from logging import StreamHandler
from logging.handlers import SysLogHandler
import puppeter
from puppeter.domain.facter import Facter
from puppeter.domain.model.osfacts import OsFamily
class App:
def __init__(self, parsed):
self._parsed = parsed
def run(self):
root = logging.getLogger()
level = self.__get_log_level(self._parsed.verbose)
root.setLevel(level)
handlers = (self.__syslog_handler(), self.__stderr_handler())
for hndl in handlers:
root.addHandler(hndl)
@staticmethod
def __stderr_handler():
handler = StreamHandler(stream=sys.stderr)
fmt = '# %(levelname)s: %(message)s'
handler.setFormatter(ColoredFormatter(fmt=fmt))
handler.setLevel(logging.NOTSET)
return handler
@staticmethod
def __syslog_handler():
osfamily = Facter.get(OsFamily)
if osfamily in (OsFamily.Debian, OsFamily.RedHat, OsFamily.Suse):
handler = SysLogHandler(address='/dev/log')
else:
handler = SysLogHandler()
fmt = puppeter.__program__ + '[%(process)d]: %(levelname)s %(name)s - %(message)s'
handler.setFormatter(logging.Formatter(fmt=fmt))
handler.setLevel(logging.INFO)
return handler
@staticmethod
def __get_log_level(verbosity):
if verbosity is None:
verbosity = 0
if verbosity > 2:
verbosity = 2
levels = {
0: logging.WARNING,
1: logging.INFO,
2: logging.DEBUG
}
return levels[verbosity]
def color_code(fg, bg=None):
if bg is not None:
return FGBG_COLOR_SEQ % (fg, bg)
else:
return FG_COLOR_SEQ % fg
RESET_SEQ = "\033[0m"
FG_COLOR_SEQ = "\033[38;5;%dm"
FGBG_COLOR_SEQ = "\033[38;5;%d;48;5;%dm"
# ref: https://unix.stackexchange.com/a/124409/145501
COLORS = {
'WARNING': color_code(fg=226),
'INFO': color_code(fg=155),
'DEBUG': color_code(fg=244),
'CRITICAL': color_code(fg=219, bg=196),
'ERROR': color_code(fg=196)
}
try:
SUPPORTS_256_COLORS = os.environ['XTERM_256_COLORS'] == '1'
except KeyError:
SUPPORTS_256_COLORS = False
class ColoredFormatter(logging.Formatter):
def __init__(self, fmt, use_color=SUPPORTS_256_COLORS):
logging.Formatter.__init__(self, fmt=fmt)
self.use_color = use_color
def format(self, record):
formatted = logging.Formatter.format(self, record)
levelname = record.levelname
if self.use_color and levelname in COLORS:
formatted = COLORS[levelname] + formatted + RESET_SEQ
return formatted