-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger.py
46 lines (30 loc) · 1.34 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
43
44
45
46
# logger.py
import logging
import os
class Logger:
def __init__(self, log_name, output_dir=None, external_logger=None):
if output_dir is None:
self.setup_logger(log_name+".log", "output", None)
else:
self.setup_logger(log_name, output_dir, external_logger)
def setup_logger(self, log_name, output_dir, external_logger=None):
if external_logger is not None:
external_logger.info(f"Creating logger: {log_name}, {output_dir}, {external_logger}")
self.logger = logging.getLogger(log_name)
self.logger.setLevel(logging.DEBUG)
# create file handler which logs even info messages
fh = logging.FileHandler(os.path.join(output_dir, log_name), 'w')
fh.setLevel(logging.INFO)
# create console handler with a higher log level
ch = logging.StreamHandler()
ch.setLevel(logging.ERROR)
# create formatter and add it to the handlers
#formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
formatter = logging.Formatter('%(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)
# add the handlers to the logger
self.logger.addHandler(fh)
self.logger.addHandler(ch)
def get_logger(self):
return self.logger