讲师中心 微信公众号
AI工具推荐 视频效率加速

如何在 pytest 中自定义日志处理器以保留 extra 字典中的上下文字段

轻丽小哥_8111

轻丽小哥_8111

发布时间:2026-10-10 11:18:29

|

605人浏览过

|

来源于php中文网

原创

如何在 pytest 中自定义日志处理器以保留 extra 字典中的上下文字段

本文介绍一种通过替换 pytest 内置日志插件、注入 structlog 处理链的方式,使 pytest 捕获的日志(而非 stdout/stderr)也能完整显示 logging.Logger.info(..., extra={...}) 中动态传入的键值对,解决默认格式器忽略 extra 字段的问题。

本文介绍一种通过替换 pytest 内置日志插件、注入 structlog 处理链的方式,使 pytest 捕获的日志(而非 stdout/stderr)也能完整显示 `logging.logger.info(..., extra={...})` 中动态传入的键值对,解决默认格式器忽略 `extra` 字段的问题。

在使用 structlog 与标准库 logging 混合架构时,常通过 extra= 参数向日志注入运行时上下文(如 {"request_id": "abc123", "user_id": 42})。这类字段在应用正常运行时能被 structlog.stdlib.ExtraAdder() 和 ConsoleRenderer 正确提取并渲染为 key='value' 形式。但当测试运行于 pytest 下并启用 --log-cli-level 时,pytest 会接管日志捕获流程,使用其内部 _pytest.logging.LoggingPlugin 创建的 Formatter —— 该格式器不感知 extra 字典结构,仅支持预设字段(如 %(name)s, %(levelname)s),导致所有动态上下文丢失。

直接配置 log_cli_format 无法解决问题,因为标准 logging.Formatter 的格式字符串(如 '%(message)s %(foo)s')要求字段名硬编码,而 extra 中的键是动态、不可预知的。因此,必须绕过原生 Formatter,改用 structlog 的处理器链(ProcessorFormatter),它能自动遍历 event_dict 并将全部剩余键值对格式化输出。

实现方案是在项目根目录的 conftest.py 中重写 pytest 日志插件:

# conftest.py
import _pytest.logging
import logging
import structlog.dev
import structlog.stdlib
import structlog.processors
import structlog.typing


def drop_structlog_fields(
    logger: logging.Logger | structlog.typing.FilteringBoundLogger,
    method_name: str,
    event_dict: structlog.types.EventDict,
) -> structlog.types.EventDict:
    """清理 structlog 特有元字段,避免污染最终日志输出"""
    _ = logger, method_name
    _ = event_dict.pop("_from_structlog", None)
    _ = event_dict.pop("_record", None)
    _ = event_dict.pop("message", None)  # message 已由 ConsoleRenderer 单独处理
    return event_dict


class MyLoggingPlugin(_pytest.logging.LoggingPlugin):
    def _create_formatter(self, log_format, log_date_format, auto_indent):
        # 使用 ProcessorFormatter 替代原生 Formatter
        return structlog.stdlib.ProcessorFormatter(
            processors=[
                structlog.stdlib.add_logger_name,
                structlog.stdlib.add_log_level,
                structlog.stdlib.PositionalArgumentsFormatter(),
                structlog.processors.StackInfoRenderer(),
                structlog.processors.format_exc_info,
                structlog.processors.UnicodeDecoder(),
                structlog.stdlib.ExtraAdder(),  # 将 logging.extra → event_dict
                drop_structlog_fields,          # 清理冗余字段
                structlog.dev.ConsoleRenderer(),  # 格式化为 'key=value' 形式
            ]
        )


def pytest_configure(config):
    # 替换 pytest 默认日志插件类(注意:依赖内部 API,需关注 pytest 版本兼容性)
    log_plugin = config.pluginmanager.get_plugin("logging")
    if log_plugin is not None:
        log_plugin.LoggingPlugin = MyLoggingPlugin

✅ 效果验证:运行 pytest --log-cli-level INFO 后,原输出:

Miller CSV TSV JSON 数据处理器
Miller CSV TSV JSON 数据处理器

Miller (mlr) 是一个命令行工具,用于查询、整形和重新格式化名称索引数据,如 CSV、TSV、JSON 和 JSON Lines。它将 awk、sed、cut、join 和 sort 的功能整合到一个专为结构化数据处理而构建的单一工具中。

下载
----------------------------------------- Captured log call -----------------------------------------
INFO     root:test_me.py:8 message here

将变为:

