python: 在从套接字读取数据的过程中捕获 SIGINT

Question

我有一个 python 脚本，我想通过使用命令行发送 "SIGINT" 信号来停止它：

kill -2 <PID>

该脚本生成一个子进程，然后进入一个无限循环，尝试从套接字接收数据。我已经安装了信号处理程序，但看起来我的程序没有捕捉到信号。这是我的代码：

import signal
import multiprocessing

def newBreakHandler(signum, frame):
    global progBreak
    progBreak = True


def runMain():

    client = multiprocessing.Client(<address>)
    # ......

    # Insert break handler
    signal.signal(signal.SIGINT, newBreakHandler)


    while not progBreak:
        try:
            # Some other stuff...
            # ...
            childProc = multiprocessing.Process(target=childProcName)                                                                                                                          
            childProc.start()


            # Main loop
            while not progBreak:
                command = client.recv_bytes()
        except:
            pass

    print "Exiting..."

问题是每当我发送：

kill -2 <PID>

我从来没有看到打印的 "Exiting" 文本并且 PID 没有被杀死。我认为这是因为内部 "while" 循环正忙于等待来自客户端的新数据，但是如果客户端没有发送新数据。

有没有办法解决这个问题？

TIA!!!

Answer 1

看起来你的内部循环正在阻塞 progBreak 检查，这段代码对我有用：

import signal

progBreak = False

def newBreakHandler(signum, frame):
    global progBreak
    progBreak = True


def runMain():
    global progBreak

    signal.signal(signal.SIGINT, newBreakHandler)


    while not progBreak:
        while not progBreak and client.poll():
            command = client.recv_bytes()

    print "Exiting..."

runMain()

python: 在从套接字读取数据的过程中捕获 SIGINT

python: catch SIGINT in process reading data from a socket

python

signals

multiprocessing

sigint