python3:日誌模塊封裝

#!/usr/bin/env python
# coding: utf-8

import logging
import time
import os


def get_log(log_model_name, style='console'):
    """
    :param log_model_name: 要打印日誌的模塊名
    :param style: 日誌打印方式,console:打印到控制檯;file:打印到文件中
    :return:
    """
    # 創建一個logger
    logger = logging.getLogger(log_model_name)
    logger.setLevel(logging.INFO)

    # 設置日誌存放路徑,日誌文件名
    # 獲取本地實際,轉換爲設置的格式
    rq = time.strftime('%Y%m%d%H%M',time.localtime(time.time()))
    # 設置所有日誌和錯誤日誌的存放路徑
    path = os.path.dirname(os.path.abspath('log_print.py'))
    all_log_path = os.path.join(path, 'All_Logs/')
    error_log_path = os.path.join(path, 'Error_Logs/')

    # 判斷要保存的日誌文件夾是否存在,不存在就創建
    if not os.path.exists(all_log_path):
        os.mkdir(all_log_path)
    if not os.path.exists(error_log_path):
        os.mkdir(error_log_path)

    # 設置日誌文件名
    all_log_name = all_log_path + rq + '.log'
    error_log_name = error_log_path + rq + '.log'

    if style == 'console':
        # 創建一個handler輸出到控制檯
        # 創建一個handler寫入所有日誌
        all_log = logging.StreamHandler()
        all_log.setLevel(logging.INFO)

        # 創建一個handler寫入錯誤日誌
        error_log = logging.StreamHandler(error_log_name)
        error_log.setLevel(logging.ERROR)

        # 定義日誌輸出格式
        # 以時間-日誌名稱-日誌級別-日誌內容的形式展示
        all_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(message)s')

        # 以時間-日誌名稱-日誌級別-文件名-函數行數-錯誤內容
        error_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(module)s-%(message)s')

        # 將定義好的輸出形式添加到handler
        all_log.setFormatter(all_log_formatter)
        error_log.setFormatter(error_log_formatter)

        # 給logger添加handler
        logger.addHandler(all_log)
        logger.addHandler(error_log)
        return logger
    else:
        # 日誌寫入到文件
        # 創建一個handler寫入所有日誌
        all_log = logging.FileHandler(all_log_name)
        all_log.setLevel(logging.INFO)

        # 創建一個handler寫入錯誤日誌
        error_log = logging.FileHandler(error_log_name)
        error_log.setLevel(logging.ERROR)

        # 定義日誌輸出格式
        # 以時間-日誌名稱-日誌級別-日誌內容的形式展示
        all_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(message)s')

        # 以時間-日誌名稱-日誌級別-文件名-函數行數-錯誤內容
        error_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(module)s-%(message)s')

        # 將定義好的輸出形式添加到handler
        all_log.setFormatter(all_log_formatter)
        error_log.setFormatter(error_log_formatter)

        # 給logger添加handler
        logger.addHandler(all_log)
        logger.addHandler(error_log)
        return logger


if __name__ == '__main__':
    logger0 = get_log('test', 'console')  # test需要打印日誌的項目名稱
    logger0.info('aaa 1213')
    logger0.info('aaa 7890')

style=='console':

style=='file':

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章