在 Python 3 中使用 SIGINT 杀死一个函数

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")

Answer 2

您可以在stop 中引发异常，这将停止counter 的执行并搜索最近的异常处理程序（您在while True 循环中设置）。

即创建自定义异常：

class SigIntException(BaseException): pass

在stop提出它：

def stop(signal, frame):
    print("You pressed ctrl-c")
    raise SigIntException

并在您的 while 循环中捕获它：

while True:
    if(input("Do you want to count? ")=="yes"):
        try:        
            counter()
        except SigIntException:
            pass

它会按照您需要的方式运行。