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
|
# Copyright (c) 2016 Weitian LI <liweitianux@live.com>
# MIT license
"""
Logging utilities.
"""
import sys
import logging
from logging import FileHandler, StreamHandler
def setup_logging(dict_config=None, level=None, stream=None, logfile=None):
"""Setup the logging.
This will override the logging configurations in the config file
if specified (e.g., by command line arguments).
Parameters
----------
dict_config : dict
Dict of logging configurations specified in the config file.
If this parameter specified, the logging will be reconfigured.
level : string;
Override the existing log level
stream : string; "stderr", "stdout", or ""
This controls where the log messages go to.
If not None, then override the old ``StreamHandler`` settings;
if ``stream=""``, then disable the ``StreamHandler``.
logfile : string
Specify the file where the log messages go to.
If ``logfile=""``, then disable the ``FileHandler``.
NOTE
----
If the logging already has ``StreamHandler`` or ``FileHandler``
configured, then the old handler will be *replaced* (i.e., remove
the old one, then add the new one).
"""
# default file open mode for logging to file
filemode = "a"
root_logger = logging.getLogger()
#
if dict_config:
# XXX:
# "basicConfig()" does NOT accept paramter "filemode" if the
# corresponding parameter "filename" NOT specified.
filemode = dict_config.pop("filemode", filemode)
# Clear existing handlers, otherwise further "basicConfig" calls
# will be ignored
for handler in root_logger.handlers:
handler.close()
root_logger.removeHandler(handler)
# Initialize/reconfigure the logging, which will automatically
# create a ``Formatter`` for handlers if necessary, and adding
# the handlers to the "root" logger.
logging.basicConfig(**dict_config)
#
# Configured logging has at least one handler with configured formatter.
# Keep the existing formatter to keep the configured format styles.
formatter = root_logger.handlers[0].formatter
if level is not None:
level_int = getattr(logging, level.upper(), None)
if not isinstance(level_int, int):
raise ValueError("invalid log level: %s" % level)
root_logger.setLevel(level_int)
#
if stream is None:
pass
elif stream in ["", "stderr", "stdout"]:
for handler in root_logger.handlers:
if isinstance(handler, StreamHandler):
# remove old ``StreamHandler``
handler.close()
root_logger.removeHandler(handler)
if stream == "":
# disable ``StreamHandler``
pass
else:
# add new ``StreamHandler``
handler = StreamHandler(getattr(sys, stream))
handler.setFormatter(formatter)
root_logger.addHandler(handler)
else:
raise ValueError("invalid stream: %s" % stream)
#
if logfile is not None:
for handler in root_logger.handlers:
if isinstance(handler, FileHandler):
filemode = handler.mode
# remove old ``FileHandler``
handler.close()
root_logger.removeHandler(handler)
if logfile == "":
# disable ``FileHandler``
pass
else:
# add new ``FileHandler``
handler = FileHandler(logfile, mode=filemode)
handler.setFormatter(formatter)
root_logger.addHandler(handler)
|
