Code Sample: python logging

ゝ一纸荒年。 2023-06-07 05:51 33阅读 0赞

# 代码介绍 #

整理的一段python的logging的简单使用代码，方便以后写一些python脚本时记录日志使用

# 代码 #

import logging
    import time
    import os
    
    
    _NEED_ROTATE = False  # 每次执行时是否需要按照时间生成新的日志文件，时间粒度为秒
    _LOG_FILE_PREFIX = 'main'  # 日志文件的前缀
    _LOG_DIR = 'logs'  # 日志文件存放的目录
    _LOG_LEVEL = logging.INFO  # 日志级别
    
    """ 日志输出示例： [2019-10-13 14:06:42,860] [MAIN] [INFO] ------------Initialize the Logger Done------------ [2019-10-13 14:06:42,860] [MAIN] [INFO] INFO [2019-10-13 14:06:42,860] [MAIN] [WARNING] WARNING [2019-10-13 14:06:42,860] [MAIN] [ERROR] ERROR """
    _LOG_FORMAT = '[%(asctime)s] [%(name)s] [%(levelname)s] %(message)s'  # 日志输出格式
    logger = None
    _LOG_NAME = 'MAIN'
    
    
    if _LOG_DIR:
        if not os.path.isdir(_LOG_DIR):
            os.makedirs(_LOG_DIR)
    else:
        _LOG_DIR = '.'
    
    
    if _NEED_ROTATE:
        _LOG_FILE = f'{_LOG_FILE_PREFIX}_{time.strftime("%Y_%m_%d_%H_%M_%S")}.log'
    else:
        _LOG_FILE = f'{_LOG_FILE_PREFIX}.log'
    
    _LOG_FILE = os.path.join(_LOG_DIR, _LOG_FILE)
    
    _common_formatter = None
    _common_streamhandler = None
    _common_filehandler = None
    
    
    def get_logger(logger_name=None):
        if not logger_name:
            return logger
        new_logger = logging.getLogger(logger_name)
        _common_formatter = logging.Formatter(_LOG_FORMAT)
        _common_streamhandler = logging.StreamHandler()
        _common_filehandler = logging.FileHandler(_LOG_FILE, 'a', 'utf8')
    
        new_logger.setLevel(_LOG_LEVEL)
        _common_filehandler.setLevel(_LOG_LEVEL)
        _common_streamhandler.setLevel(_LOG_LEVEL)
    
        _common_filehandler.setFormatter(_common_formatter)
        _common_streamhandler.setFormatter(_common_formatter)
    
        new_logger.addHandler(_common_streamhandler)
        new_logger.addHandler(_common_filehandler)
        return new_logger
    
    
    def _init():
        global logger, _common_filehandler, _common_formatter, _common_streamhandler
    
        logger = logging.getLogger(_LOG_NAME)
        _common_formatter = logging.Formatter(_LOG_FORMAT)
        _common_streamhandler = logging.StreamHandler()
        _common_filehandler = logging.FileHandler(_LOG_FILE, 'a', 'utf8')
    
        logger.setLevel(_LOG_LEVEL)
        _common_filehandler.setLevel(_LOG_LEVEL)
        _common_streamhandler.setLevel(_LOG_LEVEL)
    
        _common_filehandler.setFormatter(_common_formatter)
        _common_streamhandler.setFormatter(_common_formatter)
    
        logger.addHandler(_common_streamhandler)
        logger.addHandler(_common_filehandler)
    
        logger.info('Initialize the Logger Done'.center(50, '-'))
    
    
    def debug(msg, *args, **kwargs):
        logger.debug(msg, *args, **kwargs)
    
    
    def info(msg, *args, **kwargs):
        logger.info(msg, *args, **kwargs)
    
    
    def warn(msg, *args, **kwargs):
        logger.warning(msg, *args, **kwargs)
    
    
    def error(msg, *args, **kwargs):
        logger.error(msg, *args, **kwargs)
    
    _init()
    
    if __name__ == "__main__":
        debug('DEBUG')
        info('INFO')
        warn('WARNING')
        error('ERROR')