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
|
# 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, 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.
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"
if dict_config:
logging.basicConfig(**dict_config)
filemode = dict_config["filemode"]
#
root_logger = logging.getLogger()
if stream in ["", "stderr", "stdout"]:
for handler in root_logger.handlers:
if isinstance(handler, StreamHandler):
# remove old ``StreamHandler``
root_logger.removeHandler(handler)
if stream == "":
# disable ``StreamHandler``
pass
else:
# add new ``StreamHandler``
handler = StreamHandler(getattr(sys, stream))
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``
root_logger.removeHandler(handler)
if logfile == "":
# disable ``FileHandler``
pass
else:
# add new ``FileHandler``
handler = FileHandler(logfile, mode=filemode)
root_logger.addHandler(handler)
|
