python 自定义封装带颜色的logging模块
【python自定义封装带颜色的logging模块】自己在搭建python接口自动化框架 分享一些内容过程中想自己封装一个logger方法 根据logging进行二次封装 代码如下

import loggingimport osimport timeimport colorlogfrom logging.handlers import RotatingFileHandler# 创建文件目录cur_path = os.path.dirname(os.path.realpath(__file__))# log_path是存放日志的路径log_path = os.path.join(os.path.dirname(cur_path), 'logs')if not os.path.exists(log_path): os.mkdir(log_path)# 如果不存在这个logs文件夹，就自动创建一个# 修改log保存位置timestamp = time.strftime("%Y-%m-%d", time.localtime())logfile_name = '%s.log' % timestamplogfile_path = os.path.join(log_path, logfile_name)# 定义不同日志等级颜色log_colors_config = {'DEBUG': 'bold_cyan','INFO': 'bold_green','WARNING': 'bold_yellow','ERROR': 'bold_red','CRITICAL': 'red',}class Logger(logging.Logger):def __init__(self, name, level='DEBUG', file=None, encoding='utf-8'):super().__init__(name)self.encoding = encodingself.file = fileself.level = level# 针对所需要的日志信息 手动调整颜色formatter = colorlog.ColoredFormatter('%(log_color)s%(levelname)1.1s %(asctime)s %(reset)s| %(message_log_color)s%(levelname)-8s %(reset)s| %(''log_color)s[%(filename)s%(reset)s:%(log_color)s%(module)s%(reset)s:%(log_color)s%(funcName)s%(''reset)s:%(log_color)s%(''lineno)d] %(reset)s- %(white)s%(message)s',reset=True,log_colors=log_colors_config,secondary_log_colors={'message': {'DEBUG': 'blue','INFO': 'blue','WARNING': 'blue','ERROR': 'red','CRITICAL': 'bold_red'}},style='%')# 日志输出格式# 创建一个FileHandler，用于写到本地rotatingFileHandler = logging.handlers.RotatingFileHandler(filename=logfile_path,maxBytes=1024 * 1024 * 50,backupCount=5)rotatingFileHandler.setFormatter(formatter)rotatingFileHandler.setLevel(logging.DEBUG)self.addHandler(rotatingFileHandler)# 创建一个StreamHandler,用于输出到控制台console = colorlog.StreamHandler()console.setLevel(logging.DEBUG)console.setFormatter(formatter)self.addHandler(console)self.setLevel(logging.DEBUG)logger = Logger(name=logfile_path, file=logfile_path)

# 引入封装好的logger模块from common.logger_handler import loggerdef physical_strength(self, abnormal):"""兑换体力异常通用方法"""if self.attrs.__contains__('costType'):attrs_Type = {"costType": abnormal,"count": self.attrs["count"]}response_Type = r().response(self.send_uid, self.code, self.event, attrs_Type)# 使用时直接调用logger.info()就行logger.info(f"physical_strength_{abnormal},response_Type:{response_Type}")assert response_Type["code"] != 0time.sleep(2)attrs_count = {"costType": self.attrs["costType"],"count": abnormal}response_count = r().response(self.send_uid, self.code, self.event, attrs_count)logger.info(f"physical_strength_{abnormal},response_count:{response_count}")assert response_count["code"] != 0time.sleep(2)attrs_all = {"costType": abnormal,"count": abnormal}response_all = r().response(self.send_uid, self.code, self.event, attrs_all)logger.info(f"physical_strength_{abnormal},response_all:{response_all}")assert response_all["code"] != 0time.sleep(2)else:attrs_count = {"count": abnormal}response_count = r().response(self.send_uid, self.code, self.event, attrs_count)logger.info(f"physical_strength_{abnormal},response_count:{response_count}")assert response_count["code"] != 0time.sleep(2)

def debug(self, message):self.__console('debug', message)def info(self, message):self.__console('info', message)def warning(self, message):self.__console('warning', message)def error(self, message):self.__console('error', message)