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)