如何使用Python Click的ctx.with_resource捕获子命令/组的回溯信息

问题背景 #

使用Click的Context.with_resource实现stdout重定向到文件时功能正常，但尝试用它捕获根CLI组及子命令的异常回溯时失败：异常仍会被打印到终端，而非写入指定文件。原因是Click默认会先捕获异常，待with_resource的上下文管理器退出后再重新抛出，导致自定义的异常捕获逻辑无法生效。需求是：捕获所有组/命令的异常，将回溯信息写入指定文件，不向终端输出回溯，最后退出程序。

解决方案 #

核心是关闭Click的自动异常捕获机制，让异常在with_resource上下文的作用域内抛出，从而被自定义逻辑捕获处理。

修改后的完整代码：

import contextlib
import sys
import traceback
import click

@contextlib.contextmanager
def capture_traceback(file):
    with open(file, mode="w") as fp:       
        try:
            yield 
        except Exception:
            # 将异常回溯写入目标文件
            traceback.print_exc(file=fp)
            # 静默退出程序，不向终端输出回溯内容
            sys.exit(1)

# 关闭Click默认的异常捕获，让异常传播到上下文管理器中
@click.group(context_settings={"catch_exceptions": False})
@click.pass_context
@click.argument("file")
def cli(ctx, file):
    ctx.with_resource(capture_traceback(file))

@cli.command()
def hello():
    raise ValueError("error!")

if __name__ == "__main__":
    cli()

关键说明 #

• context_settings={"catch_exceptions": False}：禁用Click自动捕获异常的默认行为，确保异常能进入capture_traceback上下文的try/except块。
• 捕获Exception而非裸except：更严谨，避免意外捕获SystemExit等特殊基础异常，保证sys.exit(1)能正常执行退出程序。
• 移除冗余的print("exception!")：避免向终端输出额外信息，实现完全静默的异常处理。

内容的提问来源于stack exchange，提问作者Adam