TEST/common/log_util.py

import logging
import time

from common.yaml_util import get_project_path, read_config_yaml


class LoggerUtil:
    def create_log(self, logger_name='log'):

        # 创建一个日志对象
        self.logger = logging.getLogger(logger_name)
        # 设置全局的日志级别(从低到高：debug调试<info信息<warning警告<error错误<critical严重)
        self.logger.setLevel(logging.DEBUG)
        if not self.logger.handlers:
            # ------文件日志------
            # 1创建文件日志路径
            self.file_log_path = get_project_path() + "/logs/" + read_config_yaml("log", "log_name") + str(
                int(time.time())) + ".log"
            # 2创建文件日志控制器
            self.file_hander = logging.FileHandler(self.file_log_path, encoding='UTF-8')
            # 3设置文件日志级别
            file_log_level = str(read_config_yaml("log", "log_level")).lower()
            if file_log_level == "debug":
                self.file_hander.setLevel(logging.DEBUG)
            elif file_log_level == "info":
                self.file_hander.setLevel(logging.INFO)
            elif file_log_level == "warning":
                self.file_hander.setLevel(logging.WARNING)
            elif file_log_level == "error":
                self.file_hander.setLevel(logging.ERROR)
            elif file_log_level == "critical":
                self.file_hander.setLevel(logging.CRITICAL)
            else:
                self.file_hander.setLevel(logging.DEBUG)
            # 4创建文件的格式
            self.file_hander.setFormatter(logging.Formatter(read_config_yaml("log", "log_format")))
            # 将文件日志的控制器加入日志对象
            self.logger.addHandler(self.file_hander)
            # 创建控制台日志
            # 1创建控制台日志控制器
            self.console_hander = logging.StreamHandler()
            # 2设置控制台日志级别
            console_log_level = str(read_config_yaml("log", "log_level")).lower()
            if console_log_level == "debug":
                self.console_hander.setLevel(logging.DEBUG)
            elif console_log_level == "info":
                self.console_hander.setLevel(logging.INFO)
            elif console_log_level == "warning":
                self.console_hander.setLevel(logging.WARNING)
            elif console_log_level == "error":
                self.console_hander.setLevel(logging.ERROR)
            elif console_log_level == "critical":
                self.console_hander.setLevel(logging.CRITICAL)
            else:
                self.console_hander.setLevel(logging.DEBUG)
            # 3创建控制台的日志格式
            self.console_hander.setFormatter(logging.Formatter(read_config_yaml("log", "log_format")))
            # 将控制台日志加入日志对象
            self.logger.addHandler(self.console_hander)

        # 返回包含文件日志控制器和控制台日志控制器的日志对象
        return self.logger


# 错误日志的输出
def error_log(message):
    LoggerUtil().create_log().error(message)
    raise Exception(message)


# 信息日志的输出
def logs(message):
    LoggerUtil().create_log().info(message)


if __name__ == '__main__':
    error_log("车务通")