Python colorlog 库:为你的日志穿上彩色外衣

诸神缄默不语-个人技术博文与视频目录

还在看单调的黑白日志吗?给日志加点颜色,让调试变得更高效!

https://pypi.org/project/colorlog/

前言:为什么需要彩色日志?

在日常开发和调试中,日志是我们最亲密的伙伴。但是,面对密密麻麻的黑白日志文本,快速定位错误、警告或关键信息往往如同大海捞针。这时候,colorlog 库就能大显身手了!它能为不同级别的日志信息添加不同的颜色,让日志输出一目了然,大大提升调试效率。

使用 colorlog 为日志添加色彩编码,能让不同严重级别的信息一目了然,从而提升调试体验。

什么是 colorlog?

colorlog 是一个 Python 日志格式化工具,它扩展了 Python 标准库中的 logging 模块,专门用于在终端输出带颜色的日志 。通过为不同日志级别配置不同颜色,我们可以快速区分 DEBUG、INFO、WARNING、ERROR 等级别的信息。

安装 colorlog

安装 colorlog 非常简单,只需使用 pip:

pip install colorlog

基础用法:快速上手

让我们先来看一个简单的例子,了解 colorlog 的基本使用方法:

import colorlog
import logging

# 创建 logger 实例
logger = colorlog.getLogger('example')

# 创建 StreamHandler
handler = colorlog.StreamHandler()

# 设置格式化器
handler.setFormatter(colorlog.ColoredFormatter(
    '%(log_color)s%(levelname)s:%(name)s:%(message)s'
))

# 添加 handler 到 logger
logger.addHandler(handler)

# 设置日志级别
logger.setLevel(logging.DEBUG)

# 测试不同级别的日志输出
logger.debug("这是一条 DEBUG 级别的日志")
logger.info("这是一条 INFO 级别的日志")
logger.warning("这是一条 WARNING 级别的日志")
logger.error("这是一条 ERROR 级别的日志")
logger.critical("这是一条 CRITICAL 级别的日志")

运行上面的代码,你将在终端中看到不同颜色的日志输出,默认情况下:

  • DEBUG:青色
  • INFO:绿色
  • WARNING:黄色
  • ERROR:红色
  • CRITICAL:红色背景白色文字

核心组件详解

ColoredFormatter 类

ColoredFormatter 是 colorlog 的核心类,它继承自 Python 标准的 logging.Formatter,专门用于生成带颜色的日志输出 。

主要参数说明:

  • format:用于输出日志的格式化字符串(必需)
  • datefmt:可选的日期格式,传递给基类
  • reset:默认为 True,表示在消息输出结束时添加颜色重置代码
  • log_colors:日志级别到颜色名称的映射
  • secondary_log_colors:定义在格式化字符串中可用的其他颜色
  • style:在 Python 3.2 以上版本可用

颜色配置详解

colorlog 支持丰富的颜色配置选项:

基本颜色名称: black, red, green, yellow, blue, purple, cyan, white

颜色前缀:

  • fg_{color}:设置前景色(文字颜色)
  • bg_{color}:设置背景色
  • boldbold_{color}:设置粗体/亮色

特殊代码:

  • reset:清除所有格式(包括前景色和背景色)

高级配置技巧

1. 自定义日志颜色

你可以完全自定义每个日志级别的颜色:

from colorlog import ColoredFormatter

formatter = ColoredFormatter(
    "%(log_color)s%(levelname)-8s%(reset)s %(blue)s%(message)s",
    datefmt=None,
    reset=True,
    log_colors={
        'DEBUG': 'cyan',
        'INFO': 'green',
        'WARNING': 'yellow',
        'ERROR': 'red',
        'CRITICAL': 'red,bg_white',
    },
    secondary_log_colors={},
    style='%'
)

handler = logging.StreamHandler()
handler.setFormatter(formatter)

logger = logging.getLogger('custom_example')
logger.addHandler(handler)
logger.setLevel(logging.DEBUG)

2. 使用辅助颜色

secondary_log_colors 允许你为同一日志消息的不同部分设置不同的颜色规则:

formatter = ColoredFormatter(
    "%(log_color)s%(levelname)-8s%(reset)s %(message_log_color)s%(message)s",
    secondary_log_colors={
        'message': {
            'ERROR': 'red',
            'CRITICAL': 'red'
        }
    }
)

这个配置会让 ERROR 和 CRITICAL 级别的日志消息本身显示为红色,而其他级别的消息保持默认颜色 。

3. 使用字典配置

对于大型项目,推荐使用 dictConfig 进行配置:

import logging.config

logging.config.dictConfig({
    'formatters': {
        'colored': {
            '()': 'colorlog.ColoredFormatter',
            'format': "%(log_color)s%(levelname)-8s%(reset)s %(blue)s%(message)s"
        }
    },
    'handlers': {
        'stream': {
            'class': 'logging.StreamHandler',
            'formatter': 'colored',
        },
    },
    'root': {
        'handlers': ['stream'],
        'level': 'DEBUG',
    }
})

logger = logging.getLogger('dict_config_example')

4. 配置文件方式

你也可以使用 .ini 格式的配置文件:

[loggers]
keys=root

[logger_root]
handlers=stream
level=DEBUG

[formatters]
keys=color

