python接口自动化测试—logging模块，日志的处理

首先定义日志的级别

LEVELS = {
 'debug': logging.DEBUG,
 'info': logging.INFO,
 'warning': logging.WARNING, 
'error': logging.ERROR, 
'critical': logging.CRITICAL 
}

logger：日志对象，logging模块中最基础的对象，用logging.getLogger(name)方法进行初始化，name可以不填。

源码如下：

def getLogger(name=None):
    """
    Return a logger with the specified name, creating it if necessary.

    If no name is specified, return the root logger.
    """
    if name:
        return Logger.manager.getLogger(name)
    else:
        return root

定义日志类，对不同的级别进行封装

全部代码如下：

import logging
import os
import time

LEVELS = {
    'debug': logging.DEBUG,
    'info': logging.INFO,
    'warning': logging.WARNING,
    'error': logging.ERROR,
    'critical': logging.CRITICAL
}

logger = logging.getLogger()
level = 'default'


def create_file(file_name):
    path = file_name[0:file_name.rfind('/')]
    if not os.path.isdir(path):
        os.makedirs(path)
    if not os.path.isfile(file_name):
        with open('file_name', 'w', encoding='utf-8') as fd:
            fd.write(file_name)


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.abspath('.'))
    log_file = path + '/log/log_file.log'
    err_file = path + '/log/err_file.log'
    logger.setLevel(LEVELS.get(level, logging.NOTSET))
    create_file(log_file)
    create_file(err_file)
    date = '%Y-%m-%d %H:%M:%S'
    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.debug('[INFO ' + get_current_time() + ']' + log_meg)
        remove_handler('info')

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

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

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


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")