如何从bash脚本发送SIGINT给Python?

14 投票
5 回答
15537 浏览
提问于 2025-04-15 12:09

我想通过一个bash脚本来启动一个在后台运行的Python任务,然后用SIGINT信号优雅地结束它。在命令行里这样做没问题,但在脚本里就不行了。

这是我的loop.py文件:

#! /usr/bin/env python
if __name__ == "__main__":
    try:
        print 'starting loop'
        while True:
            pass
    except KeyboardInterrupt:
        print 'quitting loop'

在命令行中我可以中断这个任务:

$ python loop.py &
[1] 15420
starting loop
$ kill -SIGINT 15420
quitting loop
[1]+  Done                    python loop.py

这是我的kill.sh脚本:

#! /bin/bash
python loop.py &
PID=$!
echo "sending SIGINT to process $PID"
kill -SIGINT $PID

但是在脚本中我却无法做到:

$ ./kill.sh 
starting loop
sending SIGINT to process 15452
$ ps ax | grep loop.py | grep -v grep
15452 pts/3    R      0:08 python loop.py

而且,如果这个任务是通过脚本启动的,我就无法再从命令行中结束它了:

$ kill -SIGINT 15452
$ ps ax | grep loop.py | grep -v grep
15452 pts/3    R      0:34 python loop.py

我想我可能忽略了bash作业控制的一些细节。

5 个回答

4

我同意Matthew Flaschen的看法;问题出在Python上,显然当不是从交互式命令行调用时,它没有正确处理SIGINT信号下的KeyboardInterrupt异常。

当然,你可以像这样注册你的信号处理器:

def signal_handler(signum, frame):
    raise KeyboardInterrupt, "Signal handler"
7

除了@matthew-flaschen的回答,你还可以在bash脚本中使用exec,这样可以有效地把当前的作用域替换为正在打开的进程:

#!/bin/bash
exec python loop.py &
PID=$!
sleep 5  # waiting for the python process to come up

echo "sending SIGINT to process $PID"
kill -SIGINT $PID
17

你没有注册一个信号处理器。试试下面的代码,它似乎比较可靠。我觉得出现例外的情况很少,通常是在 Python 注册脚本的处理器之前就捕获到了信号。需要注意的是,KeyboardInterrupt 这个错误只应该在“用户按下中断键的时候”被触发。我认为它能在明确的(比如通过 kill 命令)SIGINT 信号下工作,实际上是实现上的一个意外。

import signal

def quit_gracefully(*args):
    print 'quitting loop'
    exit(0);

if __name__ == "__main__":
    signal.signal(signal.SIGINT, quit_gracefully)

    try:
        print 'starting loop'
        while True:
            pass
    except KeyboardInterrupt:
        quit_gracefully()

撰写回答