[formatter_color]
class=colorlog.ColoredFormatter
format=%(log_color)s%(levelname)s:%(name)s:%(message)s
datefmt=%H:%M:%S

[handlers]
keys=stream

[handler_stream]
class=StreamHandler
formatter=color
args=()

然后在代码中加载配置:

import logging
import logging.config

logging.config.fileConfig('logging_config.ini')
logger = logging.getLogger('config_file_example')

实际应用示例

在 FastAPI 项目中使用 colorlog

让我们看看如何在 FastAPI 项目中使用 colorlog 来美化日志输出:

import colorlog
import logging
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel

# 配置日志
handler = colorlog.StreamHandler()
formatter = colorlog.ColoredFormatter(
    "%(log_color)s%(asctime)s - %(levelname)-8s%(reset)s %(blue)s%(name)-15s %(message)s",
    datefmt="%Y-%m-%d %H:%M:%S",
    log_colors={
        'DEBUG': 'cyan',
        'INFO': 'bold_green',
        'WARNING': 'bold_yellow',
        'ERROR': 'bold_red',
        'CRITICAL': 'bold_red,bg_white',
    }
)
handler.setFormatter(formatter)

# 为 FastAPI 相关日志器配置 colorlog
uvicorn_logger = logging.getLogger("uvicorn")
uvicorn_logger.handlers = []
uvicorn_logger.addHandler(handler)
uvicorn_logger.setLevel(logging.DEBUG)

uvicorn_access = logging.getLogger("uvicorn.access")
uvicorn_access.handlers = []
uvicorn_access.addHandler(handler)

app = FastAPI(title="Colorful API", version="1.0.0")
logger = colorlog.getLogger("api")
logger.addHandler(handler)
logger.setLevel(logging.DEBUG)

class Item(BaseModel):
    name: str
    description: str = None
    price: float

@app.get("/")
async def read_root():
    logger.info("访问根路径")
    return {"message": "欢迎使用彩色日志 API!"}

@app.post("/items/")
async def create_item(item: Item):
    logger.debug(f"创建商品: {item.name}, 价格: {item.price}")
    
    if item.price <= 0:
        logger.warning(f"商品价格无效: {item.price}")
        raise HTTPException(status_code=400, detail="价格必须大于0")
    
    logger.info(f"成功创建商品: {item.name}")
    return {"status": "success", "item": item}

@app.get("/error-test")
async def error_test():
    try:
        # 模拟一个错误
        result = 1 / 0
    except Exception as e:
        logger.error(f"发生错误: {str(e)}", exc_info=True)
        raise HTTPException(status_code=500, detail="内部服务器错误")

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000, log_config=None)

同时输出到控制台和文件

有时候,我们既想要在控制台看到彩色日志,又希望将日志保存到文件中(不包含颜色代码):

import colorlog
import logging

# 创建 logger
logger = colorlog.getLogger('multi_handler')

# 控制台 Handler(带颜色)
console_handler = colorlog.StreamHandler()
console_formatter = colorlog.ColoredFormatter(
    "%(log_color)s%(asctime)s %(levelname)-8s%(reset)s %(blue)s%(message)s",
    datefmt="%H:%M:%S"
)
console_handler.setFormatter(console_formatter)

# 文件 Handler(无颜色)
file_handler = logging.FileHandler('app.log')
file_formatter = logging.Formatter(
    "%(asctime)s %(levelname)-8s %(message)s",
    datefmt="%Y-%m-%d %H:%M:%S"
)
file_handler.setFormatter(file_formatter)

# 添加两个 handler
logger.addHandler(console_handler)
logger.addHandler(file_handler)
logger.setLevel(logging.DEBUG)

# 测试日志
logger.info("这条消息会同时显示在控制台(带颜色)和文件(无颜色)中")

自定义日志级别

你甚至可以创建自定义的日志级别并为其指定颜色:

import logging
import colorlog

# 创建自定义日志级别
TRACE = 5
logging.addLevelName(TRACE, 'TRACE')

# 添加 trace 方法到 logger
def trace(self, message, *args, **kws):
    if self.isEnabledFor(TRACE):
        self._log(TRACE, message, args, **kws)

logging.Logger.trace = trace

# 配置 colorlog 包含自定义级别
formatter = colorlog.ColoredFormatter(
    log_colors={
        'TRACE': 'yellow',
        'DEBUG': 'cyan',
        'INFO': 'green',
        'WARNING': 'yellow',
        'ERROR': 'red',
        'CRITICAL': 'red,bg_white',
    },
    format='%(log_color)s%(levelname)-8s%(reset)s %(message)s'
)

handler = logging.StreamHandler()
handler.setFormatter(formatter)

logger = colorlog.getLogger('custom_level')
logger.addHandler(handler)
logger.setLevel(TRACE)

# 使用自定义级别
logger.trace("这是一条 TRACE 级别的日志")

注意事项和最佳实践

  1. 兼容性:colorlog 主要适用于终端输出,在文件日志中颜色代码可能会显示为乱码。

  2. 性能:在生产环境中,可以考虑根据环境变量决定是否启用彩色日志,以减少不必要的性能开销。

  3. 一致性:在团队项目中,建议统一颜色配置,确保所有成员对颜色含义有相同的理解。

  4. 可访问性:考虑到色盲或色弱用户,不要完全依赖颜色来区分日志级别,还应该保留文字级别的信息。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

诸神缄默不语

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值