为何Python单行while语句无法捕获异常？

按下Ctrl+C触发KeyboardInterrupt时，以下两类代码表现完全不同：

可正常捕获异常的代码 #

这类代码会正确捕获异常并打印a：

示例1：多行写法的while循环 #

try:
    while True:
        pass
except KeyboardInterrupt:
    print("a")

示例2：try块内包含前置语句的单行while #

try:
    i = 0
    while True: pass
except KeyboardInterrupt:
    print("a")

无法捕获异常的代码 #

这类代码无法捕获KeyboardInterrupt，会直接抛出回溯信息：

示例3：try块内仅包含单行while循环 #

try:
    while True: pass
except KeyboardInterrupt:
    print("a")

示例4：单行while后跟随不可达语句 #

try:
    while True: pass
    i = 0
except KeyboardInterrupt:
    print("a")

底层原因分析 #

Python 3.11新增了JUMP_BACKWARD字节码指令，该指令的行为差异是导致此问题的关键。

在Python 3.12中对两类代码反汇编后，字节码存在明显差异：

不可捕获异常的字节码 #

0           0 RESUME                   0

  2           2 NOP

  3     >>    4 JUMP_BACKWARD            1 (to 4)
        >>    6 PUSH_EXC_INFO

  4           8 LOAD_NAME                0 (KeyboardInterrupt)
             10 CHECK_EXC_MATCH
             12 POP_JUMP_IF_FALSE       11 (to 36)
             14 POP_TOP

  5          16 PUSH_NULL
             18 LOAD_NAME                1 (print)
             20 LOAD_CONST               1 ('a')
             22 CALL                     1
             30 POP_TOP
             32 POP_EXCEPT
             34 RETURN_CONST             2 (None)

  4     >>   36 RERAISE                  0
        >>   38 COPY                     3
             40 POP_EXCEPT
             42 RERAISE                  1
ExceptionTable:
  4 to 4 -> 6 [0]
  6 to 30 -> 38 [1] lasti
  36 to 36 -> 38 [1] lasti
None

可捕获异常的字节码 #

0           0 RESUME                   0

  2           2 NOP

  3           4 NOP

  4     >>    6 NOP

  3           8 JUMP_BACKWARD            2 (to 6)
        >>   10 PUSH_EXC_INFO

  5          12 LOAD_NAME                0 (KeyboardInterrupt)
             14 CHECK_EXC_MATCH
             16 POP_JUMP_IF_FALSE       11 (to 40)
             18 POP_TOP

  6          20 PUSH_NULL
             22 LOAD_NAME                1 (print)
             24 LOAD_CONST               1 ('a')
             26 CALL                     1
             34 POP_TOP
             36 POP_EXCEPT
             38 RETURN_CONST             2 (None)

  5     >>   40 RERAISE                  0
        >>   42 COPY                     3
             44 POP_EXCEPT
             46 RERAISE                  1
ExceptionTable:
  4 to 8 -> 10 [0]
  10 to 34 -> 42 [1] lasti
  40 to 40 -> 42 [1] lasti
None

两者核心差异：

• 可捕获的代码多了两条NOP指令
• JUMP_BACKWARD的目标地址不同：不可捕获的代码中该指令直接跳转到自身地址形成循环，而可捕获的代码跳转到前置的NOP地址
• 异常表的覆盖范围不同：不可捕获的代码中，循环指令所在的地址未被包含在异常处理的覆盖范围内

补充验证：嵌套try-except也无法解决 #

即使使用嵌套的try-except结构，依然无法捕获该异常，代码示例：

try:
    try:
        while True: pass
    except KeyboardInterrupt:
        print("a")
except Exception:
    print("b")

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