| 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) | |