python 日誌處理

平時做數據處理基本離不了日誌記錄功能。每次都配置一堆挺煩人，索性封裝個模塊，這裏記錄一下，與大家共享。

說明

本日誌模塊目前只有一個方法getLogger，其他配置項通過參數傳遞，包括日誌文件名，等級，日誌文件劃分方式，日誌清除配置，日誌格式等。

logger.py

import logging
from logging import handlers

# 日誌級別字典
__level_dict = {
    'critical': logging.CRITICAL,
    'fatal': logging.CRITICAL,
    'error': logging.ERROR,
    'warning': logging.WARNING,
    'warn': logging.WARNING,
    'info': logging.INFO,
    'debug': logging.DEBUG
}


def getLogger(filename, level='info', when='D', backupCount=3,
              fmt='%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s'):
    """
    獲取日誌處理對象

    :param filename: 日誌文件名稱
    :param level: 日誌等級：debug, info, warn/warning, error, critical
    :param when: 日誌文件分割的時間單位，單位有以下幾種:<br>
                - S 秒<br>
                - M 分<br>
                - H 小時<br>
                - D 天<br>
                - W 每星期<br>
                - midnight 每天凌晨<br>
    :param backupCount: 備份文件的個數，如果超過這個數量，就會自動刪除
    :param fmt: 日誌信息格式
    :return:
    """
    level = __level_dict.get(level.lower(), None)
    logger = logging.getLogger(filename)
    # 設置日誌格式
    format_str = logging.Formatter(fmt)
    # 設置日誌級別
    logger.setLevel(level)
    # 控制檯輸出
    console_handler = logging.StreamHandler()
    # 控制檯輸出的格式
    console_handler.setFormatter(format_str)
    logger.addHandler(console_handler)

    # 文件輸出
    file_handler = handlers.TimedRotatingFileHandler(filename=filename, when=when, backupCount=backupCount,
                                                     encoding='utf-8')

    # 文件輸出的格式
    file_handler.setFormatter(format_str) 
    logger.addHandler(file_handler)
    return logger


if __name__ == '__main__':
    log = getLogger('test.log')
    log.debug('debug')
    log.info('info')
    log.warning('警告')
    log.error('報錯')
    log.critical('嚴重')
    getLogger('error.log', level='error').error('error')

運行結果

2018-09-06 17:07:40,909 - D:/jetbrains/workspace-pycharm/test/logger.py[line:59] - INFO: info
2018-09-06 17:07:40,909 - D:/jetbrains/workspace-pycharm/test/logger.py[line:60] - WARNING: 警告
2018-09-06 17:07:40,909 - D:/jetbrains/workspace-pycharm/test/logger.py[line:61] - ERROR: 報錯
2018-09-06 17:07:40,910 - D:/jetbrains/workspace-pycharm/test/logger.py[line:62] - CRITICAL: 嚴重
2018-09-06 17:07:40,915 - D:/jetbrains/workspace-pycharm/test/logger.py[line:63] - ERROR: error