Python3调试：如何查看调用栈中活跃try块的嵌套结构？

Python标准库没有直接提供获取运行时活跃try块的API，但可以通过结合运行时栈帧获取+AST静态分析的方式实现需求，核心思路是：遍历调用栈的每个帧，获取当前执行行号，再对对应函数的源代码做AST解析，找到包含当前行的活跃try块（即当前代码处于try的body范围内），最后按嵌套层级输出。

实现代码 #

import inspect
import ast

def find_active_trys(node, current_line, active_trys):
    # 遍历AST节点，找到包含当前行的活跃try块
    if isinstance(node, ast.Try):
        # 判断当前行是否在try的body范围内
        body_start = node.body[0].lineno
        # 处理Python 3.8+的end_lineno，更早版本用body最后一个节点的lineno
        body_end = node.body[-1].end_lineno if hasattr(node.body[-1], 'end_lineno') else node.body[-1].lineno
        if body_start <= current_line <= body_end:
            # 收集try块的处理类型（except/finally）
            handler_info = []
            if node.except_handlers:
                for handler in node.except_handlers:
                    exc_type = ast.get_source_segment(inspect.getsource(node), handler.type) if handler.type else ""
                    handler_info.append(f'except {exc_type}'.strip())
            if node.finalbody:
                handler_info.append('finally')
            active_trys.append((node.lineno, ' / '.join(handler_info)))
            # 递归遍历try body内的节点，找嵌套的try块
            for subnode in node.body:
                find_active_trys(subnode, current_line, active_trys)
    # 遍历其他子节点
    for field, value in ast.iter_fields(node):
        if isinstance(value, list):
            for item in value:
                if isinstance(item, ast.AST):
                    find_active_trys(item, current_line, active_trys)
        elif isinstance(value, ast.AST):
            find_active_trys(value, current_line, active_trys)

def get_active_trys():
    active_try_output = []
    # 遍历调用栈，跳过当前函数自身的帧
    for frame_info in inspect.stack()[1:]:
        frame = frame_info.frame
        func_name = frame_info.function
        # 尝试获取函数源代码，无法获取则跳过（如内置函数）
        try:
            source = inspect.getsource(frame)
        except OSError:
            continue
        tree = ast.parse(source)
        current_line = frame_info.lineno
        trys_in_func = []
        find_active_trys(tree, current_line, trys_in_func)
        # 按嵌套层级添加缩进，输出信息
        base_indent = '    ' * len(active_try_output)
        for try_line, handler_str in trys_in_func:
            indent = base_indent + '    ' * trys_in_func.index((try_line, handler_str))
            active_try_output.append(f'{indent}{func_name}: try / {handler_str}')
    return '\n'.join(active_try_output)

# 示例代码
def fn1():
    try:
        fn2()
    except IOError:
        pass

def fn2():
    try:
        fn3()
    except NameError:
        pass

def fn3():
    print(get_active_trys())

if __name__ == "__main__":
    fn1()

输出结果 #

运行上述代码后，fn3中的print会输出：

fn1: try / except IOError
        fn2: try / except NameError

说明与限制 #

• 支持场景：可以识别try/except、try/finally、try/except/finally结构，以及嵌套的try块。
• 依赖条件：需要能获取到源代码（不能是仅编译为.pyc的文件或内置函数），且Python版本建议3.8+（更早版本的AST没有end_lineno，行号判断可能存在误差）。
• 备选方案：如果不需要详细的try块信息，仅需获取包含活跃try块的函数名列表，可以简化逻辑，只收集每个栈帧中存在活跃try块的函数名，返回[fn1, fn2]格式。

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