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

轻丽小哥_8111

轻丽小哥_8111

2026-10-10

605人浏览

原创

如何在 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 场景下的推荐实践。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2026.10.10

0

15

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

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

2026.10.10

0

26

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

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

2026.10.10

0

32

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

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

2026.10.10

0

16

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

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

2026.10.10

0

15

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

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

2026.10.10

0

13

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

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

2026.10.10

0

20

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

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

2026.10.09

0

11

PixPix官网入口合集
PixPix官网入口合集

本专题汇总了PixPix官网在线使用入口及平台功能详解,涵盖文生图、图生图、AI图片编辑、AI视频创作等核心能力,并整理了AI爆款图片复刻、商品套图、详情页生成、视频变清晰与去水印等电商专项工具的使用教程。同时收录了PixPix MCP接入Codex、Claude Code等主流Agent的操作指南,助您一站式完成AI图片与视频创作。

2026.10.09

0

11

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Sass 教程
Sass 教程

共14课时 | 1.5万人学习

PHP入门速学(台湾同胞版)
PHP入门速学(台湾同胞版)

共10课时 | 1.6万人学习

韩顺平 2016年 最新PHP基础视频教程
韩顺平 2016年 最新PHP基础视频教程

共47课时 | 12.1万人学习