Python基礎之logging日誌模塊封裝

　　引言

　　使用python作自動化測試的時候，對於關鍵信息輸出和記錄是必要的，方便調試和記錄查找。python有個logging模塊，咱們簡單對其進行封裝。

　　設計流程

　　話很少說，流程圖已經畫，本身領會。

 

 

　　源碼

　　

"""
封裝log方法

"""

import os
import time
import logging.handlers

# 日誌打印等級
LEVELS = {
    'debug': logging.DEBUG,
    'info': logging.INFO,
    'warning': logging.WARNING,
    'error': logging.ERROR,
    'critical': logging.CRITICAL
}

# 建立一個日誌
logger = logging.getLogger()
level = 'default'


# 建立日誌文件方法
def create_file(filename):
    path = filename[0:filename.rfind('/')]
    if not os.path.isdir(path):
        os.makedirs(path)
    if not os.path.isfile(filename):
        fd = open(filename, mode='w', encoding='utf-8')
        fd.close()
    else:
        pass



# 給logger添加handler 添加內容到日誌句柄中
def set_handler(levels):
    if levels == 'error':
        logger.addHandler(MyLog.err_handler)
    logger.addHandler(MyLog.handler)

# 在記錄日誌以後移除句柄
def remove_handler(levels):
    if levels == 'error':
        logger.removeHandler(MyLog.err_handler)
    logger.removeHandler(MyLog.handler)


def get_current_time():
    return time.strftime(MyLog.date, time.localtime(time.time()))


class MyLog:
    path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    log_file = path+'/Log/log.log'
    err_file = path+'/Log/err.log'
    logger.setLevel(LEVELS.get(level, logging.NOTSET))
    create_file(log_file)
    create_file(err_file)
    date = '%Y-%m-%d %H:%M:%S'

    # 建立一個handler，用於寫入日誌文件
    handler = logging.FileHandler(log_file, encoding='utf-8')
    err_handler = logging.FileHandler(err_file, encoding='utf-8')

    @staticmethod
    def debug(log_meg):
        set_handler('debug')
        # 文件中輸出模式
        logger.debug("[DEBUG " + get_current_time() + "]" + log_meg)
        remove_handler('debug')

    @staticmethod
    def info(log_meg):
        set_handler('info')
        logger.info("[INFO " + get_current_time() + "]" + log_meg)
        remove_handler('info')

    @staticmethod
    def warning(log_meg):
        set_handler('warning')
        logger.warning("[WARNING " + get_current_time() + "]" + log_meg)
        remove_handler('warning')

    @staticmethod
    def error(log_meg):
        set_handler('error')
        logger.error("[ERROR " + get_current_time() + "]" + log_meg)
        remove_handler('error')

    @staticmethod
    def critical(log_meg):
        set_handler('critical')
        logger.error("[CRITICAL " + get_current_time() + "]" + log_meg)
        remove_handler('critical')

    # 設置控制檯輸出格式
    formatter = logging.Formatter(
        '[%(asctime)s] [%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S')
    # 再建立一個handler，用於輸出到控制檯
    console = logging.StreamHandler()
    console.setFormatter(formatter)
    logger.addHandler(console)
    console.setLevel(logging.NOTSET)


if __name__ == "__main__":
    MyLog.debug("This is debug message")
    MyLog.info("This is info message")
    MyLog.warning("This is warning message")
    MyLog.error("This is error")
    MyLog.critical("This is critical message")

　　

　　運行結果

[2020-05-23 20:41:10] [DEBUG] [DEBUG 2020-05-23 20:41:10]This is debug message
[2020-05-23 20:41:10] [INFO] [INFO 2020-05-23 20:41:10]This is info message
[2020-05-23 20:41:10] [WARNING] [WARNING 2020-05-23 20:41:10]This is warning message
[2020-05-23 20:41:10] [ERROR] [ERROR 2020-05-23 20:41:10]This is error
[2020-05-23 20:41:10] [ERROR] [CRITICAL 2020-05-23 20:41:10]This is critical message

Process finished with exit code 0

　　

　　文件結構：

 

 　　文件內容：

 

 

 

　　總結

　　以上就是python日誌的簡單封裝，基本上知足平常的需求。

　　另外，對測試開發，自動化測試技術與思想感興趣的朋友，能夠加入QQ測開交流羣：696400122進行交流，互相學習與進步。不積跬步無以致千里！