File size: 1,367 Bytes
4fb0bd1 |
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 |
import logging
import os
def init_logger(
root_log_level=logging.DEBUG,
console_log_level=logging.NOTSET,
log_file=None,
log_file_level=logging.NOTSET):
"""This funtion initializes a customized logger
Keyword Arguments:
root_log_level {int} -- root logging level (default: {logging.DEBUG})
console_log_level {int} -- console logging level (default: {logging.NOTSET})
log_file {str} -- logging file path (default: {None})
log_file_level {int} -- logging file level (default: {logging.NOTSET})
"""
log_format = logging.Formatter("[%(asctime)s - %(filename)s - line:%(lineno)d - %(levelname)s]: %(message)s")
handlers = []
console_handler = logging.StreamHandler()
console_handler.setLevel(console_log_level)
console_handler.setFormatter(log_format)
handlers.append(console_handler)
if log_file is not None and log_file != '':
if os.path.exists(log_file):
os.remove(log_file)
elif not os.path.exists(os.path.dirname(log_file)):
os.makedirs(os.path.dirname(log_file))
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(log_file_level)
file_handler.setFormatter(log_format)
handlers.append(file_handler)
logging.basicConfig(level=root_log_level, handlers=handlers)
|