如何从Python异常对象中提取包含上层调用函数的完整调用栈字符串？

嘿，这个问题其实很好解决！先看你现在的代码，之所以输出里没显示函数A的调用信息，大概率是示例输出简化了——实际上当你在B里捕获异常时，e.__traceback__本身就包含了完整的调用链，包括A()调用B()的那一层。不过不管怎样，我们可以用更稳妥的方式获取完整栈并保存为字符串：

方法一：用traceback.format_exc()（最简便） #

这个方法会自动捕获当前异常的完整调用栈，并直接返回格式化好的字符串，完全满足你“保存为字符串”的需求：

import traceback

def B():
    try:
        raise Exception()
    except Exception as e:
        # 直接获取完整异常栈字符串并赋值给变量
        full_traceback_str = traceback.format_exc()
        # 这里可以把字符串存起来，比如写入文件或者后续使用
        return full_traceback_str

def A():
    return B()

# 获取保存好的完整栈字符串
saved_traceback = A()

# 验证一下：打印保存的字符串
print("保存的完整调用栈：")
print(saved_traceback)

运行这段代码，你会得到包含A()、B()以及顶层调用的完整栈信息，比如：

Traceback (most recent call last):
  File "test.py", line 13, in <module>
    saved_traceback = A()
  File "test.py", line 10, in A
    return B()
  File "test.py", line 4, in B
    raise Exception()
Exception

方法二：手动用format_exception确保完整栈 #

如果你还是想沿用自己最初的format_exception写法，只需要确保传入的traceback_ob是完整的（它本来就是完整的），代码调整后一样能得到完整栈：

import traceback

def B():
    try:
        raise Exception()
    except Exception as e:
        traceback_ob = e.__traceback__
        # 拼接完整栈字符串
        full_traceback_str = ''.join(traceback.format_exception(type(e), e, traceback_ob))
        return full_traceback_str

def A():
    return B()

saved_traceback = A()
print(saved_traceback)

额外场景：无异常时获取调用栈 #

如果需要在没有抛出异常的情况下获取当前调用栈，比如想记录函数的调用链路，可以用traceback.format_stack()：

import traceback

def B():
    # 获取当前调用栈并转为字符串
    stack_str = ''.join(traceback.format_stack())
    return stack_str

def A():
    return B()

saved_stack = A()
print(saved_stack)

输出会包含从顶层到B()的完整调用路径：

File "test.py", line 11, in <module>
    saved_stack = A()
  File "test.py", line 8, in A
    return B()
  File "test.py", line 5, in B
    stack_str = ''.join(traceback.format_stack())

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