如何在Python脚本报错时自动启动交互式会话以调试？

这个需求太戳调试痛点了！跑了好久的脚本突然崩了，只能对着干巴巴的报错信息瞎猜，要是能直接在报错现场扒变量、看调用栈，效率能翻好几倍。给你整理了几个靠谱的方案，从原生工具到增强版都有：

Python标准库自带的pdb调试器就能搞定这个需求，不用装任何第三方包，非常省心。

• 直接命令行启动（不用改脚本）：
  把你平时的python script.py换成下面这个命令：

  python -m pdb -c continue script.py
  

  解释下参数：-m pdb让Python以调试模式运行脚本，-c continue告诉调试器“先正常跑脚本，别一开始就停在第一行”。等脚本出错抛出异常时，程序会立刻暂停在报错的代码行，自动进入pdb的交互式会话——你可以用p 变量名看变量值，l看当前代码上下文，bt看完整调用栈，甚至修改代码后继续运行。

• 针对特定代码块设置（灵活可控）：
  如果你只想在某段风险较高的代码里启用这个功能，可以手动捕获异常并触发事后调试：

  import pdb
  import traceback
  
  try:
      # 这里放你觉得可能会崩的核心逻辑
      your_long_running_task()
  except Exception:
      # 先打印完整异常信息，再进入调试
      traceback.print_exc()
      pdb.post_mortem()
  

  这样这段代码出错时，会先把报错信息打出来，然后自动进入调试环境，其他代码不受影响。

如果你习惯IPython的语法补全、彩色输出，那ipdb绝对是你的菜——它是IPython版的pdb，调试体验提升不止一个档次。

• 先安装ipdb：

  pip install ipdb
  

• 命令行启动方式：
  和pdb几乎一样，只是换成ipdb模块：

  python -m ipdb -c continue script.py
  

  出错后进入的ipdb会话支持tab补全、语法高亮，查看复杂变量（比如字典、对象）时格式更清晰，比原生pdb好用太多。

• 脚本内设置的方式：
  逻辑和pdb一致，换成ipdb的方法就行：

  import ipdb
  import traceback
  
  try:
      your_long_running_task()
  except Exception:
      traceback.print_exc()
      ipdb.post_mortem()
  

要是你不想每次运行脚本都敲长长的命令，可以通过修改Python的全局异常钩子，让任何未捕获的异常都自动触发调试。

写一个简单的包装脚本，比如叫debug_runner.py：

import sys
import ipdb  # 想用原生pdb就换成 import pdb

def debug_on_error(type, value, tb):
    # 先打印完整异常栈
    import traceback
    traceback.print_exception(type, value, tb)
    # 进入交互式调试
    ipdb.post_mortem(tb)  # 原生pdb就写 pdb.post_mortem(tb)

# 替换默认的异常处理逻辑
sys.excepthook = debug_on_error

# 导入并运行你的脚本
import script

之后运行脚本就用：

python debug_runner.py

不管脚本哪里出错，都会自动进入调试会话，再也不用记复杂的命令了。

• 在调试会话里输入help可以查看所有可用命令，常用的比如：
  • l：查看当前代码的上下文
  • p 变量名：打印变量的值
  • bt：查看完整的调用栈
  • n：执行下一行代码
  • c：继续运行脚本（如果修复了问题可以试试）
  • q：退出调试会话

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