----------------------------------------- Captured log call -----------------------------------------
[info     ] message here                   [root] baz='qux' foo='bar'

⚠️ 注意事项:

  • 此方案依赖 _pytest.logging.LoggingPlugin 等内部类,属于 pytest 实现细节,未来版本可能变更接口。建议在 pyproject.toml 或 setup.cfg 中锁定兼容的 pytest 版本(如 pytest>=7.4,),并在升级前回归测试。
  • ConsoleRenderer 默认启用颜色和缩进;若需纯文本(如 CI 日志),可替换为 structlog.dev.ConsoleRenderer(colors=False, pad_event=0)。
  • 若项目未使用 structlog,可自行实现轻量级处理器(如遍历 event_dict 构造 f"{' '.join(f'{k}={repr(v)}' for k, v in event_dict.items())}"),但 structlog.dev.ConsoleRenderer 已提供健壮、可扩展的解决方案。

综上,通过定制 LoggingPlugin 并集成 ProcessorFormatter,我们成功将动态 extra 字段无缝注入 pytest 的捕获日志流,兼顾灵活性与可维护性,是 structlog + pytest 场景下的推荐实践。

热门AI工具

更多
SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

更多
Kratos框架HTTP与gRPC服务开发教程
Kratos框架HTTP与gRPC服务开发教程

本专题围绕Kratos框架双协议服务开发,涵盖HTTP路由与处理器编写、参数获取、gRPC服务实现与客户端调用、metadata上下文传递、encoding编解码注册、统一响应封装、超时控制与流式响应实现方法。

0

2026.10.10

Kratos框架Protobuf接口定义与代码生成合集
Kratos框架Protobuf接口定义与代码生成合集

本专题讲解Kratos框架接口定义体系,涵盖proto编写规范、proto add/client/server生成命令、http注解路由、validate校验、OpenAPI文档生成、跨服务proto复用与兼容性设计。

0

2026.10.10

C++虚函数怎么定义和调用
C++虚函数怎么定义和调用

C++虚函数是实现运行时多态的重要机制。本专题从virtual关键字的基本用法入手,介绍基类与派生类之间的函数重写、基类指针调用派生类方法,以及动态绑定的执行过程,帮助初学者掌握虚函数的核心语法。

0

2026.10.10

C++类与对象的封装方法教程
C++类与对象的封装方法教程

C++封装是面向对象编程的核心特性之一,通过类将数据与操作数据的函数组织在一起,并利用访问权限控制外部访问。本专题介绍类的定义、成员变量、成员函数以及public、private和protected的使用方法,帮助初学者掌握封装的基本原理。

0

2026.10.10

C++构造函数定义与调用方法
C++构造函数定义与调用方法

C++构造函数用于初始化类对象,是面向对象编程的重要基础。本专题从构造函数的定义、声明和调用入手,介绍默认构造函数、带参数构造函数、拷贝构造函数及成员初始化列表,帮助初学者掌握对象创建与初始化的基本方法。

0

2026.10.10

Kratos框架零基础入门教程
Kratos框架零基础入门教程

本专题整理Kratos框架入门内容,涵盖Go环境准备、kratos CLI安装升级、new命令创建项目、目录结构分层说明、服务启动与双协议端口、依赖下载报错排查,帮助开发者快速跑通第一个Kratos框架微服务应用。

0

2026.10.10

C++条件判断语句怎么写
C++条件判断语句怎么写

C++条件判断是控制程序执行流程的重要基础。本专题介绍if、if-else、else if和switch等常见分支语句,结合条件表达式、比较运算符与代码示例,帮助初学者掌握不同场景下的判断逻辑。

0

2026.10.10

C++变量怎么声明和赋值
C++变量怎么声明和赋值

C++变量是编写程序和存储数据的基础。本专题围绕变量声明、定义、初始化、赋值和类型选择等内容展开,帮助初学者理解不同变量的用法,并掌握在实际代码中定义和使用变量的方法。

0

2026.10.10

C++运算符基础入门
C++运算符基础入门

本专题详细讲解了C++运算符的类型、语法与使用方法,涵盖算术运算符、关系运算符、逻辑运算符、位运算符、赋值运算符、条件运算符及其他特殊运算符,并通过代码示例解析优先级与结合性。

0

2026.10.09

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
夸克AI浏览器使用手册
夸克AI浏览器使用手册

共0课时 | 0人学习

jQuery官方API文档
jQuery官方API文档

共0课时 | 0人学习

Manus AI 入门手册
Manus AI 入门手册

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn