-
Notifications
You must be signed in to change notification settings - Fork 0
/
log.py
executable file
·29 lines (24 loc) · 1.02 KB
/
log.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
# code from https://github.com/TooTouch/MemSeg/blob/main/log.py
import logging
import logging.handlers
class FormatterNoInfo(logging.Formatter):
def __init__(self, fmt="%(levelname)s: %(message)s"):
logging.Formatter.__init__(self, fmt)
def format(self, record):
if record.levelno == logging.INFO:
return str(record.getMessage())
return logging.Formatter.format(self, record)
def setup_default_logging(default_level=logging.INFO, log_path=""):
console_handler = logging.StreamHandler()
console_handler.setFormatter(FormatterNoInfo())
logging.root.addHandler(console_handler)
logging.root.setLevel(default_level)
if log_path:
file_handler = logging.handlers.RotatingFileHandler(
log_path, maxBytes=(1024**2 * 2), backupCount=3
)
file_formatter = logging.Formatter(
"%(asctime)s - %(name)20s: [%(levelname)8s] - %(message)s"
)
file_handler.setFormatter(file_formatter)
logging.root.addHandler(file_handler)