如何禁止Python的breakpoint()自动读取全部STDIN输入？

当执行带管道输入的Python脚本（例如seq 1 10 | python tmp.py），且脚本中包含breakpoint()时，调试器会自动读取并执行STDIN中剩余的所有内容，导致调试提前终止并抛出bdb.BdbQuit异常。

示例脚本 #

import sys
print("Hello, world!")
for i, line in enumerate(sys.stdin):
    print(line)
    print(f"Before breakpoint: {i}")
    breakpoint()
    print(f"After breakpoint: {i}")

执行后异常输出 #

seq 1 10 | python tmp.py 
Hello, world!
1

Before breakpoint: 0
> .../tmp.py(9)<module>()
-> print(f"After breakpoint: {i}")
(Pdb) 2
(Pdb) 3
(Pdb) 4
(Pdb) 5
(Pdb) 6
(Pdb) 7
(Pdb) 8
(Pdb) 9
(Pdb) 10
(Pdb) 
Traceback (most recent call last):
  File ".../tmp.py", line 9, in <module>
    print(f"After breakpoint: {i}")
  File ".../tmp.py", line 9, in <module>
    print(f"After breakpoint: {i}")
  File ".../python3.10/bdb.py", line 90, in trace_dispatch
    return self.dispatch_line(frame)
  File ".../python3.10/bdb.py", line 115, in dispatch_line
    if self.quitting: raise BdbQuit
bdb.BdbQuit

解决方案 #

方法1：直接指定pdb的输入源 #

替换脚本中的breakpoint()为手动初始化的pdb实例，将输入源指定为终端：

import pdb
# 让pdb读取终端输入，而非管道STDIN
pdb.Pdb(stdin=open('/dev/tty')).set_trace()

调试时，输入将来自终端，管道中的剩余STDIN内容会保留给后续代码处理。

方法2：临时切换STDIN并恢复 #

在进入断点前后，手动切换STDIN为终端，调试完成后恢复原始输入：

import sys
import pdb

# 保存原始STDIN
original_stdin = sys.stdin
# 替换为终端输入
sys.stdin = open('/dev/tty')
# 启动调试
pdb.set_trace()
# 恢复原始STDIN，继续处理管道数据
sys.stdin = original_stdin

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