/
logger.py
42 lines (34 loc) · 1.28 KB
/
logger.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
import os
import logging
def init_logger(config):
LOGROOT = './log/'
dir_name = os.path.dirname(LOGROOT)
if not os.path.exists(dir_name):
os.makedirs(dir_name)
logfilename = config['filename'] + '.log'
logfilepath = os.path.join(LOGROOT, logfilename)
filefmt = "%(asctime)-15s %(levelname)s %(message)s"
filedatefmt = "%d %b %Y %H:%M:%S"
fileformatter = logging.Formatter(filefmt, filedatefmt)
sfmt = "%(asctime)-15s %(levelname)s %(message)s"
sdatefmt = "%d %b %H:%M:%S"
sformatter = logging.Formatter(sfmt, sdatefmt)
if config['state'] is None or config['state'].lower() == 'info':
level = logging.INFO
elif config['state'].lower() == 'debug':
level = logging.DEBUG
elif config['state'].lower() == 'error':
level = logging.ERROR
elif config['state'].lower() == 'warning':
level = logging.WARNING
elif config['state'].lower() == 'critical':
level = logging.CRITICAL
else:
level = logging.INFO
fh = logging.FileHandler(logfilepath)
fh.setLevel(level)
fh.setFormatter(fileformatter)
sh = logging.StreamHandler()
sh.setLevel(level)
sh.setFormatter(sformatter)
logging.basicConfig(level=level, handlers=[fh, sh])