python - 使用 SIGINT 终止 Python 3 中的函数

Question

以下代码为例:

import signal
import time

def stop(signal, frame):
    print("You pressed ctrl-c")
    # stop counter()

def counter():
    for i in range(20):
        print(i+1)
        time.sleep(0.2)

signal.signal(signal.SIGINT, stop)
while True:
    if(input("Do you want to count? ")=="yes"):
        counter()

如何让 stop() 函数终止或中断 counter() 函数，使其返回到提示符？

输出示例:

Do you want to count? no
Do you want to count? yes
1
2
3
4
5
6
7
You pressed ctrl-c
Do you want to count?

我使用的是 Python 3.5.2。

Answer 1

您可以使用 KeyboardInterrupt 异常，而不是定义自己的 SIGINT 处理程序:

while input("Do you want to count? ").strip().casefold() == "yes":
    try:
        counter()
    except KeyboardInterrupt:
        print("You pressed ctrl-c")