技術(shù)背景

對(duì)于一些連續(xù)運(yùn)行或者長(zhǎng)時(shí)間運(yùn)行的Python程序而言，如服務(wù)器的后端，或者是長(zhǎng)時(shí)間運(yùn)行的科學(xué)計(jì)算程序。當(dāng)我們涉及到一些中途退出的操作時(shí)，比如使用Ctrl+C來(lái)退出正在運(yùn)行的程序。這種場(chǎng)景的出現(xiàn)一般有兩個(gè)可能性：一是程序出現(xiàn)了問(wèn)題，需要終止程序來(lái)對(duì)其進(jìn)行調(diào)整。另一種是程序本身是正確的，但是程序運(yùn)行的速度太慢了，也有可能是想提前結(jié)束，這種場(chǎng)景下很多時(shí)候我們是希望可以保留其相應(yīng)的計(jì)算結(jié)果的。但是如果我們使用的是一些第三方的數(shù)據(jù)存儲(chǔ)格式來(lái)存儲(chǔ)數(shù)據(jù)，不一定可以支持連續(xù)的存儲(chǔ)，非常常見(jiàn)的是在程序執(zhí)行結(jié)束之后，再將結(jié)果進(jìn)行保存。但是由于程序被提前終止了，此時(shí)就需要一些特殊的手段來(lái)對(duì)中途終止的程序的結(jié)果進(jìn)行保存。

基礎(chǔ)案例

我們先來(lái)看一個(gè)比較簡(jiǎn)單的案例：一個(gè)普通的打印數(shù)字的程序，每隔1s的時(shí)間就打印一個(gè)數(shù)字出來(lái)，我們可以使用python的signal.signal來(lái)捕獲這個(gè)終止信號(hào)。

# signal_exit.py
import signal
import sys
 
def signal_handler(signal, frame):
    print ('\nSignal Catched! You have just type Ctrl+C!')
    sys.exit(0)
 
if __name__ == '__main__':
    import time
 
    signal.signal(signal.SIGINT, signal_handler)
    for x in range(100):
        time.sleep(1)
        print (x)

當(dāng)我們運(yùn)行這個(gè)程序到一半時(shí)，同時(shí)按下Ctrl+C，我們會(huì)得到如下的結(jié)果：

$ python3 signal_exit.py
0
1
2
^C
Signal Catched! You have just type Ctrl+C!

這個(gè)結(jié)果表明，我們?cè)诔绦蜻\(yùn)行的過(guò)程中捕獲到了Ctrl+C的這個(gè)外部操作，并且對(duì)該操作進(jìn)行了相應(yīng)的處理之后，才終止了程序的運(yùn)行。需要注意的是，如果此時(shí)不加上sys.exit(0)這個(gè)終止的操作，這個(gè)程序不會(huì)被停止，會(huì)繼續(xù)運(yùn)行下去，相當(dāng)于只是捕獲了異常終止信號(hào)但不做任何的處理。

給終止信號(hào)傳入外部參數(shù)

在上面的一個(gè)案例中，僅僅只是捕獲了“終止運(yùn)行”的這個(gè)外部信號(hào)，但是如果更進(jìn)一步的，我們想捕獲到最后一個(gè)輸出的數(shù)字是多少，這個(gè)時(shí)候要如何操作呢？signal.signal函數(shù)本身并不支持很多的參數(shù)傳入，此時(shí)建議采取的是自行創(chuàng)建一個(gè)類(lèi)，將signal_handler函數(shù)封裝為類(lèi)的成員函數(shù)，這樣我們就可以獲取到相應(yīng)的內(nèi)部參數(shù)，如下面這個(gè)案例所示：

# signal_exit.py
import signal
import sys
import time
 
class Printer:
    def __init__(self):
        self.x = 0
        signal.signal(signal.SIGINT, self.signal_handler)
 
    def signal_handler(self, signal, frame):
        print ('\nSignal Catched! You have just type Ctrl+C! The last number is: {}'.format(self.x))
        sys.exit(0)
 
    def run(self, counter=10):
        while self.x < counter:
            print (self.x)
            time.sleep(1)
            self.x += 1
 
if __name__ == '__main__':
    printer = Printer()
    printer.run(counter=100)

此時(shí)如果在程序正在運(yùn)行的狀態(tài)下同時(shí)按下Ctrl+C，得到的結(jié)果如下：

$ python3 signal_exit.py
0
1
2
3
^C
Signal Catched! You have just type Ctrl+C! The last number is: 3

可以看到，我們成功的捕獲到了最后一個(gè)被輸出出來(lái)的參數(shù)。

總結(jié)概要

當(dāng)我們準(zhǔn)備去殺死一個(gè)進(jìn)程時(shí)，從程序設(shè)計(jì)的本身來(lái)考慮，我們應(yīng)當(dāng)要設(shè)計(jì)一定的保護(hù)方案來(lái)確保程序被非正常終止時(shí)，相應(yīng)的計(jì)算結(jié)果也能夠被很好的保存下來(lái)。在Python中可以使用signal.signal函數(shù)來(lái)實(shí)現(xiàn)這樣的功能，但是如果要實(shí)現(xiàn)數(shù)據(jù)的保存功能，需要結(jié)合一個(gè)實(shí)際的類(lèi)來(lái)實(shí)現(xiàn)。

到此這篇關(guān)于Python3實(shí)現(xiàn)捕獲Ctrl+C終止信號(hào)的文章就介紹到這了,更多相關(guān)Python捕獲Ctrl+C終止信號(hào)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論